[itint5]三数和为0

http://www.itint5.com/oj/#20

其实是3sum的变种,有重复数字,但是一开始还是写错了。其实是选定一个后,在右边剩余数组里找2sum,找到一组后继续找。

#include <algorithm>
using namespace std;
typedef tuple<int, int, int> ABC; //存放a,b,c三元组


//返回所有满足条件的(a,b,c)三元组
vector<ABC> threeSumZero(vector<int> &arr) {
    vector<ABC> ans;
    sort(arr.begin(), arr.end());
    int k = 0;
    while (k < arr.size()) {
        int i = k+1;
        int j = arr.size() - 1;
        while (i < j) {
            if (arr[i] + arr[j] == -arr[k]) {
                ans.push_back(make_tuple(arr[k], arr[i], arr[j]));
                while (i+1 < arr.size() && arr[i] == arr[i+1]) i++;
                while (j-1 >= 0 && arr[j-1] == arr[j]) j++;
                i++;
                j--;
            } else if (arr[i] + arr[j] < -arr[k]) {
                i++;
            } else {
                j--;
            }
        }
        while (k+1 < arr.size() && arr[k] == arr[k+1]) k++;
        k++;
    }
    return ans;
}

  

posted @ 2014-01-19 15:35  阿牧遥  阅读(184)  评论(0编辑  收藏  举报