15. 3Sum

这道题目，我的解法又是过于复杂了，用了map，而且把情形进行分类，结果搞得非常复杂。我的代码如下：

    vector<int> getVec(int a, int b, int c) {
        vector<int> result;
        int arr[] = {a, b, c}, tmp;
        for (int i=0; i<2; i++) {
            for (int j=i+1; j<3; j++) {
                if (arr[i] >  arr[j]) {
                    tmp = arr[i];
                    arr[i] = arr[j];
                    arr[j] = tmp;
                }
            }
        }
        result.push_back(arr[0]);
        result.push_back(arr[1]);
        result.push_back(arr[2]);

        return result;
    }
    vector<vector<int>> threeSum(vector<int>& nums) {
        map<int, int> num_cnt;
        vector<vector<int>> result;
        vector<int> item;
        int i, j, sum;
        
        // 把数字放进map
        for (i=0; i<nums.size(); i++) {
            if (num_cnt.find(nums.at(i)) != num_cnt.end()) {
                num_cnt[nums.at(i)] = num_cnt[nums.at(i)] + 1;
            } else {
                num_cnt[nums.at(i)] = 1;
            }
        }
        nums.clear();
        for (map<int, int>::iterator it = num_cnt.begin(); it != num_cnt.end(); it++) {
            nums.push_back(it->first);
        }
        
        // 三个数字都不同的情况
        for (i=0; i<(int)nums.size()-1; i++) {
            for (j=i+1; j<nums.size(); j++) {
                sum = nums.at(i) + nums.at(j);
                if (-sum > nums.at(j) &&
                    num_cnt.find(-sum) != num_cnt.end()) {
                    item = getVec(-sum, nums.at(i), nums.at(j));
                    result.push_back(item);
                }
                
            }
            
        }
        
        // 两个数字相同，一个数字不同 或者三个数字均为0
        for (map<int, int>::iterator it = num_cnt.begin(); it != num_cnt.end(); it++) {
            if (it->first == 0 && it->second >= 3) {
                item = getVec(0, 0, 0);
                result.push_back(item);
            }
            else if (it->first != 0 && it->second >= 2) {
                sum = 2*it->first;
                if (num_cnt.find(-sum) != num_cnt.end()) {
                    item = getVec(-sum, it->first, it->first);
                    result.push_back(item);
                }
            }
        }
        
        return result;
        
    }

其实，更好的思想是对vector进行排序，然后固定第一个数，假设为A[i]，接下来，我们在A[i]之后的元素中，找到两个数，使其和为-A[i]。 因为数组已经有序，所以可以通过双指针向中间进行夹逼操作。这样我们就把找出三个其和为0的数的问题，转化为在有序数组中，找到两个数其和为某个值的问题。代码如下：

    vector<vector<int>> threeSum(vector<int>& nums) {
        
        vector<vector<int>> result;
        if (nums.size() < 3)
            return result;
        // sort vector
        sort(nums.begin(), nums.end());
        
        for (int i = 0; i < nums.size() - 2; i++) {
            // //avoid duplicate solutions
            if (i == 0 || nums[i] > nums[i - 1]) {
                int x = -nums[i];
                
                int start = i + 1;
                int end = (int)nums.size() - 1;
                
                while (start < end) {
                    //case 1
                    if (nums[start] + nums[end] == x) {
                        vector<int> temp;
                        temp.push_back(nums[i]);
                        temp.push_back(nums[start]);
                        temp.push_back(nums[end]);
                        result.push_back(temp);
                        start++;
                        end--;
                        //avoid duplicate solutions
                        while (start < end && nums[end] == nums[end + 1])
                            end--;
                        while (start < end && nums[start] == nums[start - 1])
                            start++;
                        //case 2
                    } else if (nums[start] + nums[end] < x) {
                        start++;
                        //case 3
                    } else {
                        end--;
                    }
                }
                
            }
        }
        
        return result;
    }<span style="color:#ff0000;">
</span>