Two Sum && 3 Sum

来源:互联网 发布:淘宝免单群是什么意思 编辑:程序博客网 时间:2024/03/29 10:11

Two Sum 

Input: numbers={2, 7, 11, 15}, target=9
Output: index1=1, index2=2

pro:给一个数组,一个target,寻找数组中和等于target的两个数

sol:经典的2sum问题。

1.首先将数组排序。这点要想到,看给定的数组是否已经排好序了的。

2.设置两个指针i,j,分别初始化为指向数组的开头和结尾。

3.指针移动的规则如下所示,直到i与j相等了。之所以能这么做是因为比如j--的情况,j已经跟最小的i相加了还大于target,肯定不会有解了,因此需要j--。

if(num[i]+num[j]>target) j--;

ele if(num[i]+num[j]<target) i++;

else return.....

3 Sum

pro:数组中三个数的和等于target。

sol:外面多一层循环,再用2Sum O(n^2)的复杂度。

当求具体是哪些元组时,需要保证不重复,那每个指针在某个数上如果曾经用过这个数就跳过,即每个指针都只用相同的数一次~

当等于target,两边都往中间移动,因为不可能只移动一边,如j,j指向的不一样了,i仍不动的情况,和还一样,这是不可能的。

当不等于target时,只能移动一边。


code:

class Solution {public:    vector<vector<int> > threeSum(vector<int> &num) {        int i,j,k,len;        sort(num.begin(),num.end());        len = num.size();        vector< vector<int> > res;        vector<int> temp;        res.clear();        temp.clear();        if(len<3) return res;        for(i=0;i<len-2;i++)        {            temp.clear();            if(i!=0&&num[i]==num[i-1]) continue;            temp.push_back(num[i]);            j=i+1,k=len-1;            while(j<k)            {                if(num[i]+num[j]+num[k]==0)                {                    temp.push_back(num[j]);                    temp.push_back(num[k]);                    res.push_back(temp);                    temp.pop_back();                    temp.pop_back();                    j++;                    while(j<k&&num[j]==num[j-1]) j++;                    k--;                    while(j<k&&num[k]==num[k+1]) k--;                }else if(num[i]+num[j]+num[k]<0)                {                    j++;                    while(j<k&&num[j]==num[j-1]) j++;                }else                {                    k--;                    while(j<k&&num[k]==num[k+1]) k--;                }            }        }        return res;    }};




0 0
原创粉丝点击