力扣10.11

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中找，最后答案加上对应的值

代码

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;}
};