LeetCode-N数之和类问题总结(双指针法)

两数之和

给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
示例:给定 nums = [2, 7, 11, 15], target = 9。因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]。

分析：双指针法。

class Solution {
public:
	vector<int> twoSum(vector<int>& nums, int target) {
		vector<int>dump (nums);
		vector<int> temp;
		if (nums.size() < 2) {
			return temp;
		}
		sort(nums.begin(), nums.end());
		int i = 0;
		int j = nums.size() - 1;
		while(j>i)
		{
			if (nums[i] + nums[j] == target)
			{
				for (int index=0; index < dump.size(); index++)
				{
					if (dump[index] == nums[i]) {
						temp.push_back(index);
						break;
					}

				}
				for (int index = 0; index < dump.size(); index++)
				{
					if ((dump[index] == nums[j])&&(index!=temp[0]))
					{
						temp.push_back(index);
						break;
					}
				}			
				break;
			}
			else if (nums[i] + nums[j] > target) {
				j--;
			}
			else {
				i++;
			}
		}
		return temp;
	}	

};

---

三数之和

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例：给定数组 nums = [-1, 0, 1, 2, -1, -4]，满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

分析：双指针法。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res_final;
        sort(nums.begin(), nums.end());
        if (nums.size() < 3 || nums[0] > 0 || nums[nums.size()-1] < 0) return res_final;
        for (size_t i = 0; i < nums.size(); ++i) {
            if (nums[i] > 0) break;
            if (i>0 && nums[i] == nums[i-1]) continue;
            int target = 0-nums[i];
            int k = i+1;
            int m = nums.size()-1;
            while(k<m) {
                if (nums[k] > target) break;
                if ((k>i+1 && nums[k] == nums[k-1]) || nums[k]+nums[m] < target) {
                    ++k;
                } else if ((m<nums.size()-1 && nums[m] == nums[m+1]) || nums[k]+nums[m] > target) {
                    --m;
                } else {
                    res_final.push_back(vector<int>{nums[i],nums[k],nums[m]});
                    ++k;
                }
            }        
        }
        
        return res_final;
    }
};

---

最接近的三数之和

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数，使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如，给定数组 nums = [-1，2，1，-4], 和 target = 1。与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

分析：双指针法。

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        if(nums.size()<3) return INT_MAX;//输入检测
        sort(nums.begin(),nums.end());
        int res = nums[0]+nums[1]+nums[2];//初始化
        for(int i =0;i<nums.size();i++){
            int front = i+1,back = nums.size()-1;
            while(front < back){
                if(abs(nums[i]+nums[front]+nums[back]-target)<abs(res-target)) res = nums[i]+nums[front]+nums[back];
                if(nums[i]+nums[front]+nums[back]>target) back--;
                else if (nums[i]+nums[front]+nums[back]<target) front++;
                else if (nums[i]+nums[front]+nums[back]==target) return target;
            }
        }
        return res;
    }
};

---

四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。
注意：答案中不可以包含重复的四元组。

示例：给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。满足要求的四元组集合为：
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]

分析：三数之和的解法外加一层循环。

class Solution {
public:
	    vector<vector<int>> fourSum(vector<int>& nums, int target) {
		vector<vector<int>> res_final;
		int temptarget = 0;
		sort(nums.begin(), nums.end());
		if (nums.size() < 4 )
			return res_final;
		for (size_t i = 0; i < nums.size()-3; ++i) {
			if (i > 0 && nums[i] == nums[i - 1]) 
				continue;
			for (size_t j = i + 1; j < nums.size() - 2; j++) {
				if (j-i > 1 && nums[j] == nums[j- 1]) 
					continue;
				temptarget = target- nums[i]-nums[j];
				int k = j + 1;
				int m = nums.size() - 1;
				while (k < m) {
					if ((k > j + 1 && nums[k] == nums[k - 1]) || nums[k] + nums[m] < temptarget) {
						++k;
					}
					else if ((m < nums.size() - 1 && nums[m] == nums[m + 1]) || nums[k] + nums[m] > temptarget) {
						--m;
					}
					else {
						res_final.push_back(vector<int>{nums[i], nums[j], nums[k], nums[m]});
						++k;
					}
				}
			}
		}

		return res_final;
	}
};

---

n数之和

题目描述：检查一个数组里是否存在n个数的和等于某个值target

算法分析：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int n, int target) {
        vector<vector<int>> res;
        vector<int> cur;
        sort(nums.begin(),nums.end());
        findNsum(nums,target,n,cur,res);
        return res;
    }
    void findNsum(vector<int>& nums,int target,int N,vector<int> cur,vector<vector<int>>& res)
    {
        if(N<2 || nums.size()<N || nums[0]*N>target)
            return ;
        if(N==2)
        {
            int i=0,j=nums.size()-1;
            while(i<j)
            {
                if(nums[i]+nums[j]==target)
                {
                    vector<int> temp;               //注意新建一个数组，不改变cur
                    for(int p=0;p<cur.size();p++)
                        temp.push_back(cur[p]);
                    temp.push_back(nums[i]);
                    temp.push_back(nums[j]);        //python中list合并’+’的实现
                    res.push_back(temp);
                    while(i<j && nums[i]==nums[i+1]) i++;
                    while(i<j && nums[j]==nums[j-1]) j--;
                    i++;j--;
                }
                else if(nums[i]+nums[j]<target)
                    i++;
                else
                    j--;
            }
        }
        else
        {
            for(int k=0;k<nums.size()-N+1;k++)
            {
                if(k==0 || nums[k]!=nums[k-1])
                {
                    vector<int> tempNums,tempCur;
                    for(int t=k+1;t<nums.size();t++)
                        tempNums.push_back(nums[t]);
                    for(int z=0;z<cur.size();z++)         //注意新建一个数组，不改变cur
                        tempCur.push_back(cur[z]);
                    tempCur.push_back(nums[k]);
                    findNsum(tempNums,target-nums[k],N-1,tempCur,res);
                }
            }
        }
    }
};