CF 666C & 牛客 36D
给定字符串S, 求有多少长为$n$的字符串T, 使得S为T的子序列.
可以得到转移矩阵为
$\begin{equation}
A
=\begin{bmatrix}
25 & 0 & 0 &\cdots\ &0 &0\\
1 & 25 & 0 &\cdots\ &0 &0\\
0 & 1 & 25 & \cdots\ & 0 & 0\\
\vdots & \vdots & \vdots & \ddots & \vdots &\vdots\\
0 & 0 & 0 &\cdots\ &1& 26\\
\end{bmatrix}
\end{equation}$
设$|S|=m$, 答案就为$A^n$的$(m,0)$项, 也就是说答案只与$S$的长度有关, 但是用矩阵幂的话复杂度是$O(m^3logn)$显然过不去.
实际上我们可以直接设$f(n)$为长为$n$的字符串的答案, 不去维护匹配的状态.
可以得到$f(n) = \begin{cases} 0, & n< m \\26f(n-1)+25^{n-m}\binom{n-1}{m-1}, & n\ge m \end{cases}$
前一部分表示在前$n-1$位已经有子序列等于$S$的情形, 那么第$n$位可以任取值.
后一部分表示在第$n$位时第一次出现子序列等于$S$的情形, 那么枚举前$m-1$位在$T$中的第一次出现位置, 其余位置可以任取其余的$25$个值.
#include <iostream>
#include <sstream>
#include <algorithm>
#include <cstdio>
#include <math.h>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <string.h>
#include <bitset>
#define REP(i,a,n) for(int i=a;i<=n;++i)
#define PER(i,a,n) for(int i=n;i>=a;--i)
#define hr putchar(10)
#define pb push_back
#define lc (o<<1)
#define rc (lc|1)
#define mid ((l+r)>>1)
#define ls lc,l,mid
#define rs rc,mid+1,r
#define x first
#define y second
#define io std::ios::sync_with_stdio(false)
#define endl '\n'
#define DB(a) ({REP(__i,1,n) cout<<a[__i]<<' ';hr;})
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int P = 1e9+7, P2 = 998244353, INF = 0x3f3f3f3f;
ll gcd(ll a,ll b) {return b?gcd(b,a%b):a;}
ll qpow(ll a,ll n) {ll r=1%P;for (a%=P;n;a=a*a%P,n>>=1)if(n&1)r=r*a%P;return r;}
ll inv(ll x){return x<=1?1:inv(P%x)*(P-P/x)%P;}
inline int rd() {int x=0;char p=getchar();while(p<'0'||p>'9')p=getchar();while(p>='0'&&p<='9')x=x*10+p-'0',p=getchar();return x;}
//head
const int N = 1e5+10;
int t, clk;
char s[N];
vector<pii> g[N];
int ans[N], f[N];
ll p25[N], fac[N], ifac[N];
ll C(int n, int m) {
return fac[n]*ifac[n-m]%P*ifac[m]%P;
}
int main() {
p25[0] = fac[0] = ifac[0] = 1;
REP(i,1,N-1) {
p25[i] = p25[i-1]*25%P;
fac[i] = fac[i-1]*i%P;
ifac[i] = inv(fac[i]);
}
scanf("%d%s", &t, s);
int now = strlen(s);
REP(i,1,t) {
int op, x;
scanf("%d", &op);
if (op==1) scanf("%s", s), now = strlen(s);
else scanf("%d", &x), g[now].pb(pii(x,++clk));
}
REP(i,1,N-1) if (g[i].size()) {
sort(g[i].begin(),g[i].end());
int sz = g[i].size(), now = 0;
while (now<sz&&g[i][now].x<i) ++now;
f[i-1] = 0;
REP(j,i,N-1) {
f[j] = (f[j-1]*26ll+C(j-1,i-1)*p25[j-i])%P;
while (now<sz&&g[i][now].x==j) ans[g[i][now++].y] = f[j];
if (now>=sz) break;
}
}
REP(i,1,clk) printf("%d\n",ans[i]);
}

浙公网安备 33010602011771号