402. 移掉 K 位数字 c++

题目描述

给你一个以字符串表示的非负整数 num 和一个整数 k ，移除这个数中的 k 位数字，使得剩下的数字最小。请你以字符串形式返回这个最小的数字。

示例 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 <= k <= num.length <= 105
num 仅由若干位数字（0 - 9）组成
除了 0 本身之外，num 不含任何前导零

解法

class Solution
{
public:
    string removeKdigits(string num, int k)
    {
        int n = num.size();
        if (n <= k)
        {
            return "0";
        }
        string res;
        int i = 0;
        int remain = n - k;
        while (i < n)
        {
            while (k > 0 && !res.empty() && res.back() > num[i])
            {
                res.pop_back();
                k--;
            }
            res.push_back(num[i]);
            i++;
        }
        while (k > 0)
        {
            res.pop_back();
            k--;
        }
        i = 0;
        while (i < res.size())
        {
            if (res[i] != '0')
            {
                break;
            }
            i++;
        }
        if (i == res.size())
        {
            return "0";
        }
        res = res.substr(i);
        return res;
    }
};