Leetcode刷题笔记 402. 移掉K位数字

时间：2020年11月15日
知识点：贪心、模拟
题目链接：https://leetcode-cn.com/problems/remove-k-digits/

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

注意:

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

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

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

示例 3 :
输入:
num = “10”, k = 2
输出:
“0”
解释:
从原数字移除所有的数字，剩余为空就是0。

解法

1. 我们先来看如何删除一个数字
2. 当num=537 我们会删除5 当num=261 我们删除6 可以看到删除的是第一个左边比右边大的
3. 当num=234 我们删除最后一个
4. 这里我们采用往里放的思路 放入的数字小于数组中的末尾时 数组末尾的数字弹出 循环执行 注意不要删除太多了
5. 注意判断一开始是0的情况 和全0的情况

代码

#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
    string removeKdigits(string num, int k) {
        int count_k = 0;
        vector<int> ans;
        string str="";
        for(char x:num){
            while(ans.size() && ans.back() > x && count_k < k){
                ans.pop_back();
                count_k++;
            }
            ans.push_back(x);
        }
        while(k-count_k){
            ans.pop_back();
            count_k++;
        }
        int flag = 1;
        for(int x:ans){
            if (flag && x == '0')
                continue;
            flag = false;
            str += x;
        }
        if(str=="")
            str = '0';
        return str;
    }
};
int main()
{
    string num{"10"};
    int k = 2;
    Solution s;
    cout<<s.removeKdigits(num, k)<<endl;
    return 0;
}

今天也是爱zz的一天哦！