两指针--减少数组循环

题目(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());
    }
原文地址:https://www.cnblogs.com/backinfile/p/5886263.html