问题描述:给定一个以字符串表示的非负整数num和一个整数k,移除这个数中的k位数字,使得剩下的数字最小,请以字符串形式返回这个最小的数字。
示例1:
示例2:
解题思路:
假设给定一个n位的数字(n>1),要求移除k位数字(1<k<n)。我们考虑先移除一位数字的情况。
有两种移除方式,要么移除①位置数字,要么移除①位置之后的数字。
如果我们移除①位置的数字,那么剩下的字符串就是以②位置的数字为最高位的数字。
如果移除①位置之后的数字,那么剩下的字符串就是以①位置的数字为最高位的数字。
这样我们判断剩下字符串最高位置的数字大小,很明显最高位小的整体最小。因此我们需要移除一位数字之后,剩余字符串最高位要尽可能最小。
基于这种想法,可以比较当前数字与它的前一位数字的大小,如果前一位数字更大,那么就删除前一位数字,否则保留当前数字,进入下一次迭代。
这是一种贪心的思想。每一次我们都取当前情况下的最优解。
那么,这样每次取当前情况下的最优解,最后能够得到最终的最优解吗?
我们假设两种情况:
1.存在最小的数字,它是以上某个贪心操作中当前数字小于之前的数字(如xx62xxx),却没有删除它之前的数字(没有删除6)得到的,最后结果是(xx6xxx)。但是这个数字一定是小于贪心算法得到的(xx2xxx),这与之前假设的(xx6xxx)最小相悖,因此经过贪心操作得到的数字最小。
2.存在最小的数字,它是以上某个贪心操作中当前数字大于等于之前的数字(如xx267xx),却删除了当前数字(删除了6)得到的,最后结果是(xx27xx),贪心算法是保留了当前数字,得到(xx26xx),这与之前假设的(xx27xx)最小相悖,因此经过贪心操作得到的数字最小。
当然还存在另一种可能(xx261xx),这时删除了当前数字,得到(xx21xx)。贪心算法保留了当前数字6,但是进入下一轮迭代,当前数字为1,之前数字为6,当前数字小于之前数字,应该删除之前数字,得到(xx21xx)。即该假设与贪心算法相同。
综上所述,该问题经过贪心算法是能够求得全局最优解的。
在实际实现中,我们可以使用栈来存入保存的数字,因此,栈顶元素就可以视为当前数字之前的数字,如果是舍弃的数字,则不如栈。
代码如下:
class Solution {
public:
string removeKdigits(string num, int k) {
vector<char> stk;
for(auto & digit:num){
while(stk.size()>0&&stk.back()>digit&&k){
stk.pop_back();
k-=1;
}
stk.push_back(digit);
}
for(;k>0;k--){
stk.pop_back();
}
string ans="";
bool isLeadingZero=true;
for(auto&digit:stk){
if(isLeadingZero&&digit=='0'){
continue;
}
isLeadingZero=false;
ans+=digit;
}
return ans==""?"0":ans;
}
};