存在一个由 n 个不同元素组成的整数数组 nums ,但你已经记不清具体内容。好在你还记得 nums 中的每一对相邻元素。
给你一个二维整数数组 adjacentPairs ,大小为 n - 1 ,其中每个 adjacentPairs[i] = [ui, vi] 表示元素 ui 和 vi 在 nums 中相邻。
题目数据保证所有由元素 nums[i] 和 nums[i+1] 组成的相邻元素对都存在于 adjacentPairs 中,存在形式可能是 [nums[i], nums[i+1]] ,也可能是 [nums[i+1], nums[i]] 。这些相邻元素对可以 按任意顺序 出现。
返回 原始数组 nums 。如果存在多种解答,返回 其中任意一个 即可。
示例1:
输出:[1,2,3,4]
解释:数组的所有相邻元素对都在 adjacentPairs 中。
特别要注意的是,adjacentPairs[i] 只表示两个元素相邻,并不保证其 左-右 顺序。
示例2:
输出:[-2,4,1,-3]
解释:数组中可能存在负数。
另一种解答是 [-3,1,4,-2] ,也会被视作正确答案。
示例3:
输入: adjacentPairs = [[100000,-100000]]
输出: [100000,-100000]
提示:
nums.length == nadjacentPairs.length == n - 1adjacentPairs[i].length == 22 <= n <= 105-105 <= nums[i], ui, vi <= 105- 题目数据保证存在一些以
adjacentPairs作为元素对的数组nums
思考过程:
首先题目保证了result[]一定存在,其次题目给出了数组中每一个元素的相邻元素信息。那么我们在构建result[]的时候,假如result[0] ~ result[i - 1]已经正确构建,我们想要获得result[i],就需要知道result[i - 1]的邻居信息与res[i -2]的自身信息。具体操作上我们可以用HashMap<i,j>辅助完成,i表示数组元素i,j表示元素i的两个邻居的和,则有result[i] = map.get(result[i - 1]) - result[i - 2];
时间复杂度:O(N)
空间复杂度:O(N)
代码:
public int[] restoreArray(int[][] adjacentPairs) {
//HashMap计数
HashMap<Integer, Integer> map = new HashMap<>();
for (int[] arr : adjacentPairs) {
map.put(arr[0], map.getOrDefault(arr[0], 0) + 1);
map.put(arr[1], map.getOrDefault(arr[1], 0) + 1);
}
//构建邻接关系表, map<i, j>表示元素i的左右邻居的和为j
HashMap<Integer, Integer> adj = new HashMap<>();
for (int[] arr : adjacentPairs) {
adj.put(arr[0], adj.getOrDefault(arr[0], 0) + arr[1]);
adj.put(arr[1], adj.getOrDefault(arr[1], 0) + arr[0]);
}
//寻找开始元素
int start = -1;
for (int i : map.keySet()) {
if (map.get(i) == 1) {
start = i;
break;
}
}
//res[i]的值为res[i - 1]的值的相邻元素的和减去res[i - 2]
int[] res = new int[adjacentPairs.length + 1];
res[0] = start;
for (int i = 1; i < res.length; i++) {
if (i == 1) {
res[i] = adj.get(res[i - 1]);
} else {
res[i] = adj.get(res[i - 1]) - res[i - 2];
}
}
return res;
}
}