抓狂中
前言
两数之和-三数之和-四数之和三个题算是一系列的,主要掌握双指针法,配合一些剪枝操作,解题思路就有了,以后n个数之和也能轻松解决;
一、两数之和
题目
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
思路
循环嵌套遍历也可,但是时间复杂度是O(n2),采用HashMap,遍历一遍时把元素加进map里,然后找targer-当前元素值是否在map里即可;
注:由于可能存在相同值,所以要先检查所要找的值是否在map里,在把当前值加进map,如nums=[…3,3,…],target=6,如果在把第一个3先加进map,则判断map里有无3会返回两个相同的位置值。
代码
class Solution {
public int[] twoSum(int[] nums, int target) {
Map<Integer,Integer> m = new HashMap<Integer,Integer>();//map的键是元素,值是位置
int len=nums.length;
for(int i=0;i<len;i++)
{
if(m.containsKey(target-nums[i]))
return new int[]{m.get(target-nums[i]),i};
//如果有相同元素,且target的值等于这两元素之和,则必须在判断完存在之后才能将当前遍历的元素加入map中
m.put(nums[i],i);//遍历过程中把每个元素都添加进map中
}
return new int[]{};
}
}
二、三数之和
题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
思路
双指针法
代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> l = new ArrayList<>();
for(int begin=0;begin<nums.length-2;begin++ )
{
if(nums[begin]>0)
break;
if(begin>0&&nums[begin]==nums[begin-1])
continue;//如果当前项和前一项相等,说明重复了,跳过
int left=begin+1,right=nums.length-1,sum=0;
while(left<right)
{
sum=nums[left]+nums[right]+nums[begin];
if(sum==0)
{
l.add(new ArrayList<Integer>(Arrays.asList(nums[begin],nums[left],nums[right])));
while(left<right&&nums[left]==nums[++left]);
while(left<right&&nums[right]==nums[--right]);
}
else if(sum<0)
while(left<right&&nums[left]==nums[++left]);
else
while(left<right&&nums[right]==nums[--right]);
}
}
return l;
}
}
三、四数之和
题目
给你一个由 n 个整数组成的数组 nums ,和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] (若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
思路
同样是双指针法,不过多了很多剪枝
代码
class Solution {
//双指针法
public List<List<Integer>> fourSum(int[] nums, int target){
List<List<Integer>> res = new ArrayList<List<Integer>>();//返回的答案
if(nums==null||nums.length<4)//如果数组为空或者长度小于4,返回空数组
return res;
int len=nums.length;
Arrays.sort(nums);
for(int i=0;i<len-3;i++)
{
//剪枝
if(i>0&&nums[i]==nums[i-1])//如果和上一个元素的值相等,表示该值已经判断过了
continue;
if(nums[i]+nums[i+1]>target-(nums[i+2]+nums[i+3]))//四个元素和如果超出了int的范围,改成减法活用long强转,下同
break;//如果前四个元素相加已经比target大了,则后面肯定没有小于等于target的数了,结束循环
if(nums[i]+nums[len-1]<target-(nums[len-2]+nums[len-3]))
continue;//如果第一个和最后三个数相加比target小,则第一个数肯定不满足,直接从下一个数开始
for(int j=i+1;j<len-2;j++)
{
//剪枝方法同上
if(j>i+1&&nums[j]==nums[j-1])
continue;
if(nums[i]+nums[j]>target-(nums[j+1]+nums[j+2]))
break;
if(nums[i]+nums[j]<target-(nums[len-2]+nums[len-1]))
continue;
int left=j+1,right=len-1;
//双指针法
while(left<right)
{
int sum1 = nums[i]+nums[j];
int sum2 = target-(nums[left]+nums[right]);
if(sum1<sum2)
while(left<right&&nums[left]==nums[++left]);//跳过重复元素
else if(sum1>sum2)
while(left<right&&nums[right]==nums[--right]);//跳过重复元素
else
{
res.add(Arrays.asList(nums[i],nums[j],nums[left],nums[right]));
while(left<right&&nums[left]==nums[++left]);
//另一种麻烦的写法
// while(left<right&&nums[left+1]==nums[left])
// left++;
//left++;//不加这一步跳不出循环,永远卡在最后一个相等的left
// 通过在nums[++left]做自加,包正最后能指项下一个不同的元素
while(left<right&&nums[right]==nums[--right]);
//另一种麻烦的写法
// while(left<right&&nums[right-1]==nums[right])
// right--;
//right--;//不加这一步跳不出循环,永远卡在最后一个相等的right
//改进方法同上
}
}
}
}
return res;
}
}