402. 移掉K位数字
给定一个以字符串表示的非负整数 num,移除这个数中的 k 位数字,使得剩下的数字最小。
注意:
num 的长度小于 10002 且 ≥ k。
num 不会包含任何前导零。
贪心的思路,每次从开头循环,如果当前字符大于之后的一个字符,则删除,即可满足题意,但是速度比较慢,需要优化
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
while(k):
k -= 1
flag = True
for i in range(len(num)-1):
if flag and num[i]>num[i+1]:
num = num[:i]+num[i+1:]
flag = False
if flag:
num = num[:-1]
num = num.lstrip("0")
return num if num else "0"
加个位置标识符 index,可以大大减少迭代时间
比如 num = “1234567890”
如果按照上边的代码,每次都会从第0位开始循环,但是如果每次只去除一个字符的话,只需要从删除的字符上一个位置开始搜索即可
这样可以把理论上的时间复杂度从O(nk)变成O(n),其中k表示删除的字符数,n表示字符串长度,只需注意如果删除的是第0个字符时,要将index赋值成0,而不是-1
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
index = 0
while(k):
k -= 1
flag = True
for i in range(index,len(num)-1):
if num[i]>num[i+1]:
num = num[:i]+num[i+1:]
flag = False
index = i-1 if i-1>=0 else 0
break
if flag:
num = num[:-1]
num = num.lstrip("0")
return num if num else "0"