【LeetCode】402. 移掉K位数字

题目链接：https://leetcode-cn.com/problems/remove-k-digits/description/

题目描述

给定一个以字符串表示的非负整数 num，移除这个数中的 k 位数字，使得剩下的数字最小。

注意:

• num 的长度小于 10002 且 ≥ k。
• num 不会包含任何前导零。

示例

输入: num = “1432219”, k = 3
输出: “1219”
解释: 移除掉三个数字 4, 3, 和 2 形成一个新的最小的数字 1219。

输入: num = “10200”, k = 1
输出: “200”
解释: 移掉首位的 1 剩下的数字为 200. 注意输出不能有任何前导零。

解决方法

贪心算法。贪心策略：每次从第0个元素起，移走第一个递减序对[pos,pos+1](num[pos]>num[pos+1])中的pos这个元素

class Solution {
public:
    string removeKdigits(string num, int k) {
        //贪心算法。贪心策略：每次从第0个元素起，移走第一个递减序对[pos,pos+1](num[pos]>num[pos+1])中的pos这个元素
        if (k==0) return num; 
        if (num.size()==k) return "0";
        while(k--){
            bool flag=false;
            for (int i=0;i<num.size();i++){ //找到第一个递减序对
                if (num[i]>num[i+1]){
                    num.erase(i,1);
                    flag=true;
                    break;
                }
            }
            if (!flag) num.erase(num.size()-1,1);//没有递减序对时，删除最后一个元素
        }
        //处理前导零
        int len=0;bool fa=false;
        for (int i=0;i<num.size();i++){
            if (num[i]!='0'){
                len=i;
                fa=true;
                break;
            }
        }
        if (!fa) return "0";//结果全为0，则输出一个0就好
        num.erase(0,len);
        return num;
    }
};