Leetcode水题——3Sum

题目

题解

这题还是比较简单的，首先进行排序（对于没有按下标输出要求的题目，向考虑排序），排好序后，找第一个数，设第一个数下标为i，则第二个数从i+1开始找，因为i之前的数字的所有组合都已经找出来了，计算第一个数与第二数的和，取反，根据二分法查找，查找范围为第二个数的下标到数组结尾，另外当第一个数为正数时就可以退出了，因为正数后面都为正数，不可能为0，另外题目要求不能重复，一些边界条件还是要注意

代码

class Solution {
public:
    
    int binary(vector<int>&a,int first,int end,int number) {
	int mid;
	while (first <= end) {
		mid = (first + end) / 2;
		if (a[mid] == number)
			return mid;
		if (a[mid] < number)
			first = mid + 1;
		else
			end = mid-1;
	}
	return -1;
    }
    vector<vector<int>> threeSum(vector<int>& a) {
        	sort(a.begin(), a.end());
            vector<vector<int>> result0;
            int i,j,n=a.size();
            for (i = 0;i < n;i++) {
                if (a[i] > 0) break;
                if (i == 0 || a[i] != a[i - 1]) {
                    for (j = i + 1;j < n;j++) {
                        int temp = 0 - a[i] - a[j];
                        int result = binary(a, j + 1, n - 1, temp);
                        if ((j==i+1||a[j]!=a[j-1])&&result != -1) {
                            vector<int> temp;
                            temp.push_back(a[i]);
                            temp.push_back(a[j]);
                            temp.push_back(a[result]);
                            result0.push_back(temp);
                        }
                    }
                }    
         }
        return result0;
    }
};