leetcode 周赛 171 5307. 将整数转换为两个无零整数的和 Convert Integer to the Sum of Two No-Zero Integers - python 解法

目录

1. 解答

赛题链接 英文赛题 题库本题目链接

「无零整数」是十进制表示中 不含任何 0 的正整数。

给你一个整数 n，请你返回一个 由两个整数组成的列表 [A, B]，满足：

• A 和 B 都是无零整数
• A + B = n

题目数据保证至少有一个有效的解决方案。

如果存在多个有效解决方案，你可以返回其中任意一个。

示例 1：

输入：n = 2
输出：[1,1]
解释：A = 1, B = 1. A + B = n 并且 A 和 B 的十进制表示形式都不包含任何 0 。

示例 2：

输入：n = 11
输出：[2,9]

示例 3：

输入：n = 10000
输出：[1,9999]

示例 4：

输入：n = 69
输出：[1,68]

示例 5：

输入：n = 1010
输出：[11,999]

提示：

• 2 <= n <= 10^4

解答

语言：python3

设答案两个数字是 n1 和 n2，定义变量 cf 代表上一位是否有进位。策略是：

1. 从低位到高位遍历，开始设置进位标志位（变量 cf）为 0

2. 计算当前位数字减去上一位的进位作为目标数字，如果为 -1，当前目标数字设为 9：

   1. 如果当前位是最高位，直接把这个数字作为 n1 的当前位，n2 不变。
   2. 如果目标数字为 0，n1 这一位为 1，n2 为 9，进位 1（就是把 cf 设为 1）
   3. 如果目标数字为 1，n1 这一位为 2，n2 为 9，进位 1（就是把 cf 设为 1）
   4. n1 这一位为 1，n2 为当前目标数字减一，进位 1（就是把 cf 设为 1）

class Solution:
    def getNoZeroIntegers(self, n: int) -> List[int]:
        digits = list(str(n))
        if len(digits) == 1:
            return [n-1, 1]
        n1 = []
        n2 = []
        cf = 0
        for i in range(len(digits)-1, -1, -1):
            cur_val = int(digits[i]) - cf
            cf = 0
            if cur_val == -1:
                cur_val = 9
                cf = 1
            if i == 0:
                n1.append(str(cur_val))
            elif cur_val == 0:
                n1.append('1')    
                n2.append('9')
                cf = 1
            elif cur_val == 1:
                n1.append('2')
                n2.append('9')
                cf = 1
            else:
                n1.append('1')
                n2.append(str(cur_val-1))
        n1.reverse()
        n2.reverse()
        return [int(''.join(n1)), int(''.join(n2))]