一、题目:从相邻元素对还原数组
1. 描述
存在一个由
n
n
n 个不同元素组成的整数数组 nums
,但你已经记不清具体内容。好在你还记得 nums
中的每一对相邻元素。给你一个二维整数数组 adjacentPairs
,大小为
n
−
1
n - 1
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. 思路
- 需要考虑原数组的第一个元素
- 使用 HahMap 记录下每个数的相邻元素
- 只有一个相邻元素的数是数组的第一个元素。*(只有一个相邻元素的数有开头和结尾两个,按照题意这两个数都可以作为第一个元素) *
- 根据找到的原数组的第一个数 e 1 e_{1} e1,因为它只有一个相邻元素,所以可以简单地得到HashMap得到与其相邻的第二个元素 e 2 e_{2} e2。
- 显然 e 2 e_{2} e2 拥有两个相邻元素,只需要排除掉已经和 e 2 e_{2} e2 连接上的 e 1 e_{1} e1,就可以得到 e 3 e_{3} e3。
- 以此类推就可以得到完整的原数组
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;
}
}