LeetCode 1743 从相邻元素对还原数组[Map] HERODING的LeetCode之路

Posted 2021-08-25 HERODING23

解题思路：
从逻辑上来说还是很清晰的一道题，但是思路不太容易想出来，由于顺序没有严格规定，所以我们只需要找好第一个就行，其他位置根据数组提供的信息依次插入即可，步骤如下：

1. 统计所有的数入map中，一个数对应它的相邻数列表；
2. 找到相邻数列表为1的，设置为第一个数；
3. 依次把相邻数放入到数组中。

代码如下：

class Solution {
public:
    vector<int> restoreArray(vector<vector<int>>& adjacentPairs) {
        unordered_map<int, vector<int>> mp;
        int len = adjacentPairs.size();
        for(int i = 0; i < len; i ++) {
            mp[adjacentPairs[i][0]].push_back(adjacentPairs[i][1]);
            mp[adjacentPairs[i][1]].push_back(adjacentPairs[i][0]);
        }
        vector<int> res(len + 1);
        for(auto& [n, list] : mp) {
            if(list.size() == 1) {
                res[0] = n;
                break;
            }
        }
        res[1] = mp[res[0]][0];
        for(int i = 2; i <= len; i ++) {
            auto& list = mp[res[i - 1]];
            res[i] = res[i - 2] == list[0] ? list[1] : list[0];
        }
        return res;
    }
};