程式語言 - LeetCode - C++ - 454. 4Sum II



題目:


解答:

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> mp;

        for (int n1 : nums1) {
            for (int n2 : nums2) {
                mp[n1 + n2] += 1;
            }
        }

        int ans = 0;

        for (int n3: nums3) {
            for (int n4 : nums4) {
                int c = -(n3 + n4);

                if (mp.count(c)) {
                    ans += mp[c];
                }
            }
        }

        return ans;
    }
};