LeetCode刷题记录7-从数组中找出指定条件的3个数字

题目

给定一个数组nums的n个整数,a,b,c在nums中,并且a + b + c = 0。在数组中找到所有唯一的三个数字,它们的总和为零。(并且每一组由3个数字组成并每一组不存在完全相同的情况)
如下:

For example, given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

三层for循环

这种可能是能想到的最简单的方法了,遍历每一组数,然后计算结果,并去除重复(这也是本人想到的笨方法)。

我想要介绍的主要是第二种解法,具体如下:

跳过已扫描的数字

因为a+b+c=0,所以我们需要固定一个数字a,取反(-a),在剩余的数字中查找b+c和其相等的数字,并且遇到重复我们需要继续扫描下一个数字(该解法来自leedCode,链接会在文本最下方):

vector<vector<int> > threeSum(vector<int> &num) {
    //[-1, 0, 1, 2, -1, -4]                         

    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(3, 0);
                triplet[0] = num[i];
                triplet[1] = num[front];
                triplet[2] = 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;
}

原地址在这
https://leetcode.com/problems/3sum/discuss/

猜你喜欢

转载自blog.csdn.net/gx864102252/article/details/78826221