b_lc_插入后的最大值(贪心+区分正负数)

给定一个整数n,要求插入一个数字x,是的n最大化

思路:

  • 如果是正数,则尽量让高位变大,也就是说如果有数位y比x要小,则可以让x放在y前面,越高越好
  • 如果是负数,则与上反之
class Solution:
    def maxValue(self, n: str, x: int) -> str:
        flag = False
        if n[0] == '-':
            flag = True

        if flag: #负数找到第一个大于
            less_idx = -1
            n = n[1:]
            for i in range(len(n)):
                if int(n[i]) > x:
                    less_idx = i
                    break
            if less_idx == -1:
                less_idx = len(n)
            str_list = list(n)
            str_list.insert(less_idx, str(x))
            n = ''.join(str_list)
        else:
            gree_idx = -1
            for i in range(len(n)):
                if int(n[i]) < x:
                    gree_idx = i
                    break
            if gree_idx == -1:
                gree_idx = len(n)
            str_list = list(n)
            str_list.insert(gree_idx, str(x))
            n = ''.join(str_list)
        if flag:
            n = '-' + n
        return n

写的太长了,优化一下:

class Solution:
    def maxValue(self, n: str, x: int) -> str:

        if n[0] == '-':
            for i in range(1, len(n)):
                if int(n[i]) > x:
                    return n[:i] + str(x) + n[i:]
        else:
            for i in range(len(n)):
                if int(n[i]) < x:
                    return n[:i] + str(x) + n[i:]
        return n + str(x)
posted @ 2021-05-30 16:24  童年の波鞋  阅读(55)  评论(0编辑  收藏  举报