leetcode 15. 3Sum

这是一个寻找Mr/Mrs.Right的故事。

先排序便于搜索。

先 fix 一个数， 寻找另外两个匹配的数。

用两个指针 丈夫和妻子。

丈夫尺寸为 fix 的下一个数。妻子的尺寸为最后一个数， 他们的目的是合起来等于第 fix 个数的相反数。

丈夫可以变大，妻子可以缩小。（通过某种手术，咳咳……）

他们的初始和偏大，因为丈夫指向开头，所以无法变小，只能妻子变小为能和丈夫match的尺寸，直到找到这个尺寸，或者变得太小跳过这个尺寸，（因为前边的尺寸都检测过，并不match，所以妻子不用变大，当时思路卡在这里），妻子变得太小过后只能让丈夫变大，同理，知道两人相遇，结束。

#include <cstdio>
#include <vector>
#include <stdlib.h>
#include <time.h>
#include <algorithm>
using namespace std;

class Solution{    //主代码
public :
    vector< vector<int> > threeSum(vector<int>& arr){
        vector< vector<int> > res;
        vector<int> t;
        sort(arr.begin(), arr.end());
        if(arr.empty() || arr.back() < 0 || arr.front() > 0) return res;
        for(int fix = 0;fix < arr.size() && arr[fix] <= 0; fix++){
            if(fix > 0 && arr[fix] == arr[fix-1]) continue;
            int left = fix + 1, right = arr.size() - 1;
            while(left < right){
                if(arr[left] + arr[right] == -arr[fix]){
                    t.clear();
                    t.push_back(arr[fix]);
                    t.push_back(arr[left]);
                    t.push_back(arr[right]);
                    res.push_back(t);
                    while(left < right && arr[left] == arr[left + 1]) left++;
                    while(left < right && arr[right] == arr[right - 1]) right--;
                    left++;right--;
                }else if(arr[left] + arr[right] > -arr[fix])
                    right--;
                else
                    left++;
            }
        }
        return res;
    }

};

//生成随机数组
void Vct(vector<int>& arr){
    srand((int)time(0));
    for(int i = rand()%20; i > 0; i--)
        arr.push_back(rand()%40 - 20);
    return;

}

int main()
{
    Solution q;
    vector<int> arr;
    Vct(arr);
    for(int i = 0; i < arr.size(); i++){
        printf("%d ",arr[i]);
        if(i%5 == 0) printf("\n");
    }//生成随机数组并打印

    vector< vector<int> > a = q.threeSum(arr);//solve并打印
    for (int i = 0; i < a.size(); i++){
        for (int j = 0; j < a[i].size(); j++)
            printf("%d ",a[i][j]);
        printf("\n");
    }
    return 0;
}