zoukankan      html  css  js  c++  java
  • LeetCode 454. 四数相加II

    题目描述

    给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0
    为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 (-2^{28})(2^{28} - 1) 之间,最终结果不会超过 (2^{31} - 1)

    例如:

    输入:
    A = [ 1, 2]
    B = [-2,-1]
    C = [-1, 2]
    D = [ 0, 2]
    
    输出:
    2
    
    解释:
    两个元组如下:
    1. (0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
    2. (1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0
    

    来源:力扣(LeetCode)
    链接:https://leetcode-cn.com/problems/4sum-ii

    思路解析

    使用一个哈系表 unordered_map<int, int> countAB 存储所有 A + B 的值,countABkey 为 A + B 的可能的结果,value 为该结果出现的次数。
    然后遍历所有 C + D 的值,如果出现与 countABkey 相同的值,则总可能的结果应加上 value

    代码实现

    class Solution {
    public:
        int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
            unordered_map<int, int> countAB;
            for(int u : A) {
                for(int v : B) {
                    countAB[u + v]++;
                }
            }
            int res = 0;
            for(int u : C) {
                for(int v : D) {
                    res += countAB[-u - v];
                }
            }
            return res;
        }
    };
    
  • 相关阅读:
    NFS服务
    DNS服务
    git使用笔记(七)版本回退和撤销
    git使用笔记(六)github
    git使用笔记(五)打标签
    git使用笔记(四)远程操作
    git使用笔记(三)文件忽略
    git使用笔记(二)分支与合并
    git使用笔记(一)入门
    GDB使用小记
  • 原文地址:https://www.cnblogs.com/xqmeng/p/14047734.html
Copyright © 2011-2022 走看看