两指针--减少数组循环

Posted 2020-08-09 backinfile

题目(lintcode)：

1.二数之和

2.三数之和

3.最接近的三数之和

4.四数之和

 

　　 取三数之和为例：

（一）

　　普通算法，多重遍历数组，需要多重for嵌套，但严重超时。

（二）

　　剪枝，在外层循环中考虑有没有“不可能符合要求”的情况。

　　例如，在三数之和中，若两个数之和大于0，直接break。但题目没说不会出现负数，所以不对。

（三）

　　接（二），判断两个数之和大于0无效，是因为数组无序，所以先进行排序

　　 sort(nums.begin(),nums.end()); 

　　之后可以在保证大于0时，break掉。像这样：

for (int i = 0; i < nums.size(); i++) {
    if (nums[i] > 0) break;
    for (int j = i;j < nums.size(); j++) {
        if (nums[j] >= 0 && nums[j] +nums[i] > 0) break;
            for (int k = j; k < nums.size(); k++) {

（四）

　　设置两个指针，分别指向“数组首”(front)和“数组尾”(rear)，计算两指针处的和，再加上for循环的一个数，得到三数和sum：

　　若sum大于零，使和变小，rear-- 即可；若sum小于零，使和变大，front++ 即可。

vector<vector<int> > threeSum(vector<int> &nums) {
        set<vector<int> > re;
        vector<int> in;
        sort(nums.begin(), nums.end(), greater<int>());
        int front, rear, sum;
        for (int i = 0; i < nums.size(); i++) {
            front = 0;
            rear = nums.size() -1;
            while (front < rear) {
                if (front == i) front++;
                else if (rear == i) rear--;
                if (front == rear) break;
                sum = nums[front] + nums[rear] + nums[i];
                if (sum == 0) {
                    in.clear();
                    in.push_back(nums[front]);
                    in.push_back(nums[rear]);
                    in.push_back(nums[i]);
                    sort(in.begin(), in.end());
                    re.insert(in);
                    do {front++;}
                    while (front < rear && nums[front] == nums[front - 1]);
                }
                else if (sum > 0) front++;
                else rear--;
            }
        }
        return vector <vector<int>>(re.begin(),re.end());
    }