力扣10.11 - 技术栈

力扣10.11

益达爱喝芬达2024-10-15 18:24

3164. 优质数对的总数 II

给你两个整数数组 nums1 和 nums2，长度分别为 n 和 m。同时给你一个正整数 k。

如果 nums1[i] 可以被 nums2[j] * k 整除，则称数对 (i, j) 为优质数对（0 <= i <= n - 1, 0 <= j <= m - 1）。

返回优质数对的总数。

数据范围

1 <= n, m <= 105
1 <= nums1[i], nums2[j] <= 106
1 <= k <= 103

分析

枚举倍数，记录nums1和nums2每个数字出现的个数记录在cnt1和cnt2中，对于nums2的每个数kk，枚举它的倍数，与k相乘后去cnt1中找，最后答案加上对应的值

代码

c 复制代码

typedef long long LL;
class Solution {
public:
    const static int N = 1e6 + 6;
    int cnt[N];
    map<LL, LL> cnt2;
    long long numberOfPairs(vector<int>& nums1, vector<int>& nums2, int k) {
        LL maxx = 0;
        for(int i = 0; i < nums1.size(); i ++ ) {
            cnt[nums1[i]] ++ ;
            maxx = max(maxx, (LL)nums1[i]);
        }
        for(auto kk : nums2) {
            cnt2[kk] ++;
        }
        LL res = 0;
        for(auto [kk, v] : cnt2) {
            LL t = kk * k;
            for(LL j = t; j <= maxx; j += t ) { // 枚举倍数
                res += v * cnt[j]; // 加法原理
            }
        }
        return res;
    }
};