LeetCode 454. 四数相加 II

题目描述:

解法一(暴力法,显然超时):

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        int res=0;
        for(int i=0;i<A.size();i++){
            for(int j=0;j<B.size();j++){
                for(int k=0;k<C.size();k++){
                    for(int l=0;l<D.size();l++){
                        if(A[i]+B[j]+C[k]+D[l]==0)
                            res++;
                    }
                }
            }
        }
        return res;
    }
};

解法二(空间换时间,使用记录表):

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int,int> record;
        int res=0;
        for(int i=0;i<A.size();i++){
            for(int j=0;j<B.size();j++){
                record[A[i]+B[j]]++;
            }
        }
        for(int i=0;i<C.size();i++){
            for(int j=0;j<D.size();j++){
                if(record.count(0-C[i]-D[j]))
                    res+=record[0-C[i]-D[j]];
            }
        }
        return res;
    }
};

 

posted @ 2019-09-01 20:01  DH_HUSTer  阅读(11)  评论(0编辑  收藏  举报