Problem P21. [算法课贪婪]移掉 K 位数字

image

用单调栈的想法,最先进栈的肯定比最后进栈的小。
贪心的话就是确保第一个数是最小的就可以确保整个数是最小的。比如: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;
}

posted @ 2022-09-14 14:01  白缺  阅读(95)  评论(0编辑  收藏  举报