给你两个整数数组 nums1
和 nums2
,长度分别为 n
和 m
。同时给你一个正整数 k
。
如果 nums1[i]
可以除尽 nums2[j] * k
,则称数对 (i, j)
为 优质数对(0 <= i <= n - 1
, 0 <= j <= m - 1
)。
返回 优质数对 的总数。
示例 1:
输入:nums1 = [1,3,4], nums2 = [1,3,4], k = 1
输出:5
解释:
5个优质数对分别是 (0, 0)
, (1, 0)
, (1, 1)
, (2, 0)
, 和 (2, 2)
。
示例 2:
输入:nums1 = [1,2,4,12], nums2 = [2,4], k = 3
输出:2
解释:
2个优质数对分别是 (3, 0)
和 (3, 1)
。
思路
遍历 可以除尽,也就是进行取余计算得到的结果为0 计算结果中0的数量,就是优质数对的总数
解题过程
使用列表推导式计算每个元素取余的结果,其实就是两层for循环,nums1中的第一个元素和nums2中的所有元素进行取余计算,然后是nums1中的第二个元素....以此类推,得到的是一个二维列表 还是两层for循环,找到二维列表中的0并计数,每有一个0记一个1,最后把所有1加起来得到最终结果
class Solution(object):
def numberOfPairs(self, nums1, nums2, k):
"""
:type nums1: List[int]
:type nums2: List[int]
:type k: int
:rtype: int
"""
# 使用列表推导式计算每个元素取余的结果
a = [[nums1[i] % (nums2[j]*k) for j in range(len(nums2))] for i in range(len(nums1))]
b = sum(1 for row in a for element in row if element == 0)
return b