一、需求
输入 a = [[0], [3,4], [1,2]],要求输出为 [0, 2, 2, 1, 1]
解释:[0] 是第 0 位,所以 0 位置的输出为0,[1,2] 的 index= 2,所以 1 和 2 对应位置的输出是2
二、代码
a = [[0],[3,4],[1,2]]
max_len = 5
rs = [0] * max_len
for i,x in enumerate(a):
# x 分别为 [0],[3,4],[1,2]
for idx in x:
# idx 为 list x 中的每个元素
rs[idx] = i
此时 rs = [0, 2, 2, 1, 1]。