原题地址:http://acm.hdu.edu.cn/showproblem.php?pid=6586
题意:给出一个字符串str和k,以及一些限制条件,要求从选择str的一个子序列,使得满足限制条件,长度为k,且字典序最小。
思路:可以先通过构造一个序列自动机,这样子可以方便知道某个位置后任意字符的位置。
然后我们贪心的选取每一位,判断当前可不可行即可。
判断可不可行就是需要满足选择这个字母之后,在这之后的每种字母的出现+已经出现的次数>必须出现的次数,并且满足出现次数<=最多次数,以及后面的位置数量要>还没有填的必须的数量。
#include <bits/stdc++.h>
#define eps 1e-8
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define lson l,mid,rt<<1
#define rson mid+1,r,(rt<<1)+1
#define CLR(x,y) memset((x),y,sizeof(x))
#define fuck(x) cerr << #x << "=" << x << endl
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int seed = 131;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
char str[maxn];
int nxt[maxn][30];//序列自动机
int suf[maxn][30];//记录后缀的数量
int l[30], r[30], k;
int use[30];
char ans[maxn];
int alp[30];//记录每个字母出现的次数
void init(int len) {
int now[30];
int num[30];
CLR(now, -1);
CLR(num, 0);
for (int i = len; i >= 0; i--) {
for (int j = 0; j < 26; j++) {
nxt[i][j] = now[j];
suf[i][j] = num[j];
}
num[str[i] - 'a']++;
now[str[i] - 'a'] = i;
}
}
int check(int x, int c, int cnt) { //在第x位判断加入字符c可不可行,上次在str的cnt位置
if (use[c] + 1 > r[c]) return 0; //超过最大范围
int need = 0;//必须字符
//尝试插入
use[c]++;
int id = nxt[cnt][c];//插入id位置的c字符
if (id == -1) return 0;
for (int i = 0; i < 26; i++) {
need += max(0, l[i] - use[i]);
}
if (need > k - x) {//当前必须的多于剩下的位置
use[c]--;
return 0;
}
for (int i = 0; i < 26; i++) {
if (use[i] + suf[id][i] < l[i]) {//当前选择的+后面有的<至少的
use[c]--;
return 0;
}
}
return 1;
}
int main() {
while (~scanf("%s%d", str + 1, &k)) {
CLR(alp, 0);
CLR(suf, 0);
CLR(use, 0);
CLR(nxt, 0);
for (int i = 0; i < 26; i++) {
scanf("%d%d", &l[i], &r[i]);
}
int len = strlen(str + 1);
for (int i = 1; i <= len; i++) {
alp[str[i] - 'a']++;
}
init(len);
int flag = 1;
int cnt = 0;//上一次的位置
for (int i = 1; i <= k; i++) {//枚举k位
int can = 0;
for (int j = 0; j < 26; j++) {//枚举是否在第i位插入字符j
if (check(i, j, cnt)) {
cnt = nxt[cnt][j];
ans[i] = j + 'a';
can = 1;
break;
}
}
if (!can) {
flag = 0;
break;
}
}
if (!flag) printf("-1\n");
else {
ans[k + 1] = 0;
printf("%s\n", ans + 1);
}
}
return 0;
}