Problem P21. [算法课贪婪]移掉 K 位数字
用单调栈的想法,最先进栈的肯定比最后进栈的小。
贪心的话就是确保第一个数是最小的就可以确保整个数是最小的。比如:200>100
最后做一个出栈处理,保证数值最后输出是200,不是0200,还有就是如果栈里面什么都没有即所有位的数值都被消去的话,要输出0。
#include<iostream>
#include<bits/stdc++.h>
#include<cstdio>
using namespace std;
int main()
{
string str;
stack<char> sta;
cin >> str;
int k;
cin >> k;
int l = str.length();
while(k>0&&l>0){
if(sta.size()>0&&str[str.length()-l]<sta.top()){
sta.pop();
k--;
}else {
sta.push(str[str.length()-l]);
l--;
}
}
while(k>0){
sta.pop();
k--;
}
while(l>0){
sta.push(str[str.length()-l]);
l--;
}
string s;
while(sta.size()>0){
s=sta.top()+s;
sta.pop();
}
bool flag = true;
for (int i = 0; i < s.length(); i++){
if (flag&&s[i]=='0'){
continue;
}
flag = false;
printf("%c", s[i]);
}
if (flag == true){
printf("0");
}
return 0;
}