2023华为OD机试真题-统计匹配的二元组个数(JAVA、Python、C++)

题目描述：
给定两个数组A和B，若数组A的某个元素A[i]与数组B中的某个元素B[j]满足 A[i] == B[j], 则寻找到一个值匹配的二元组(i, j). 请统计在这两个数组A和B中，一共存在多少个这样的二元组。 
输入描述：
第一行输入数组A的长度M；第二行输入数组B的长度N；第三行输入数组A的值；第四行输入数组B的值。
1 <= M, N <= 100000
A, B数组中数值的取值均小于100000；
输出描述：
输出匹配的二元组个数。 
补充说明：
若不存在相等的值，则输出0. 所采用的算法复杂度需小于O(N^2)，否则会超时。输入数组中允许出现重复数字，一个数字可以匹配多次。

 收起
示例1
输入：
5
4
1 2 3 4 5
4 3 2 1
输出：
4
说明：
若下标从0开始，则匹配的二元组分别为(0, 3), (1, 2), (2, 1), (3, 0), 共计4个。
示例2
输入：
6
3
1 2 4 4 2 1
1 2 3
输出：
4
说明：
若下标从0开始，则匹配的二元组分别为(0, 0), (1, 1), (4, 1), (5, 0) 共计4个。
 

import java.util.*;
 
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int numA = in.nextInt();
        int numB = in.nextInt();
        //数组a
        int[] numAs = new int[numA];
        for (int i = 0; i < numA; i++) {
            numAs[i] = in.nextInt();
        }
        //数组b
        int[] numBs = new int[numB];
        for (int i = 0; i < numB; i++) {
            numBs[i] = in.nextInt();
        }
 
        //数组a放入map
        HashMap<Integer, Integer> aMap = new HashMap<>();
        for (int a : numAs) {
            aMap.put(a, aMap.getOrDefault(a, 0) + 1);
        }
        //遍历数组b
        int sum = 0;
        for (int b : numBs) {
            sum += aMap.getOrDefault(b, 0);
        }
        
         System.out.print(sum);
    }
}

#include<bits/stdc++.h>
using namespace std;
int n,m,temp;
int main(){
    while(cin>>n>>m){
        unordered_map<int, int> a,b;
        unordered_set<int> c;
        for(int i=0;i<n;i++){
            cin>>temp;
            c.insert(temp);
            a[temp]+=1;
        }
        int ans=0;
        for(int i=0;i<m;i++){
            cin>>temp;
            c.insert(temp);
            b[temp]+=1;
            ans+=a[temp];
        }
        cout<<ans<<endl;
    }
}