1. 题目来源
2. 题目解析
数组元素各不相同,显然使用哈希表。
数组首元素、尾元素与其相邻的元素仅有一个,其余中间的相邻数都是两个。以此可以确定数组的头、尾元素是多少,与其相邻的唯一一个元素也就确定了,顺序递推即可。
递推时,从下标 2 开始,先得到下标 i-1
的两个相邻元素,i-2
已经填过了,把没填过的其中一个元素填到 i
位置即可。
时间复杂度: O ( n ) O(n) O(n)
空间复杂度: O ( n ) O(n) O(n)
二维数组写法:
class Solution {
public:
vector<int> restoreArray(vector<vector<int>>& adjacentPairs) {
int n = adjacentPairs.size();
unordered_map<int, vector<int>> S;
for (auto a : adjacentPairs) {
S[a[0]].push_back(a[1]);
S[a[1]].push_back(a[0]);
}
vector<int> res(n + 1);
for (auto [k, v] : S) {
if (v.size() == 1) { // 数组首、尾仅有一个元素与其相邻,以此可以找到头、尾的值
res[0] = k;
break;
}
}
res[1] = S[res[0]][0];
for (int i = 2; i <= n; i ++ ) {
auto &a = S[res[i - 1]];
if (res[i - 2] == a[0]) res[i] = a[1];
else if (res[i - 2] == a[1]) res[i] = a[0];
}
return res;
}
};