NO1743.从相邻元素对还原数组

497 阅读2分钟

存在一个由 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;
    }
}