Leetcode 3704. Count No-Zero Pairs That Sum to N

• Leetcode 3704. Count No-Zero Pairs That Sum to N
  • 1. 解题思路
  • 2. 代码实现

• 3704. Count No-Zero Pairs That Sum to N

1. 解题思路

这一题思路上就是一个动态规划的思路，我们只需要考虑每一位上两个数各自取什么非零的位数即可。

唯一需要注意的是，需要注意一下余数以及开头为0的情况即可。

2. 代码实现

给出python代码实现如下：

class Solution:
    def countNoZeroPairs(self, n: int) -> int:
        s = str(n)
        l = len(s)

        @lru_cache(None)
        def dp(idx, r, is_num1_empty, is_num2_empty):
            if idx >= l:
                return 0 if (r != 0 or is_num1_empty or is_num2_empty) else 1
            tgt = int(s[idx]) + 10 * r
            if tgt <= 1 and (not is_num1_empty) and (not is_num2_empty):
                return 0
            ans = 0
            for i in range(min(tgt+1, 10)):
                j = tgt-i
                if 0 < i < 10 and 0 < j < 10:
                    ans += dp(idx+1, 0, False, False)
                elif is_num1_empty and i == 0 and 0 < j < 10:
                    ans += dp(idx+1, 0, True, False)
                elif is_num2_empty and j == 0 and 0 < i < 10:
                    ans += dp(idx+1, 0, False, True)

                j = tgt-1-i
                if 0 < i < 10 and 0 < j < 10:
                    ans += dp(idx+1, 1, False, False)
                elif is_num1_empty and i == 0 and 0 < j < 10:
                    ans += dp(idx+1, 1, True, False)
                elif is_num2_empty and j == 0 and 0 < i < 10:
                    ans += dp(idx+1, 1, False, True)
                elif is_num1_empty and is_num2_empty and i == 0 and j == 0:
                    ans += dp(idx+1, 1, True, True)
            return ans
        
        return dp(0, 0, True, True)

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