leetcode 454.四数相加II

关注

原创最新推荐文章于 2026-03-15 00:22:05 发布 138 阅读 AI 写同款· GEO 优化 ›

代码随想录算法训练营Day6 | Leetcode 454四数相加II、383赎金信、15三数之和、18四数之和 Leetcode454四数相加II、383赎金信、15三数之和、18四数之和解题思路与技巧、代码阅读详情

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int count=0;

        for(int i=0;i<nums1.size();i++){
            for(int j=0;j<nums2.size();j++){
                for(int k=0;k<nums3.size();k++){
                    for(int l=0;l<nums4.size();l++){
                        if(i>=0&&i<nums1.size()&&j>=0&&j<nums1.size()&&k>=0&&k<nums1.size()&&l>=0&&l<nums1.size()&&(nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0)){
                            count++;
                        }
                    }
                }
            }
        }
        return count;
    }
};

超时

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int count=0;
        unordered_map<int,int> ab;
        for(int a:nums1){
            for(int b:nums2){
                ab[a+b]++;
            }
        }
        for(int c:nums3){
            for(int d:nums4){
                if(ab.find(0-c-d)!=ab.end()){
                    count+=ab[0-c-d];
                }
            }
        }
        return count;
    }
};

时间复杂度降到n^2