思路:
这个题目挺妙的,注意到就是相邻包括了左相邻和右相邻,在数组开头和结尾的数字只有1次相邻。因此我们可以把所有数字对应的相邻元素作为哈希表的值存储,然后遍历哈希表,如果值维度是1,说明这个是开头或者结尾得元素。中间得元素相邻是2个得。
然后从后续依次,选择相邻中没走过的元素连接即可。
class Solution {
public:
vector<int> restoreArray(vector<vector<int>>& adjacentPairs) {
unordered_map<int, vector<int>> mp;
for (auto& adjacentPair : adjacentPairs) {
mp[adjacentPair[0]].push_back(adjacentPair[1]); //记录0相邻的有这个
mp[adjacentPair[1]].push_back(adjacentPair[0]); //记录和1相邻的有这个
}
int n = adjacentPairs.size() + 1;
vector<int> ret(n);
for (auto& [e, adj] : mp) {
if (adj.size() == 1) {
ret[0] = e;
break;
}
}
ret[1] = mp[ret[0]][0];
for (int i = 2; i < n; i++) {
auto& adj = mp[ret[i - 1]];
ret[i] = ret[i - 2] == adj[0] ? adj[1] : adj[0];
}
return ret;
}
};