【Leetcode】K sum problem

1. Two Sum

BruteForce

首先我們可以想到一個暴力破解的方法，每兩個數字之間求和看是不是目標值，時間複雜度爲$O(n^2)$

vector<int> twoSum_BruteForce(vector<int> &nums, int target)
{
    vector<int> res;
    for (int i = 0; i < nums.size() - 1; ++i)
    {
        for (int j = i + 1; j < nums.size(); j++)
        {
            if ((nums[i] + nums[j]) == target)
            {
                res.push_back(i);
                res.push_back(j);
            }
        }
    }
    return res;
}

Hash Table

利用Hash表存儲索引與對應值，時間複雜度$O(n)$

vector<int> twoSum(vector<int> &nums, int target)
{
    vector<int> res;
    unordered_map<int, int> hash;
    for (int i = 0; i < nums.size(); i++)
    {
        if (hash.find(target - nums[i]) != hash.end())
        {
            res.push_back(i);
            res.push_back(hash[target - nums[i]]);
            return res;
        }
        hash[nums[i]] = i;
    }
    return res;
}

15. 3Sum

BruteForce

我們首先也可以想到一種暴力破解的方式，時間複雜度爲$O(n^3)$，這樣的方法肯定會超時。

vector<vector<int>> threeSum_BruteForce(vector<int> &nums)
{
    vector<vector<int>> res;
    sort(nums.begin(), nums.end());
    for (int i = 0; i < nums.size() - 2; ++i)
    {
        for (int j = i + 1; j < nums.size() - 1; ++j)
        {
            for (int k = j + 1; j < nums.size(); ++k)
            {
                if ((nums[i] + nums[j] + nums[k]) == 0)
                {
                    vector<int> tmp;
                    tmp.push_back(nums[i]);
                    tmp.push_back(nums[j]);
                    tmp.push_back(nums[k]);
                    res.push_back(tmp);
                }
            }
        }
    }
    res.erase(unique(res.begin(), res.end()), res.end());//Remove duplicate
    return res;
}