知识点:双指针、排序、数组、枚举
题目
给你两个整数数组 nums1
和 nums2
。
从 nums1
中移除两个元素,并且所有其他元素都与变量 x
所表示的整数相加。如果 x
为负数,则表现为元素值的减少。
执行上述操作后,nums1
和 nums2
相等 。当两个数组中包含相同的整数,并且这些整数出现的频次相同时,两个数组 相等 。
返回能够实现数组相等的最小整数 x
。
示例 1:
输入:nums1 = [4,20,16,12,8], nums2 = [14,18,10]
输出:-2
解释:
移除 nums1 中下标为 [0,4] 的两个元素,并且每个元素与 -2 相加后,nums1 变为 [18,14,10] ,与 nums2 相等。
示例 2:
输入:nums1 = [3,5,5,3], nums2 = [7,7]
输出:2
解释:
移除 nums1 中下标为 [0,3] 的两个元素,并且每个元素与 2 相加后,nums1 变为 [7,7] ,与 nums2 相等。
提示:
3 <= nums1.length <= 200
nums2.length == nums1.length - 2
0 <= nums1[i], nums2[i] <= 1000
测试用例以这样的方式生成:存在一个整数 x,nums1 中的每个元素都与 x 相加后,再移除两个元素,nums1 可以与 nums2 相等。
题解
class Solution {
public:
int minimumAddedInteger(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size(), n = nums2.size();
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
for (int i: {2, 1, 0}) {
int left = i + 1, right = 1;
while (left < m && right < n) {
if (nums1[left] - nums2[right] == nums1[i] - nums2[0]) {
++right;
}
++left;
}
if (right == n) {
return nums2[0] - nums1[i];
}
}
return 0;
}
};
总结:
i
d
0
id_0
id0一定是nums1下标为0、1、2的某一个元素。可以进行遍历。遍历顺序从2开始是因为nums1[2]>=nums1[1]>=nums1[0]。nums1的值越大,x的值越小,保证返回的是最小的整数x。同时采用双指针对nums1(left指针)和nums2(right指针)遍历,判断差值与nums1[i] - nums2[0]的大小。
不一样则说明需要去除nums1当前指针的元素,因此left++而right不动。一样就同时向前迭代。最终判断的标准是right指针是否遍历整个nums2。