题目描述:
给定一个长度为 n 的整数数组和一个目标值 target,寻找能够使条件 nums[i] + nums[j] + nums[k] < target 成立的三元组 i, j, k 个数(0 <= i < j < k < n)。
示例:
输入: nums = [-2,0,1,3], target = 2
输出: 2
解释: 因为一共有两个三元组满足累加和小于 2:
[-2,0,1]
[-2,0,3]
**进阶:**是否能在 O(n2) 的时间复杂度内解决?
方法1:使用三层循环
主要思路:
(1)按照题意,统计的数量,应该和元素的顺序没有关系,只和是否存在有关系,故可以先对数组进行排序;
(2)对于排序的数组,直观的想,就是三层循环统计满足要求的元素;
class Solution {
public:
int threeSumSmaller(vector<int>& nums, int target) {
//处理特殊的情形
if(nums.size()<3)
return 0;
//对原数组进行排序
sort(nums.begin(),nums.end());
int count=0;
//三层循环
//第一层是固定第一元素
for(int head=0;head<nums.size()-2;++head){
if(nums[head]+nums[head+1]+nums[head+2]>=target)//提前终止条件
break;
//第二次是固定第二个元素
for(int left=head+1;left<nums.size()-1;++left){
if(nums[head]+nums[left]+nums[left+1]>=target)//提前终止条件
break;
int tmp=target-nums[left]-nums[head];//第三层需要比较的值
//第三层是第三个需要判断的值
for(int right=left+1;right<nums.size();++right){
if(nums[right]>=tmp)//若该值不满足要求,则后面的都不满足要求,提前终止
break;
++count;//满足要求的组成,统计加1
}
}
}
return count;
}
};
方法2:双指针
主要思路:
(1)同样需要先对数组进行排序;
(2)然后固定第一数,对后面的两个数之和使用正常的双指针进行判断:
(a)当满足条件nums[left]+nums[right]<tmp时,说明right和left之间的数字都可以和left组合满足要求,则满足要求的组合增加为 count+=(right-left);然后将left自增1;
(b)若不满足要求,则将right自减 1;
class Solution {
public:
int threeSumSmaller(vector<int>& nums, int target) {
//处理特殊的情形
if(nums.size()<3)
return 0;
sort(nums.begin(),nums.end());//将数组进行排序
//辅助变量
int left=0;
int right=0;
int count=0;
for(int head=0;head<nums.size()-2;++head){//需要固定的第一个数字
//后面的双指针的范围
left=head+1;
right=nums.size()-1;
int tmp=target-nums[head];//新的目标值
while(left<right){//终止条件
if(nums[left]+nums[right]<tmp){//若满足要求
count+=(right-left);//需要增加的满足要求的组合的数量
++left;//更新左指针
}
else{
--right;//更新右指针
}
}
}
return count;//返回结果
}
};