А один знакомый JS-ник взял и предложил красивый вариант без рекурсии, и я считаю, он достоин того, чтобы вы его увидели.
Переведено с JS:
def unflatten(result):
li = result[0][0]
final = [[]]
for i in range(len(result)):
if result[i][0] == li:
final[-1].append(result[i][1])
elif result[i][0] > li:
while result[i][0] > li:
final.append([])
li += 1
final[-1].append(result[i][1])
else:
while result[i][0] < li:
final[-2].append(final.pop())
li -= 1
final[-1].append(result[i][1])
li = result[i][0]
return final[0]
У меня не получилось. Ошибка перевода на питон? print(unflatten([[0, 'a'], [1, 'b']])) > ['a']
какой тяжелый и плохочитаемый код. а формулировку задачки можно?
вроде попроще получилось def unflatten(l): result = [] def dig(target, lvl): if lvl == 0: return target for _ in range(lvl): if not isinstance(target[-1], list): target.append([]) target = target[-1] return target for lvl, value in l: target = dig(result, lvl) target.append(value) return result
Обсуждают сегодня