解析:假设当前已经放了i个字符,c[p]为当前以p结尾的字符串的个数,sum = c[0] + .... c[25]。
在放入第i+1个字符后,应该使sum尽可能大。
假设放入的是字符t,那么新的c[t]的值为sum+1(所有串+自己),sum的变化量为1+sum-c[t]。
找到最大的变化量,进行更新即可。但是这里有取模的操作,没法求出最大的变化量。
所以,贪心的思考,使得变化量最大的那个字符必然是最后一次出现该字符最早的那个,开个优先队列查找就好。
[code]:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
typedef long long LL;
const LL MOD = 1e9+7;
const int maxn = 1e6+5;
int n,m,k,b[26];
char s[maxn];
LL c[26],sum,d[26];
struct Comp{
bool operator ()(const int &i,const int &j) const{
return b[i] > b[j];
}
};
priority_queue<int,vector<int>,Comp> Q;
int main(){
int i,j,p;
scanf("%d%d%s",&n,&k,s+1);
m = strlen(s+1);
for(i = 0;i < k;i++) b[i] = 0;
for(i = 1;i <= m;i++){
p = s[i]-'a';
b[p] = i;
d[p] = (1+sum-c[p])%MOD;
c[p] = (1+sum)%MOD;
sum = (sum+d[p])%MOD;
}
for(i = 0;i < k;i++) Q.push(i);
for(i = 1;i <= n;i++){
p = Q.top();Q.pop();
b[p] = m+i;Q.push(p);
d[p] = (1+sum-c[p])%MOD;
c[p] = (1+sum)%MOD;
sum = (sum+d[p])%MOD;
}
sum = (sum+1)%MOD;
sum = (sum+MOD)%MOD;
printf("%I64d\n",sum);
return 0;
}
更多推荐
Codeforces 645E Intellectual Inquiry (贪心+DP)
发布评论