[M哈希表] lc1743. 从相邻元素对还原数组(哈希表+递推+周赛226_2)

Posted 2021-08-27 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：1743. 从相邻元素对还原数组

2. 题目解析

数组元素各不相同，显然使用哈希表。

数组首元素、尾元素与其相邻的元素仅有一个，其余中间的相邻数都是两个。以此可以确定数组的头、尾元素是多少，与其相邻的唯一一个元素也就确定了，顺序递推即可。

递推时，从下标 2 开始，先得到下标 i-1 的两个相邻元素，i-2 已经填过了，把没填过的其中一个元素填到 i 位置即可。

---

时间复杂度：$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;
    }
};