题意:
给定一个序列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; }