LeetCode 1743. 从相邻元素对还原数组

一、题目：从相邻元素对还原数组

1. 描述

存在一个由 $n$ 个不同元素组成的整数数组 nums ，但你已经记不清具体内容。好在你还记得 nums 中的每一对相邻元素。给你一个二维整数数组 adjacentPairs ，大小为 $n-1$ ，其中每个 adjacentPairs[i] = [num1, num2] 表示元素 num1 和 num2 在数组 nums 相邻，且一定存在于adjacentPairs ，存在形式可能是 [...num1, num2...] ，也可能是 [...nums2, num1...]（adjacentPairs[i]只表示两个元素相邻，并不保证其 左-右 顺序） 。返回原始数组 nums。如果存在多种解答，返回 其中任意一个即可。

2. 提示和要求

• nums.length == n
• adjacentPairs.length == n - 1
• adjacentPairs[ i ].length == 2
• 2 <= n <= 105
• -105 <= nums[ i ] <= 105

3. 示例

• 示例一

输入：adjacentPairs = [[2,1],[3,4],[3,2]]
输出：[1,2,3,4]

• 示例二

输入：adjacentPairs = [[4,-2],[1,4],[-3,1]]
输出：[-2,4,1,-3]
//另一种解答是 [-3,1,4,-2] ，也是正确答案。

• 示例三

输入：adjacentPairs = [[100000,-100000]]
输出：[100000,-100000]

二、解题

1. 思路

• 需要考虑原数组的第一个元素
  1. 使用 HahMap 记录下每个数的相邻元素
  2. 只有一个相邻元素的数是数组的第一个元素。*（只有一个相邻元素的数有开头和结尾两个，按照题意这两个数都可以作为第一个元素） *
• 根据找到的原数组的第一个数 $e_1$，因为它只有一个相邻元素，所以可以简单地得到HashMap得到与其相邻的第二个元素 $e_2$。
• 显然 $e_2$ 拥有两个相邻元素，只需要排除掉已经和 $e_2$ 连接上的 $e_1$，就可以得到 $e_3$。
• 以此类推就可以得到完整的原数组nums。

2. 代码

class Solution {
    public int[] restoreArray(int[][] adjacentPairs) {
        Map<Integer, List<Integer>> map = new HashMap<Integer,List<Integer>>();
        int n = adjacentPairs.length+1;
        int[] res = new int[n];// 结果数组，代表原数组nums
        
     
        for(int[] adjacentPair:adjacentPairs){
            map.putIfAbsent(adjacentPair[0], new ArrayList());
            map.putIfAbsent(adjacentPair[1], new ArrayList());
            map.get(adjacentPair[0]).add(adjacentPair[1]);
            map.get(adjacentPair[1]).add(adjacentPair[0]);
        }

		//找到原数组nums的第一个元素，并赋值给res[0]
        for(Map.Entry<Integer,List<Integer>> entry : map.entrySet()){
            if(entry.getValue().size()==1){
                res[0] = entry.getKey();
                break;
            }
        }

		//通过res[1]可以简单地得到res[2],因为res[1]只有一个相邻元素
        res[1] = map.get(res[0]).get(0);
        for(int i=2;i<n;i++){
            List<Integer> lastNum = map.get(res[i-1]);
            res[i] = 
                lastNum.get(0)==res[i-2]?lastNum.get(1):lastNum.get(0);
        }
        return res;
    }
}

三、 结果