从序列 N 个数中顺序选出 M 个不同的数, 使得这 M 个数的字典序最小。
链接:字典序最小
做法:从前往后遍历,如果栈空,将a[i]入栈,如果栈顶大于a[i],且栈顶在a[i]之后还出现过,那么栈顶就可以删除。
原本我也想到个差不多的做法,可惜方向错了,当时想的是,如果a[i]在后面还出现过,而且不是当前所缺的最小的数,那a[i]就不放入栈中。。这样是会漏解的,如果此时a[i]不放,后面可能出现更大的数字必须要放,字典序就变大了。
#include <bits/stdc++.h>
// #define int long long
using namespace std;
const int N = 1e6 + 5, mod = 998244353;
deque<int>q;
int a[N], t[N];
bool f[N];
signed main()
{
int n, m;
cin >> n >> m;
for(int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
t[a[i]]++;
}
q.push_back(0);
for (int i = 1; i <= n; i++) {
t[a[i]]--;
if (f[a[i]]) continue;
while(a[i] <= q.back() && t[q.back()]) {
f[q.back()] = 0;
q.pop_back();
}
q.push_back(a[i]);
f[a[i]] = 1;
}
q.pop_front();
cout << q.front();
q.pop_front();
while(!q.empty()) {
printf(" %d", q.front());
q.pop_front();
}
return 0;
}