题目
思路
这道题题干中描述的是“移除”,也就是剩下的字符依然要保持原有的排列顺序。同时要使得剩下的数字最小,那需要保证小的数尽量排位在前面。综合考虑本题可以使用单调栈的思路,遍历字符串,如果当前遍历到的字符小于栈顶字符,此时弹出栈顶字符可以使得小数尽量排位在前。在遍历过程中要注意考虑K次的限制,当K减为0时,表明已经不能再删除字符,此时要退出循环。同时如果字符中递增的字符数量较多,可能会出现遍历完后K还没减到0的情况,这种情况需要删除字符末尾的k个字符。在解答时直接使用切片返回答案数组中n前n-k个字符即可。
代码
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
n=len(num)
K=k
q=[]
if n==K: #当n=k时,所有字符都会被弹出,因此结果为0
return '0'
for i in range(n):
#print(num[i],k,q)
if not q:
q.append(num[i])
elif q and k and ord(num[i])<ord(q[-1]):
while q and k and ord(num[i])<ord(q[-1]):
q.pop()
k-=1
q.append(num[i])
else:
q.append(num[i])
return str(int(''.join(q[:n-K])))