题意:
给定一个序列A,(ai:【0,255】),可以组合其中连续的一些数,这些数个数不能超过m;这些数可以以最小的数作为key
问这个序列A对应的字典序最小的 key 序列是啥;
思路:
贪心的想法,从左到右,
对于当前数a[i],如果已经合入某个组,那直接输出其key;
否则:为了字典序最小:我们找到它前面(包括它)的第m个数,从这个数往后找是否有一个数还没加入某个组,或者加入的某个组能够包含a[i],那把a[i]和这个数之间的数加入新的组;
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 7;
typedef long long ll;
int n, m;
vector<int> vec;
map<int, int> mp;
set<int> st;
int id[277], a[maxn], ans[maxn];
bool vis[maxn];
int main() {
for(int i = 0; i <= 255; ++i) id[i] = 0;
memset(vis, 0, sizeof vis);
scanf("%d%d", &n, &m);
for(int i = 0; i < n; ++i) {
scanf("%d", &a[i]);
int t = max(0, a[i]-m+1);
if(vis[a[i]]) {
ans[i] = id[a[i]];
} else if(!vis[t]) {
ans[i] = t;
for(int j = t; j <= a[i]; ++j) {
vis[j] = 1;
id[j] = t;
}
} else {
int j = t;
for( ; j <= a[i]; ++j) {
if(!vis[j]) break;
if(a[i]-id[j]+1 <= m) break;
}
ans[i] = j;
for(int k = j; k <= a[i]; ++k) {
id[k] = j;
vis[k] = 1;
}
}
}
printf("%d", ans[0]);
for(int i = 1; i < n; ++i) printf(" %d", ans[i]);
return 0;
}