Leetcode15 3Sum

这里提供两种方法

第一中非常的耗时，没有经过优化

bool equalto(vector<vector<int>>num1, vector<int> num2)
{
    //if (num2.size() == 0)
    //{
    //    return false;
    //}
    sort(num2.begin(), num2.end());
    for (int i = 0; i < num1.size(); i++)
    {
        sort(num1[i].begin(), num1[i].end());
        if (num1[i] == num2) return false;
    }
    return true;
}

vector<vector<int>> threeSum1(vector<int>& nums) {
    vector<vector<int>> result(NULL);
    vector<int> tmp;
    if (nums.size() <= 2) return result;
    for (int i = 0; i < nums.size(); i++)
    {
        tmp.push_back(nums[i]);
        for (int j = i + 1; j < nums.size(); j++)
        {
            tmp.push_back(nums[j]);
            for (int k = j + 1; k < nums.size(); k++)
            {
                tmp.push_back(nums[k]);
                if ((tmp[0] + tmp[1] + tmp[2]) == 0)
                {
                    if (result.size() == 0)
                    {
                        result.push_back(tmp);
                        tmp.pop_back();
                        continue;
                    }
                    else if (equalto(result, tmp))// 没有就放进去
                    {
                        result.push_back(tmp);
                        tmp.pop_back();
                        continue;
                    }

                }
                tmp.pop_back();
                continue;
            }
            tmp.pop_back();
            continue;
        }
        tmp.pop_back();
        continue;
    }
    return result;
}

 

第二种方法经过优化

vector<vector<int>> threeSum(vector<int>& nums) {
    vector<vector<int>> result(NULL);
    int len = nums.size();
    if (nums.size() <= 2) return result;
    sort(nums.begin(), nums.end());//排序方便后面处理
    for (int i = 0; i < nums.size() - 2; i++)
    {
        if ((nums[i] + nums[i + 1] + nums[i + 2]) > 0) break;//这种情况一出现，说明i后面的任意3个数相加都是大于0
        if ((nums[i] + nums[len - 2] + nums[len - 1]) < 0) continue;// 因为nums[len - 2]，nums[len - 1]是nums中最大的两个数，说明i太小
        if (i > 0 && nums[i] == nums[i - 1]) continue;// 防止第一个数字重复的情况，[-1,-1,0,1]会发生[-1,0,1],[-1,0,1]
        if (nums[i] > 0) break;// 只要出现这种情况，说明包括i以后的任意的数字都是大于0
        int k = len - 1;
        for (int j = i + 1; j < k;)
        {
            int sum = nums[i] + nums[j] + nums[k];
            //说明最后一个值太大了，往前
            if (sum > 0) k--;// 最后一个值太大，往前调整
            else if (sum < 0) ++j;// 第一个数字太小，往后调整
            else 
            {
                result.push_back({nums[i],nums[j],nums[k]});//表示已经找到了一串记为 A
                do { ++j; } while ((nums[j] == nums[j - 1]) && (j < k));// j 表示的是三个数中第二个数字，说明和A一样，不能再次放入
                do { --k; } while (nums[k] == nums[k + 1] && (j < k));// k 表示的是三个数中第三个数字，说明和A一样，不能再次放入
            }
        }
    }
    return result;
}