题目来源:https://leetcode-cn.com/problems/restore-the-array-from-adjacent-pairs/
大致题意:
给一个二维数组,其中有n-1个大小为2的一维数组,每个一维数组代表一个相邻关系。通过相邻关系确定一个大小为n的数组。所有数字不会重复,答案不唯一。
思路
可知一个数组中,首尾元素只有一个邻点,其他元素有两个邻点。
那么在所有的相邻关系中,首尾元素只会出现一次,其他元素出现两次。
注意答案不唯一,首尾颠倒不影响正确。
哈希表
- 第一次遍历二维数组,使用哈希表存下所有元素及其邻点。
- 第二次遍历哈希表,找出第一个出现次数为1的元素(也就是邻点数为1),将其作为数组首元素,然后再从哈希表中找出该元素的邻点作为第二个元素
- 第三次遍历执行n-2次,每次都找当前数组已存元素的末尾元素的邻点,将其加入数组
public int[] restoreArray(int[][] adjacentPairs) {
Map<Integer, List<Integer>> numMap = new HashMap<Integer, List<Integer>>();
int n = adjacentPairs.length + 1;
int[] array = new int[n];
for (int[] adjacentPair : adjacentPairs) {
// 若当前元素不在哈希表中则将其放入
numMap.putIfAbsent(adjacentPair[0], new ArrayList<Integer>());
numMap.putIfAbsent(adjacentPair[1], new ArrayList<Integer>());
// 放入邻点信息
numMap.get(adjacentPair[0]).add(adjacentPair[1]);
numMap.get(adjacentPair[1]).add(adjacentPair[0]);
}
// 获取首元素
for (Map.Entry<Integer, List<Integer>> entry : numMap.entrySet()) {
int num = entry.getKey();
// 若邻点个数为1
if (entry.getValue().size() == 1) {
array[0] = num;
break;
}
}
// 获取第二个元素
array[1] = numMap.get(array[0]).get(0);
// 完善数组
for (int i = 2; i < n; i++) {
// 找出当前数组末尾的邻点集合
List<Integer> list = numMap.get(array[i-1]);
// 找出未放入的点
array[i] = array[i-2] == list.get(0) ? list.get(1) : list.get(0);
}
return array;
}