【Lintcode】693. Remove Dights

题目地址：

https://www.lintcode.com/problem/remove-dights/description

给定一个字符串表示的非负整数$s$，再给定一个正整数$k$，现在要求在$s$中删去$k$位，使得剩下来的数达到最小。返回这个最小值（返回字符串即可，要求不能有开头$0$）。如果$k$大于等于字符串的长度则规定答案是$0$。题目保证给定的数没有开头$0$（除非本身就是$0$）。

思路是单调栈。先考虑如果只允许删除一个数字，那应该怎样做。设原数是$a_n{a}_{n-1}...a_0$，如果$a_n\le a_{n-1}\le ...\le a_0$，那么我们可以证明要删掉的数就是$a_0$，所得数就是$s^{\prime }=\overline{a_n{a}_{n-1}...a_1}$，不然的话，设删除的数是$a_i$，其中$i\ne 0$，那么得数$s^{\prime \prime }=\overline{a_n...a_{i+1}{a}_{i-1}...a_0}\ge s^{\prime }=\overline{a_n{a}_{n-1}...a_{i+1}{a}_i...a_1}$，所以$s^{\prime }$就是最优解；否则，设 k = min ⁡ { k : a k > a k + 1 } k=\min\{k:a_k>a_{k+1}\} k=min{k:ak​>ak+1​}，则用相同的证明方法可以证明要删除的数就是$a_k$。

接下来考虑删除多个数字的情况。首先由上面只删除一个数字的方法，我们得到一个方案，即先从左到右遍历这个字符串，每次一遇到形如$a_j>a_{j+1}$的情况就删除$a_j$（如果没有这样的pair，就直接删掉尾端的$k$个数）。接下来我们证明这种方法能得到最优解。可以用反证法。假设用上面的方案删除的数字分别是$a_{i_1},a_{i_2},...,a_{i_k}$所得新数为$s^{\prime }$，还有一个更好方案，删除的数字是$a_{j_1},a_{j_2},...,a_{j_k}$所得新数为$s^{\prime \prime }$。取最小的$r$使得$i_r\ne j_r$，如果$i_r<j_r$，那么一定有$a_{i_r}>a_{i_r+1}$，则$s^{\prime }<s^{\prime \prime }$矛盾；如果$i_r>j_r$，则一定有$a_{j_r}\le a_{j_r+1}$，所以$s^{\prime }\le s^{\prime \prime }$矛盾。综上所述按照上面的方案得到的就是最优解。

具体代码，其实就是一边从左到右遍历字符串，一边维护一个字典序单调上升的栈。代码如下：

public class Solution {
    /**
     * @param num: a number
     * @param k: the k digits
     * @return: the smallest number
     */
    public String removeKdigits(String num, int k) {
        // write your code here.
        // 要把num删光的话，则直接返回0
        if (k >= num.length()) {
            return "0";
        }
        
        // 直接用一个StringBuilder来作为栈使用
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < num.length(); i++) {
            char c = num.charAt(i);
            // 只要遇到了比栈顶小的字符，就把栈顶pop出来，同时记还可以删除的数字个数减1
            while (sb.length() > 0 && sb.charAt(sb.length() - 1) > c && k > 0) {
                sb.setLength(sb.length() - 1);
                k--;
            }
            
            sb.append(c);
        }
        // 此时sb已经单调了，如果还需要删除多于其长度个数的数字，则直接返回0
        if (k >= sb.length()) {
            return "0";
        }
        
        // 否则先把末尾的k个字符切掉，然后去掉开头0
        sb.setLength(sb.length() - k);
        int idx = 0;
        while (idx < sb.length() && sb.charAt(idx) == '0') {
            idx++;
        }
        
        return idx == sb.length() ? "0" : sb.substring(idx);
    }
}

时空复杂度$O(n)$。