思路:单调(递增)栈+贪心
移除一位数字原则:①尽可能靠左②尽可能大
操作:找出第一处发生递减的地方,删除前一个数,一直递增的 话就删除最后一个数
如:1231,删除3
重复k次,时间复杂度最差会达到 O(nk)
class Solution {
public String removeKdigits(String num, int k) {
if(num.length()==k) return "0";
StringBuilder s=new StringBuilder(num);
for(int i=0;i<k;i++){
int temp=0;
while(temp<s.length()-1 && s.charAt(temp)<=s.charAt(temp+1)){
temp++;
}
if(temp==s.length()-1){
s.delete(s.length()-1,s.length());
}
s.delete(temp,temp+1);
}
//删除前置零
while (s.length() > 1 && s.charAt(0) == '0') s.delete(0, 1);
return s.toString();
}
}
优化:用单调(递增)栈维护,栈中的元素代表截止到当前位置,删除不超过 k 个数字后,所能得到的最小整数
操作:
- 对于每个数字,如果该数字小于栈顶元素,我们就不断地弹出(删除)栈顶元素,直到①栈为空②新的栈顶元素不大于当前数字③我们已经删除了 k位数字。
- 如果该数字大于栈顶,表示此时仍是递增,入栈即可
- 此时栈底到栈顶是单调递增的,如果弹出了不到k个元素,那再从末尾删除几个,最后记得删除最前面的0,如果没剩下的数了就返回0
class Solution:
def removeKdigits(self, num: str, k: int) -> str:
stack=[]
for n in num:
while stack and k and n<stack[-1]:#维护栈内递增(不减)
stack.pop()
k-=1
stack.append(n)
if k>0:#删除的不够k个
stack=stack[:-k]#留下从0到倒数第k个数
return ''.join(stack).lstrip('0') or "0"#删除前置0
#若 "".join(finalStack).lstrip('0') 这一句结果有值(真)
#则返回该结果
#若计算结果为空字符 '',为假,返回 or 后面的结果 '0'
class Solution {
public String removeKdigits(String num, int k) {
if(num.length()==k) return "0";
StringBuilder stack=new StringBuilder();
for(char c:num.toCharArray()){
while(stack.length()!=0 && k!=0 && c<stack.charAt(stack.length()-1)){
stack.delete(stack.length()-1,stack.length());
k--;
}
stack.append(c);
}
if(k!=0){
stack.delete(stack.length()-k,stack.length());
}
while (stack.length() > 1 && stack.charAt(0) == '0') stack.delete(0, 1);
return stack.toString();
}
}