LeetCode_3Sum

description:给定一个保存整形数据的vector,编写程序检查是否有三个数的和等于0的,返回一个vector保存不重复的答案

解法1:暴力迭代

vector<vector<int>> threeSum(vector<int> &nums)
{
    vector<vector<int>> res;
    for (int i = 0; i != nums.size() - 1; ++i)
    {
        for (int j = i + 1; j != nums.size(); ++j)
        {
            for (int k = 0; k != nums.size(); ++k)
            {
                if (k != i && k != j && nums[i] + nums[j] + nums[k] == 0)
                {
                    vector<int> temp{nums[i], nums[j], nums[k]};
                    sort(temp.begin(), temp.end());
                    res.push_back(temp);
                }
            }
        }
    }
    sort(res.begin(),res.end());
    res.erase(unique(res.begin(), res.end()), res.end());
    return res;
}

复杂度:O(N3),嵌套的三个for循环是查找所有问题的解,然后全部添加到res容器中,然后的两行表达式是去掉重复的解,不推荐该方法,输入数据偏大时运行时间会超出限制

解法2:利用Two Sum的解法,将一个元素作为基数,然后在剩下的元素中找和为基数相反数的元素

vector<vector<int> > threeSum(vector<int> &num) {
    
    vector<vector<int> > res;

    std::sort(num.begin(), num.end());

    for (int i = 0; i < num.size(); i++) {
        
        int target = -num[i];
        int front = i + 1;
        int back = num.size() - 1;

        while (front < back) {

            int sum = num[front] + num[back];
            
            // Finding answer which start from number num[i]
            if (sum < target)
                front++;

            else if (sum > target)
                back--;

            else {
                vector<int> triplet{num[i],num[front],num[back]};
                res.push_back(triplet);
                
                // Processing duplicates of Number 2
                // Rolling the front pointer to the next different number forwards
                while (front < back && num[front] == triplet[1]) front++;

                // Processing duplicates of Number 3
                // Rolling the back pointer to the next different number backwards
                while (front < back && num[back] == triplet[2]) back--;
            }
            
        }

        // Processing duplicates of Number 1
        while (i + 1 < num.size() && num[i + 1] == num[i]) 
            i++;

    }
    
    return res;
    
}

外层for循环是将num[i]作为基数,target是要在剩下元素中找的两个数的和,该方法用两个指针来寻找,通过提交表明第一种方法的删除重复元素的方式会使运行时间增加几乎两倍,所以在第二种方法中使用while循环来跳过重复的元素。
最坏情况下为O(N2);

猜你喜欢

转载自blog.csdn.net/lancelot0902/article/details/92585712