JZOJ 5167 【NOIP2017模拟6.26】下蛋爷

题目大意：

有n个单词，有一个长度为m的文章，现在你先需要知道每个单词在文章内出现的次数。
现在一共进行k轮，每一轮有p的概率保留出现次数最小的那些单词（不保留就是把那些单词删掉）。
求n轮以后，每个单词保留下来的概率.

1<=n<=200,1<=单词长度<=20,1<=m<=10^6

题解：

分两个部分。
第一部分就直接上ac自动机，可以打up来优化。
ac自动机up优化：
up[x]表示x和x的fail链上的第一个是单词结尾的点。
这样查询的时候就会很快（贴瓷砖也有用的上）。
注意单词可能会有重复，单词结尾需要用前向星维护。

第二部分就是概率dp.
设f[i][j]表示第i轮,出现次数第j大的单词们活下来的概率（出现次数相同的算同一个）
f[i][j] = f[i - 1][j - 1] +f[i - 1][j] * p

Code：

#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#define fo(i, x, y) for(int i = x; i <= y; i ++)
using namespace std;

const int Maxn = 205, Maxm = 1000005;

double p;
char a[Maxn][25], s[Maxm];
int n, ls, cc, ans[205], b[205], d[10000], next[10000][27], fail[10000], up[10000], tt;
int final[10000], tot;
struct edge {
    int to, next;
}e[205];
double f[1005][205], sum[1005];

void link(int x, int y) {
    e[++ tot].next = final[x], e[tot].to = y, final[x] = tot;   
}

void insert(int p, char *a) {
    int len = strlen(a + 1), k = 1;
    fo(i, 1, len) {
        if(!next[k][a[i] - 'a']) next[k][a[i] - 'a'] = ++ tt;
        k = next[k][a[i] - 'a'];
    }
    link(k, p);
}

void make_ac() {
    int l = 0, r = 0; d[++ r] = 1;
    while(l < r) {
        l ++; int x = d[l];
        fo(i, 0, 25) {
            int y = next[x][i];
            if(!y) continue;
            int j = fail[x];
            while(j && !next[j][i]) j = fail[j];
            if(next[j][i]) j = next[j][i];
            fail[y] = j; if(!fail[y]) fail[y] = 1;
            up[y] = y; if(!final[y] && y != 1) up[y] = up[fail[y]];
            d[++ r] = y;
        }
    }
}

void Init() {
    tt = 1;
    scanf("%d", &n);
    fo(i, 1, n) {
        scanf("%s", a[i] + 1);
        insert(i, a[i]);
    }
    make_ac();
    scanf("%s", s + 1);
    ls = strlen(s + 1);
}

void Work() {
    int j = 1;
    fo(i, 1, ls) {
        while(j != 1 && !next[j][s[i] - 'a']) j = fail[j];
        if(next[j][s[i] - 'a']) j = next[j][s[i] - 'a'];
        int z = up[j];
        while(z) {
            for(int k = final[up[z]]; k; k = e[k].next)
                ans[e[k].to] ++;
            z = up[fail[z]];
        }
    }
}

bool rank_b(int x, int y) {
    return ans[x] < ans[y]; 
}

void End() {
    scanf("%lf", &p); scanf("%d", &cc);
    fo(i, 1, n) b[i] = i;
    sort(b + 1, b + n + 1, rank_b);
    fo(j, 2, n) f[1][j] = 1;
    f[1][1] = p;
    fo(i, 2, cc) {
        int k = 0;
        fo(j, 1, n) {
            if(j == 1 || ans[b[j]] != ans[b[j - 1]]) {
                k ++;
                f[i][k] = f[i - 1][k - 1] + (f[i - 1][k] - f[i - 1][k - 1]) * p;
            }
        }
    }
    int k = 0;
    fo(i, 1, n) {
        if(i != 1 && ans[b[i]] == ans[b[i - 1]])
            sum[b[i]] = sum[b[i - 1]]; else k ++, sum[b[i]] = f[cc][k];
    }
    fo(i, 1, n) printf("%.3lf ", sum[i]);
}

int main() {
    Init();
    Work();
    End();
}