代码随想录训练营第七天|双指针

454.四数相加II

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> cnt;
        for(int i:nums1)
            for(int j:nums2)
                cnt[i+j]++;

        int res=0;
        for(int i:nums3){
            for(int j:nums4){
                if(cnt.count(0-(i+j))!=0)
                    res+=cnt[0-(i+j)];
            }
        }
        return res;

    }
};

空间换时间，利用map将O(n^4)问题降为 O(n^2)

383. 赎金信

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        unordered_map<char,int> seen;
        for(auto x:magazine){
            seen[x]++;
        }
        for(auto x:ransomNote){
            if(seen[x]==0)
                return false;
            seen[x]--;
        }
        return true;
    }
};

在C++中，使用std::map访问一个不存在的key会自动插入该key，并将其值初始化为value_type的默认值。所以seen[x]==0代表两种情况：

1.magazine中从未出现过x 

2.magazine中出现过x，但已被消耗完毕

以上两种情况均应返回false，除此之外，表明当前字符可被正常构造，所有字符构造完毕后返回true。

15. 三数之和

class Solution {
public: 
    vector<vector<int>> result;
    vector<int> temp;
    int sum;

    void dfs(vector<int>& nums,int startIdx){
        if(temp.size()==3){
            if(sum==0)
                result.push_back(temp);
            return;
        }
        for(int i=startIdx;i<=nums.size()-(3-temp.size())&&sum+nums[i]<=0;i++){
            sum+=nums[i];
            temp.push_back(nums[i]);
            dfs(nums,i+1);
            temp.pop_back();
            sum-=nums[i];
            while(i+1<nums.size()&&nums[i+1]==nums[i])
                i++;
        }
    }

    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        dfs(nums,0);
        return result;
    }
};

回溯解法，超时了Orz .. 

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> result;
        for (int i = 0; i <= nums.size() - 3; i++) {
            int left = i + 1, right = nums.size() - 1;
            while (left < right) {
                if (nums[i] + nums[left] + nums[right] == 0) {
                    result.push_back({nums[i], nums[left], nums[right]});
                    while (left < right && nums[left + 1] == nums[left])
                        left++;
                    while (left < right && nums[right - 1] == nums[right])
                        right--;
                    left++;
                    right--;
                } else if (nums[i] + nums[left] + nums[right] > 0) {
                    right--;
                } else {
                    left++;
                }
            }
            while (i + 1 <= nums.size() - 3 && nums[i + 1] == nums[i])
                i++;
        }
        return result;
    }
};

三数之和=暴搜首位+两数之和（双指针） 

18. 四数之和

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        if(nums.size()<4)
            return {};
        for(int i=0;i<=nums.size()-4;i++){
            for(int j=i+1;j<=nums.size()-3;j++){
                int left=j+1, right=nums.size()-1;
                while(left<right){
                    long val=(long)nums[i]+nums[j]+nums[left]+nums[right];
                    if(val==target){
                        result.push_back({nums[i],nums[j],nums[left],nums[right]});
                        while(left<right&&nums[left+1]==nums[left])
                            left++;
                        while(left<right&&nums[right-1]==nums[right])
                            right--;
                        left++;
                        right--;
                    }
                    else if(val>target)
                        right--;
                    else    
                        left++;
                }
                while(j+1<=nums.size()-3&&nums[j+1]==nums[j])
                    j++;
            }
            while(i+1<=nums.size()-4&&nums[i+1]==nums[i])
                i++;
        }
        return result;
    }
};

四数之和=暴搜前两位+两数之和（双指针）