每日一題(day1)

Leetcode001

Two Sum問題
知識更新:find_if、sort、lambda表達式

方法1:暴力方法(失敗)

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

失敗原因:時間複雜度太高

方法2:哈希方法(失敗)

    vector<int> twoSum(vector<int>& nums, int target) 
    {
        vector<int> res(2);
        unordered_map<int, int> nums_map;
        for (int i = 0; i < nums.size(); i++)
        {
            auto j = find_if(nums_map.begin(), nums_map.end(), [&](const std::pair<int, int> &pair)
            {
                return pair.second == target - nums[i];
            });

            if (j != nums_map.end())
            {
                res[0] = j->first;
                res[1] = i;
                break;
            }           
            nums_map[i] = nums[i];
        }
        return res;
    }

失敗原因:map不支持按值查找,用find_if代替

方法3:雙指針方法(成功)

vector<int> twoSum(vector<int>& nums, int target) 
    {
        vector<int> res(2);
        vector<pair<int,int>> nums_sorted;
        int head = 0;
        int tail = nums.size() - 1;
        for (int i = head; i <= tail; i++)
            nums_sorted.push_back(make_pair(i, nums[i]));
        sort(nums_sorted.begin(), nums_sorted.end(), [&](pair<int,int> a, pair<int,int> b)
        {   
            //從小到大排序
            return a.second < b.second;
        });

        while (head < tail)
        {
            int sum = nums_sorted[head].second + nums_sorted[tail].second;
            if (sum == target)
            {
                res[0] = nums_sorted[head].first;
                res[1] = nums_sorted[tail].first;
                break;
            }
            else
                sum > target ? tail-- : head++;
        }
        return res;
    }

不足之處:結果的順序可能會不同

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章