NO1743.從相鄰元素對還原數組

存在一個由 n 個不一樣元素組成的整數數組 nums ，但你已經記不清具體內容。好在你還記得 nums 中的每一對相鄰元素。

給你一個二維整數數組 adjacentPairs ，大小爲 n - 1 ，其中每一個 adjacentPairs[i] = [ui, vi] 表示元素 ui 和 vi 在 nums 中相鄰。

題目數據保證全部由元素 nums[i] 和 nums[i+1] 組成的相鄰元素對都存在於 adjacentPairs 中，存在形式多是 [nums[i], nums[i+1]] ，也多是 [nums[i+1], nums[i]] 。這些相鄰元素對能夠 按任意順序 出現。

返回 原始數組 nums 。若是存在多種解答，返回 其中任意一個 便可。

示例1：

輸出：[1,2,3,4]
解釋：數組的全部相鄰元素對都在 adjacentPairs 中。
特別要注意的是，adjacentPairs[i] 只表示兩個元素相鄰，並不保證其 左-右 順序。
複製代碼

示例2：

輸出：[-2,4,1,-3]
解釋：數組中可能存在負數。
另外一種解答是 [-3,1,4,-2] ，也會被視做正確答案。
複製代碼

示例3：

輸入： adjacentPairs = [[100000,-100000]]
輸出： [100000,-100000]
複製代碼

提示：

• nums.length == n
• adjacentPairs.length == n - 1
• adjacentPairs[i].length == 2
• 2 <= n <= 105
• -105 <= nums[i], ui, vi <= 105
• 題目數據保證存在一些以 adjacentPairs 做爲元素對的數組 nums

---

思考過程： 首先題目保證了result[]必定存在，其次題目給出了數組中每個元素的相鄰元素信息。那麼咱們在構建result[]的時候，假如result[0] ~ result[i - 1]已經正確構建，咱們想要得到result[i]，就須要知道result[i - 1]的鄰居信息與res[i -2]的自身信息。具體操做上咱們能夠用HashMap<i,j>輔助完成，i表示數組元素i，j表示元素i的兩個鄰居的和，則有result[i] = map.get(result[i - 1]) - result[i - 2];

時間複雜度：O(N)
空間複雜度：O(N)

代碼：

public int[] restoreArray(int[][] adjacentPairs) {
        //HashMap計數
        HashMap<Integer, Integer> map =  new HashMap<>();
        for (int[] arr : adjacentPairs) {
            map.put(arr[0], map.getOrDefault(arr[0], 0) + 1);
            map.put(arr[1], map.getOrDefault(arr[1], 0) + 1);
        }

        //構建鄰接關係表, map<i, j>表示元素i的左右鄰居的和爲j
        HashMap<Integer, Integer> adj = new HashMap<>();
        for (int[] arr : adjacentPairs) {
            adj.put(arr[0], adj.getOrDefault(arr[0], 0) + arr[1]);
            adj.put(arr[1], adj.getOrDefault(arr[1], 0) + arr[0]);
        }

        //尋找開始元素
        int start = -1;
        for (int i : map.keySet()) {
            if (map.get(i) == 1) {
                start = i;
                break;
            }
        }

        //res[i]的值爲res[i - 1]的值的相鄰元素的和減去res[i - 2]
        int[] res = new int[adjacentPairs.length + 1];
        res[0] = start;
        for (int i = 1; i < res.length; i++) {
            if (i == 1) {
                res[i] = adj.get(res[i - 1]);
            } else {
                res[i] = adj.get(res[i - 1]) - res[i - 2];
            }
        }

        return res;
    }
}
複製代碼