莽撞了~一直在试结果。。。做题不可如此。
下面解法的关键思想是高位的元素如果越小越好,所以如果比之前位上的值更小,那么就把 之前位上的值 去掉~~ 所以比栈顶大的元素直接入栈, 如果比栈顶小,就把栈顶弹出来~~
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
if len(num) == k:
return "0"
res = []
sign = 0
for i in range(len(num)):
if i == 0:
res.append(num[i])
else:
if num[i] >= res[-1]:
res.append(num[i])
else:
while res and sign < k and num[i] < res[-1]:
res.pop()
sign += 1
res.append(num[i])
if sign < k:
res = res[0:len(res)-k+sign]
while len(res) > 1 and res[0] == "0":
res = res[1:]
return "".join(res)