leetcode专题训练 15. 3Sum

1.超时的方法，最开始想的方法

class Solution {
public:
    
    int BinarySearch(int left, int right, int result, vector<int>& nums) {
        int l = nums.size();
        int mid;
        while (left <= right) {
            mid = (right + left) / 2;
            if (nums[mid] == result) {
                return mid;
            }
            if (nums[mid] < result) { 
                left++;
            }
            if (nums[mid] > result) {
                right--;
            }
        }
        return -1;
    }
    
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort (nums.begin(), nums.end());
        int l = nums.size();
        vector<vector<int>> result;
        vector<int> tmp_result;
        result.clear();
        for (int i = 0; i < l; i++) {
            if (nums[i] > 0) break;
            for (int j = l - 1; j != i; j--) {
                if (nums[j] < 0) break;
                int k = BinarySearch(i + 1, j - 1, -nums[i] - nums[j], nums);
                if (k == -1) continue;
                tmp_result.clear();
                tmp_result.push_back(nums[i]);
                tmp_result.push_back(nums[j]);
                tmp_result.push_back(nums[k]);
                vector<vector<int>>::iterator it = find(result.begin(),result.end(),tmp_result);
                if (it!=result.end()) continue; //vector中已有
                result.push_back(tmp_result);
            }
        }
        return result;
    }
};

2.超时2.0， 双指针+排序 O($n^2$)

class Solution {
public:
    
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort (nums.begin(), nums.end());
        int l = nums.size();
        vector<vector<int>> result;
        vector<int> tmp_result;
        result.clear();
        for (int i = 0; i < l; i++) {
            if (nums[i] > 0) break;
            int j = i + 1;
            int k = l - 1;
            while (j < k) {
                //cout << i << " " << j << " " << k << endl;
                if (nums[k] < 0) break;
                //cout << nums[i] << " " << nums[j] << " " << nums[k] << endl << endl;
                if (nums[i] + nums[j] + nums[k] == 0) {
                    tmp_result.clear();
                    tmp_result.push_back(nums[i]);
                    tmp_result.push_back(nums[j]);
                    tmp_result.push_back(nums[k]);
                    result.push_back(tmp_result);
                    j++;
                    k--;
                } else if (nums[i] + nums[j] + nums[k] < 0) {
                    j++;
                } else {
                    k--;
                }
            }
        }
        set<vector<int>> st(result.begin(), result.end());
        result.assign(st.begin(), st.end());
        return result;
    }
};

3.通过1.0， 双指针+排序 O($n^2$)
相对于超时2.0只加了一句话：

            if (i != 0 && nums[i] == nums[i - 1]) continue;

因为该死的数据中有全都是0的情况，不加这句话就会超时。加上这句话就代表着如果这个数值已经被处理过，那么就不再处理这个数值了。

class Solution {
public:
    
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort (nums.begin(), nums.end());
        int l = nums.size();
        vector<vector<int>> result;
        vector<int> tmp_result;
        result.clear();
        for (int i = 0; i < l; i++) {
            if (nums[i] > 0) break;
            if (i != 0 && nums[i] == nums[i - 1]) continue;
            int j = i + 1;
            int k = l - 1;
            while (j < k) {
                if (nums[k] < 0) break;
                if (nums[i] + nums[j] + nums[k] == 0) {
                    tmp_result.clear();
                    tmp_result.push_back(nums[i]);
                    tmp_result.push_back(nums[j]);
                    tmp_result.push_back(nums[k]);
                    result.push_back(tmp_result);
                    j++;
                    k--;
                } else if (nums[i] + nums[j] + nums[k] < 0) {
                    j++;
                } else {
                    k--;
                }
            }
        }
        set<vector<int>> st(result.begin(), result.end());
        result.assign(st.begin(), st.end());
        return result;
    }
};

4.优化了时间，加入了第二个和第三个指针所扫描元素的判重，判重之后同时也保证了加入result里的序列都是没有出现过的，就不用再用set进行去重了。

class Solution {
public:
    
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort (nums.begin(), nums.end());
        int l = nums.size();
        vector<vector<int>> result;
        vector<int> tmp_result;
        result.clear();
        for (int i = 0; i < l; i++) {
            if (nums[i] > 0) break;
            if (i != 0 && nums[i] == nums[i - 1]) continue;
            int j = i + 1;
            int k = l - 1;
            while (j < k) {
                if (nums[k] < 0) break;
                if (nums[i] + nums[j] + nums[k] == 0) {
                    tmp_result.clear();
                    tmp_result.push_back(nums[i]);
                    tmp_result.push_back(nums[j]);
                    tmp_result.push_back(nums[k]);
                    result.push_back(tmp_result);
                    while (j < k && nums[j] == nums[j + 1]) j++;
                    while (j < k && nums[k] == nums[k - 1]) k--;
                    j++;
                    k--;
                } else if (nums[i] + nums[j] + nums[k] < 0) {
                    while (j < k && nums[j] == nums[j + 1]) j++;
                    j++;
                } else {
                    while (j < k && nums[k] == nums[k - 1]) k--;
                    k--;
                }
            }
        }
        //set<vector<int>> st(result.begin(), result.end());
        //result.assign(st.begin(), st.end());
        return result;
    }
};

老了，没法熬夜了，困成皮皮虾。
下回再也不白天打游戏晚上敲代码了。
当然是要白天打游戏晚上也打游戏了。
还是白天敲代码，晚上打游戏吧。
熬夜使人语无伦次。