1. 程式人生 > 其它 >搜尋(bfs,dfs)

搜尋(bfs,dfs)

✔做題思路 or 感想 :

1.理解了上一題的三數之和後,這一題其實也就差不多了,同樣的套路,同樣的方法,不過就多加了一層for的事情罷了,如果往後要求5數之和,n數之和,亦是如此了

2.這一題的思路仍然是 利用四個指標求和 + 去重操作,只要程式碼不寫歪,大體思路都能對

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>>result;
        if (nums.size() == 0)return result; //防止奇怪測試用例
        sort(nums.begin(), nums.end());     //一定要排序,為了下面的去重操作以及四個指標的操作
        //這裡設計了四個指標的操作,且四個指標有大小順序
        //nums[i] < nums[j] < nums[left] < nums[right]
        for (int i = 0; i < nums.size(); i++) {
            if (i > 0 && nums[i] == nums[i - 1])continue;   //第一個指標的去重
            for (int j = i + 1; j < nums.size(); j++) {
                if (j > i + 1 && nums[j - 1] == nums[j])continue;   //第二個指標的去重
                int left = j + 1;
                int right = nums.size() - 1;
                while (left < right) {
                    int aim = target - (nums[i] + nums[j]); //這裡的操作是防止四個int相加後範圍超過int的範圍
                    int sum = nums[left] + nums[right];
                    if (sum > aim)right--;
                    else if (sum < aim)left++;
                    else {
                        result.push_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
                        //要找完一組資料後再去重,不然會漏
                        while (right > left && nums[right - 1] == nums[right])right--;  //第三個指標的去重
                        while (right > left && nums[left + 1] == nums[left])left++; //第四個指標的去重
                        //兩個指標在完成操作後同時收縮
                        right--;
                        left++;
                    }
                }
            }
        }
        return result;
    }
};