Leetcode 3725. Count Ways to Choose Coprime Integers from Rows

• Leetcode 3725. Count Ways to Choose Coprime Integers from Rows
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3725. Count Ways to Choose Coprime Integers from Rows

1. 解题思路

这一题虽然是一个hard的题目，但是思路上实际就是一个比较常规的动态规划的思路，对着代码看一下就行了，就没什么好多做展开的了。

2. 代码实现

给出python代码实现如下：

MOD = 10**9+7

class Solution:
    def countCoprime(self, mat: List[List[int]]) -> int:
        n, m = len(mat), len(mat[0])

        @lru_cache(None)
        def dp(idx, _gcd):
            if _gcd == 1:
                return pow(m, n-idx, mod=MOD)
            elif idx == n:
                return 0
            ans = 0
            for i in range(m):
                ans = (ans + dp(idx+1, gcd(_gcd, mat[idx][i]))) % MOD
            return ans

        ans = 0
        for i in range(m):
            ans = (ans + dp(1, mat[0][i])) % MOD
        return ans

提交代码评测得到：耗时1770ms，占用内存25.96MB。