2122. 还原原数组(枚举&双指针)

Posted 2022-02-04 Harris-H

2122. 还原原数组(枚举&双指针)

排序后最小值一定属于$lower$。

考虑枚举$a_0$ 配对的$a_i$。

这样可以确定$k$。

然后我们用一个标记数组 和 双指针 进行检验。

每次找到当前剩余的最小值 和对应的$a_j=a_i+2k$

时间复杂度：$O(n^2)$

class Solution 
public:
    vector<int> recoverArray(vector<int>& nums) 
        int n = nums.size();
        sort(nums.begin(), nums.end());
        for (int i = 1; i < n; ++i) 
            if (nums[i] == nums[0] || (nums[i] - nums[0]) % 2 != 0) 
                continue;
            
            
            vector<int> used(n);
            used[0] = used[i] = true;
            int k = (nums[i] - nums[0]) / 2;
            vector<int> ans;
            ans.push_back(nums[0] + k);
            
            int left = 0, right = i;
            for (int j = 2; j + j <= n; ++j) 
                while (used[left]) 
                    ++left;
                
                while (right < n && (used[right] || nums[right] - nums[left] != k * 2)) 
                    ++right;
                
                if (right == n) 
                    break;
                
                ans.push_back(nums[left] + k);
                used[left] = used[right] = true;
            
            
            if (ans.size() == n / 2) 
                return ans;
            
        
        
        // 题目保证一定有解，不会到这一步
        return ;
    
;