手写代码:数组的2-sum,3-sum,问题(leetcode原题)
参考回答:
最常见的是2SUM问题(1. 2SUM),就是数组中找出两个数相加和为给定的数。
这道题有两种思路:
1. 一种思路从首尾搜索两个数的和,并逐步逼近。
2. 另外一种思路是固定一个数A,看SUM-A是否在这个数组之中。
此方法是先将数组从小到大排序
设置两个指针,一个指向数组开头,一个指向数组结尾,从两边往中间走。直到扫到满足题意的为止或者两个指针相遇为止。
此时这种搜索方法就是类似于杨氏矩阵的搜索方法,就是从杨氏矩阵的左下角开始搜索,直到找到为止。
如果此时题目条件变为如果没有找出最接近的2SUM和问题,解决方法跟上面是一样的
用这种方法2SUM问题的时间复杂度是O(nlogn)O(nlogn)的,主要在于排序的时间。
对数组中的每个数建立一个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; }