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;
}