题目描述
给定一个十进制的正整数number,选择从里面去掉一部分数字,希望保留下来的数字组成的正整数最大。
输入描述:
输入为两行内容,第一行是正整数number,1 ≤ length(number) ≤ 50000。第二行是希望去掉的数字数量cnt 1 ≤ cnt < length(number)。
输出描述:
输出保留下来的结果。
示例1
输入
325 1
输出
35
题解
方法1. 因为想要最后剩下的数尽量大,所以贪心地从前往后找到某位数比后一位小就删掉这个数,但是这样需要 O(n*m) (n 是总位数,m 是删除的个数)。我们可以利用一个栈来达到 O(n)的时间复杂度:遍历每一位,当还能删除时且栈内的数比当前数小就出栈,直到栈内的数比当前数大,或者栈空,就将当前的数入栈。如果全部数都入过栈时还需要删除,那就从栈顶删。
sta = []
num = '0123456789'
s = input()
n = m = int(input())
for i in s:
while len(sta) != 0 and num.index(sta[-1]) < num.index(i) and m > 0:
m -= 1
sta.pop()
sta.append(i)
print (''.join(sta[:(len(s) - n)]))
方法2. 利用10个队列记录0~9出现的位置,例如9843648,那么4出现的位置就是2,5,8出现的位置就是1,6。然后贪心地从大到小枚举每一位可以放的数字,
#include <bits/stdc++.h>
using namespace std;
queue <int> q[10];
char s[50010];
int n, m, len;
int main() {
ios::sync_with_stdio(false);
cin >> s >> m;
len = strlen(s), n = len - m;
for (int i = 0; i < len; i ++)
q[s[i] - '0'].push(i);
for (int c, k = 0, j = -1; k < n; k ++) {
for (int i = 9; ~i; i --) {
while (!q[i].empty() && q[i].front() <= j) q[i].pop();
if (!q[i].empty() && m >= q[i].front() - j - 1) {
c = '0' + i;
m -= q[i].front() - j - 1;
j = q[i].front();
break;
}
}
putchar(c);
}
return 0;
}