首页 > 其他 > 详细

[itint5]三数和为0

时间:2014-01-20 08:43:24      阅读:401      评论:0      收藏:0      [点我收藏+]

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

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#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;
}

  

[itint5]三数和为0

原文:http://www.cnblogs.com/lautsie/p/3525963.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!