从 Python 中的相邻对恢复数组的程序
假设我们有一个大小为n-1的名为adPair的二维数组,其中每个adPair[i]有两个元素[ui,vi]表示元素ui和vi在名为nums的数组中相邻,在nums中有n个唯一元素。我们必须找到数组nums。如果有多个解决方案,则返回其中任何一个。
所以,如果输入像adPair=[[3,2],[4,5],[4,3]],那么输出将是[2,3,4,5]
示例
让我们看看以下实现以获得更好的理解-
from collections import defaultdict
def solve(adPair):
   my_map = defaultdict(list)
   for a, b in adPair:
      my_map[a].append(b)
      my_map[b].append(a)
   for a, l in my_map.items():
      if len(l) == 1:
         nums = [a, l[0]]
         break
   for i in range(1, len(adPair)):
      a, b = my_map[nums[-1]]
      if a == nums[-2]:
         nums.append(b)
      else:
         nums.append(a)
   return nums
adPair = [[3,2],[4,5],[4,3]]
print(solve(adPair))输入
[[3,2],[4,5],[4,3]]输出结果
[2, 3, 4, 5]
