这题算是贪心思想的一个经典运用,也是所有学贪心的OIer们的必经之题(学贪心都1年了,复现还WA了4发。。。)思维难度一般,但是比较复杂,需要考虑得非常周到才可以。
贪心思路:很容易想到遍历字符串,针对每个数与其前面的数比较,前面的要是大的话就把前面的踢掉,由于可能会“连环踢",所以i–。结果你踌躇满志的交了一发,结果WA了。hack一组数据:
input: 178543 4
output: 13
结果你输出的是17!!!
我们模拟一遍代码的过程:
i的值 | 操作 | 字符串 |
---|---|---|
i==1 | 1<=7,不管 | 178543 |
i==2 | 7<=8,不管 | 178543 |
i==3,i– | 8>5,8踢掉 | 17543 |
i==3 | 5>4,5踢掉 | 1743 |
唉?明明5比7小,结果7还一直霸者位子呢?因为有8这个数挡在中间,使7在可以被踢掉的时候没被踢掉,等5跑到7后面时,i已经跑到3去了,鞭长莫及了。所以,我们可以尝试把if改成while,这样,就可以在5踢掉8的之后顺便把7踢掉了。
代码如下:
#include <iostream>
using namespace std;
string s;
int k, cnt;
int main() {
cin >> s >> k;
for(int i=1; i<s.length(); i++) {
while(!s.empty() && s[i-1] > s[i] && cnt < k) {
cnt ++;
s.erase(i-1, 1);
i --;
}
}
if(cnt < k) {
s.erase(s.length()-k+cnt, k-cnt);
}
while(!s.empty() && s[0] == '0') {
s.erase(0, 1);
}
if(s.empty()) cout << 0; // 不得不说真坑
else cout << s;
return 0;
}