leetcode周赛273记录-还原数组

最新推荐文章于 2024-09-14 18:52:56 发布

nth2000

最新推荐文章于 2024-09-14 18:52:56 发布

阅读量446

点赞数

本文链接：https://blog.csdn.net/nth2000/article/details/122344295

版权

leetcode 算法排序算法

题干

在这里插入图片描述

思路-（双指针-通常为寻找问题的子结构）

子结构

在这里插入图片描述

具体思路

现在的数组可以分为两部分，其中一部分每个元素与另一部分每个元素都小 $2 k$ （分别对应），即原始数组中的元素为现在第一部分的元素加上第二部分对应元素之和除以2
而现在的数组的最小元素一定属于第一部分，将其加上 $k$ 也一定是原始数组的最小元素
首先对数组排序
令left指针指向第一部分的当前元素，right指针指向使得nums[left]+2k的元素，同时使用mask记录哪些元素被选取过（也就等价于从数组中删除选中的元素）。初始阶段left指向排序后的nums[0]，right分别遍历nums中所有元素。
不变量的维持:将left向右移动到第一个没被选择过的元素，这个元素一定处于第一部分中。

代码

    public int[] recoverArray(int[] nums) {
        Arrays.sort(nums);
        int left = 0;
        int[] ans = new int[nums.length / 2];
        for(int right = 1;right < nums.length;right++)
        {
            left = 0;
            boolean[] mask = new boolean[nums.length];
            mask[0] = true;
            int right_1 = right;
            mask[right_1]= true;
            int k;
            if((nums[right_1] - nums[left])%2 == 0 && nums[right_1] - nums[left] > 0) {
                k = (nums[right_1] - nums[left]) / 2;
                int len = 0;    //当前数组长度
                ans[len++] = (nums[right_1] + nums[left]) / 2;
                if(len == nums.length / 2)
                    return ans;
                while(true)
                {
                    while (left < nums.length && mask[left])
                        left++;
                    if(left < nums.length)
                        mask[left] = true;
                    else break;
                    while(right_1 < nums.length && (mask[right_1] || nums[left] + 2 * k != nums[right_1]))
                        right_1++;
                    if(right_1 < nums.length)
                        mask[right_1] = true;
                    else break;
                    ans[len++] = (nums[right_1] + nums[left])/2;
                    if(len == nums.length / 2)
                        return ans;
                }
            }
        }
        return ans;
    }