leetcode刷题系列-738单调递增的数字

题目描述

给定一个非负整数 N，找出小于或等于 N 的最大的整数，同时这个整数需要满足其各个位数上的数字是单调递增。

（当且仅当每个相邻位数上的数字 x 和 y 满足 x <= y 时，我们称这个整数是单调递增的。）

示例 1:

输入: N = 10
输出: 9

示例 2:

输入: N = 1234
输出: 1234

示例 3:

输入: N = 332
输出: 299

初次解法

从左到右找到第一个递减的数字，记录其下标为pos。结束时则有nums[pos] < nums[pos - 1]。比如nums=[3，3，2，2，2]。则pos=1。将pos=1处的数字减1。则nums=[3,2,2,2,2]。从pos依次往前找，直到nums[pos - 1] < nums[pos]为止。

class Solution:
    def monotoneIncreasingDigits(self, N: int) -> int:
        bits = list(str(N))
        pos = -1
        for i, e in enumerate(bits[:-1]):
            if e > bits[i + 1]:
                pos = i
                bits[pos] = str(int(bits[pos]) - 1)
                break
        if pos == -1:
            return N
        while pos > 0:
            if bits[pos] < bits[pos - 1]:
                bits[pos - 1] = str(int(bits[pos - 1]) - 1)
                pos -= 1
            else:
                break

        #字符串给列表元素赋值
        bits[pos + 1:] = '9' * (len(bits) - 1 - pos)
        return int(''.join(bits))

循环改进：

def monotoneIncreasingDigits(self, N: int) -> int:
    bits = list(str(N))
    pos = 1
    while pos < len(bits) and bits[pos - 1] <= bits[pos]:
        pos += 1
    if pos == len(bits):
        return N
    while pos > 0 and bits[pos] < bits[pos - 1]:
            bits[pos - 1] = str(int(bits[pos - 1]) - 1)
            pos -= 1
    #字符串给列表元素赋值
    bits[pos + 1:] = '9' * (len(bits) - 1 - pos)
    return int(''.join(bits))

下面的做法通过将for改为while。通过while控制退出条件,使得退出条件更加清晰。同时避免了break等语句使用。代码可读性更强。更加简洁。

注意第一个循环中的小于等于。表明退出条件一定是num[pos - 1] > num[pos]