2122. 还原原数组(枚举&双指针)
排序后最小值一定属于 l o w e r lower lower。
考虑枚举 a 0 a_0 a0 配对的 a i a_i ai。
这样可以确定 k k k。
然后我们用一个标记数组 和 双指针 进行检验。
每次找到当前剩余的最小值 和对应的 a j = a i + 2 k a_j = a_i +2k aj=ai+2k
时间复杂度: O ( n 2 ) O(n^2) O(n2)
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 {};
}
};