LintCode ：删除数字

LintCode ： 删除数字

题目

给出一个字符串 A, 表示一个 n 位正整数, 删除其中 k 位数字, 使得剩余的数字仍然按照原来的顺序排列产生一个新的正整数。
找到删除 k 个数字之后的最小正整数。
N <= 240, k <= N

样例

给出一个字符串代表的正整数 A 和一个整数 k, 其中 A = 178542, k = 4

返回一个字符串 “12”

思路

为了得到更小的数字我们需要把小数字放在前面，那么我们就要把在当前数字前并且比当前数字大的数删掉，所以我们从头开始遍历，找到比当前数大的就删除，直到删除k个数为止。

代码

string DeleteDigits(string A, int k) {
    if(k > A.size())
        return "0";
    int _k = 0;
    for(string::iterator it = A.begin(); it < A.end() && _k < k; it++) {
        string::iterator back;
        char x = *it;
        if(it != A.begin()) {
            for(back = it - 1; back >= A.begin() && *back > x && _k < k; back--) {
                _k++;
            }
            back++;
            while(*back != x)
                A.erase(back);
            it = back;
        }
    }
    while(_k < k){
        A.erase(A.end() - 1);
        _k++;
    }
    while(*(A.begin()) == '0')
        A.erase(A.begin());
    if(A == "")
        return "0";
    return A;
}