LeetCode每日一题4.17

2176.统计数组中相等且可以被整除的数对

问题分析

理解问题:

给定一个整数数组 nums 和一个整数 k。

需要找到所有满足 0 <= i < j < n 且 nums[i] == nums[j] 且 (i * j) 能被 k 整除的数对 (i, j) 的数目

思路

算法思路:

遍历数组中的每一对元素 (i, j)(其中 i < j)。

检查 nums[i] 是否等于 nums[j]。

如果相等,再检查 i * j 是否能被 k 整除。

如果上述条件都满足,则计数器加一。

代码

python 复制代码
class Solution:
    def countPairs(self, nums: List[int], k: int) -> int:
        ans = 0
        for i in range(len(nums)-1):
            for j in range(i + 1, len(nums)):
                if nums[i] == nums[j] and (i * j) % k == 0:
                    ans += 1
        return ans

复杂度分析

时间复杂度:两个for循环: ( O(n^2) )。

空间复杂度:( O(1) )

学习

优化方法:枚举右维护左

如果数组长度n = 10^5 ,暴力枚举超时

问题简化:

代码:

python 复制代码
# 预处理每个数的因子
MX = 101
divisors = [[] for _ in range(MX)]
for i in range(1, MX):
    for j in range(i, MX, i):
        divisors[j].append(i)

class Solution:
    def countPairs(self, nums: list[int], k: int) -> int:
        ans = 0
        cnt = defaultdict(int)
        for j, x in enumerate(nums):  # 枚举 j,计算左边有多少个符合要求的 i
            if j and x == nums[0]:
                ans += 1  # 单独统计 i=0 的情况
            k2 = k // gcd(k, j)  # i 必须是 k2 的倍数
            ans += cnt[(x, k2)]
            for d in divisors[j]:  # j 是 d 的倍数
                cnt[(x, d)] += 1
        return ans

复杂度分析

方法学习:

作者:灵茶山艾府

链接:https://leetcode.cn/problems/count-equal-and-divisible-pairs-in-an-array/solutions/1277713/mo-ni-by-endlesscheng-wegn/

来源:力扣(LeetCode)

著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

相关推荐
桦016 分钟前
[除自身以外数组的乘积]
算法
CoovallyAIHub1 小时前
原来工业 AI 异常检测只做了一半?AnomalyNCD 补上了“最关键一环”
深度学习·算法·计算机视觉
shenghaide_jiahu1 小时前
数学建模——01规划/整数规划
算法·数学建模
CoovallyAIHub1 小时前
数据集分享 | PCB缺陷检测与玻璃缺陷实例分割数据集分享
深度学习·算法·计算机视觉
蒟蒻小袁2 小时前
力扣面试150题--只出现一次的数字
数据结构·算法·leetcode
Star在努力2 小时前
16-C语言:第17天笔记
c语言·笔记·算法
啊阿狸不会拉杆2 小时前
《Java 程序设计》第 11 章 - 泛型与集合
java·开发语言·jvm·数据结构·算法
NuyoahC2 小时前
笔试——Day23
c++·算法·模拟
恣艺3 小时前
LeetCode 68:文本左右对齐
算法·leetcode·c#
Alfred king3 小时前
Leetcode 四数之和
算法·leetcode·职场和发展·数组·排序·双指针