194

问答题 194 /413

手写代码:数组的2-sum,3-sum,问题(leetcode原题)

参考答案

参考回答:

2SUM问题

最常见的是2SUM问题(1. 2SUM),就是数组中找出两个数相加和为给定的数。
这道题有两种思路:
1. 一种思路从首尾搜索两个数的和,并逐步逼近。
2. 另外一种思路是固定一个数A,看SUM-A是否在这个数组之中。

对于第一种思路如下:

此方法是先将数组从小到大排序
设置两个指针,一个指向数组开头,一个指向数组结尾,从两边往中间走。直到扫到满足题意的为止或者两个指针相遇为止。
此时这种搜索方法就是类似于杨氏矩阵的搜索方法,就是从杨氏矩阵的左下角开始搜索,直到找到为止。
如果此时题目条件变为如果没有找出最接近的2SUM和问题,解决方法跟上面是一样的
用这种方法2SUM问题的时间复杂度是O(nlogn)O(nlog⁡n)的,主要在于排序的时间。

第二种思路方法如下:

对数组中的每个数建立一个map/hash_map 然后再扫一遍这个数组,判断target-nums[i]是否存在,如果存在则说明有,不存在继续找。当然这样做的话,需要处理一个细节:判重的问题。
代码如下【注意因为题目中说一定有解所以才下面这样写,如果不一定有解,则还要再加一些判断】

vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,vector<int>> mark;
for(int i=0;i<nums.size();i++)
mark[nums[i]].push_back(i);
for(int i = 0;i<nums.size();i++){
if(target-nums[i] == nums[i]){
if(mark[nums[i]].size() > 1){
vector<int> tmp{i,mark[nums[i]][1]};
return tmp;
}
}else{
if(mark.find(target-nums[i]) != mark.end()){
vector<int> tmp{i,mark[target-nums[i]][0]};
return tmp;
}
}
}
}


3-SUM 问题

Question
Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.The solution set must not contain duplicate triplets.

根据给定的容量为 n 的整数数组,找到所有满足 a + b + c = 0 的三个元素a、b 、c组合,需去重;

解法

修改问题为:满足 a + b + c = target,target是给定数,原题即 target = 0;

根据题目可知,与 2-SUM 问题类似,在整数数组中不放回的取出三个元素,其和等于给定数(0),不同的是,满足条件的解有多个而且需要去重;

首先想到的解法是,遍历数组,然后调用 2-SUM 问题中的方法寻找两个元素相加等于当前元素的补足,最后执行去重操作;这样的话,查询的时间复杂度是$O(n^2)$,空间复杂度是$O(n^2)$,去重的时间复杂度是$O(n^2)$,空间复杂度是$O(1)$,这绝对不能算一个好方案;

思考其他思路,既然要去重,可以先对数组执行一次排序,这样的话在遍历的时候可以跳过相同元素;在确定一个当前元素后,找另外两个元素相加作为当前元素的补足,此时的解可能是多个的,采用首尾标记的方式可以一次遍历完成查找;

public List<List<Integer>> threeSum(int[] nums, int target){
int length = nums.length;
List<List<Integer>> result = new ArrayList<>();
Arrays.sort(nums);
for(int i = 0; i < length - 2; i++) {
if(nums[i] + nums[i+1] + nums[i+2] > target)break; // too large
if(nums[i] + nums[length-1] + nums[length-2] < target)continue; // too small
if(i > 0 && nums[i] == nums[i - 1]) continue;
int left = i + 1;
int right = length - 1;
while(left < right){
int diff = target - nums[i] - nums[left] - nums[right];
if (diff == 0){
result.add(new ArrayList<Integer>(Arrays.asList(nums[i], nums[left], nums[right])));
while(left < right && nums[left] == nums[left + 1]) left++;
while(left < right && nums[right] == nums[right - 1]) right--;
left++;
right--;
}else if (diff > 0){
left++;
}else {
right--;
}
}
}
return result;
}