Codeforces #349 div1 C. Codeword 组合数学 递推

题目

题目链接：http://codeforces.com/problemset/problem/666/C

题目来源：Codeforces #349 div1

题解

首先需要看出方案数只和模式串的长度有关，和具体的字符无关。

由于模式串的长度和规模为$10^5$，则不同长度模式串最多有$\sqrt {10^5}$种，因而如果如果我们若能在线性时间内推出某长度下的结果，复杂就会变成$\Theta(n\sqrt n)$。

具体记数的过程比较复杂，需要考虑枚举模式串为第一次匹配的情况。

若$S$为主串，模式串为$P$，$a_i$为$|S|=i$时的方案数。

$a_i= \begin{cases} 0 & \text{ if } i < |P| \\ 26a_{i-1}+ {{n-1} \choose {i-1}}\cdot 25^{i-|P|} & \text{ if } i \geqslant |P| \end{cases}$

$26a_{i-1}$包含了第一次匹配在$S_1\cdots S_{i-1}$出现的情况，最后一个位置放啥都行。

${{n-1} \choose {i-1}}\cdot 25^{i-|P|}$ 是在$S_i$才出现第一次匹配的情况。$S_1\cdots S_{i-1}$能匹配$P_i\cdots P_{|P|-1}$且$S_i=P_{|P|}$。最后一位确定，然后在$S_1\cdots S_{i-1}$选择$P_i\cdots P_{|P|-1}$的匹配，即${{n-1} \choose {i-1}}$，剩余的位置每个位置都含有$25$种可能，因为不能是它之后第一个确定的位置的字符。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
// head
const int N = 1e5+5;
const int MOD = 1e9+7;

LL powmod(LL a, LL b) {
    LL ans = 1;
    for (; b; b >>= 1) {
        if (b & 1) ans = (a * ans) % MOD;
        a = a * a % MOD;
    }
    return ans;
}

int fac[N], inv[N], p25[N];

void init() {
    inv[0] = fac[0] = p25[0] = 1;
    for (int i = 1; i < N; i++) {
        fac[i] = (LL)fac[i-1] * i % MOD;
        inv[i] = powmod(fac[i], MOD-2);
        p25[i] = (LL)p25[i-1] * 25 % MOD;
    }
}

LL C(int n, int m) {
    return (LL)fac[n] * inv[m] % MOD * inv[n-m] % MOD;
}

int a[N], ans[N];
vector<PII> query[N];

void solve(int n) {
    a[n] = 1;
    for (int i = 0; i < n; i++) {
        a[i] = 0;
    }
    for (int i = n+1; i < N; i++) {
        a[i] = (LL)a[i-1] * 26 % MOD;
        a[i] = (a[i] + C(i-1, n-1) * p25[i-n]) % MOD;
    }
    for (auto q : query[n]) {
        ans[q.fi] = a[q.se];
    }
}

char s[N];

int main() {
    init();
    int n, cur, q, x, cnt = 0;
    scanf("%d", &n);
    scanf("%s", s);
    cur = strlen(s);
    for (int i = 0; i < n; i++) {
        scanf("%d", &q);
        if (q == 1) {
            scanf("%s", s);
            cur = strlen(s);
        } else {
            scanf("%d", &x);
            query[cur].push_back(make_pair(cnt++, x));
        }
    }
    for (int i = 0; i < N; i++) {
        if (query[i].size() == 0) continue;
        solve(i);
    }
    for (int i = 0; i < cnt; i++) {
        printf("%d\n", ans[i]);
    }
    return 0;
}