洛谷P3041 视频游戏的连击Video Game Combos [USACO12JAN] AC自动机+dp

正解:AC自动机+dp

解题报告:

传送门!

算是个比较套路的AC自动机+dp趴,,,

显然就普普通通地设状态,普普通通地转移,大概就f[i][j]:长度为i匹配到j

唯一注意的是,要加上所有子串的贡献,就在结构体中新加一个变量d表示在跟到这个节点的串以及后缀中完整串的个数

直接在bfs求fail的时候加上fail指针的d就欧克

啊当然只是为了方便描述所以说另开一个d,,,实际实现的话直接把标记结尾的那个值改成int类型搞下就欧克

然后说下细节,,,

其实就一个细节,就赋初值问题,要把除了f[i][0]的都设成-inf

原因挺显然的趴?拿样例为例好了,如果不这么设就会搞得明明只选了一个A就被计入了ABA一个串的贡献这样的乌龙QwQ

overr

#include<bits/stdc++.h>
using namespace std;
#define il inline
#define gc getchar()
#define ll long long
#define ri register int
#define rb register bool
#define rc register char
#define rp(i,x,y) for(ri i=x;i<=y;++i)
#define my(i,x,y) for(ri i=x;i>=y;--i)

const int N=+,inf=1e9;
int n,K,f[N][N*],nod_cnt,as;
struct nod{int to[],fail,flg;}tr[N*];
char ch[N];
queue<int>Q;

il int read()
{
    rc ch=gc;ri x=;rb y=;
    while(ch!='-' && (ch>'' || ch<''))ch=gc;
    if(ch=='-')ch=gc,y=;
    while(ch>='' && ch<='')x=(x<<)+(x<<)+(ch^''),ch=gc;
    return y?x:x;
}
il void insert(char *s)
{
    ri lth=strlen(s+),nw=;
    rp(i,,lth)
    {
        if(!tr[nw].to[s[i]-'A'+])tr[nw].to[s[i]-'A'+]=++nod_cnt;
        nw=tr[nw].to[s[i]-'A'+];
    }
//    printf("nw=%d\n",nw);
    tr[nw].flg=;
}
il void bfs()
{
    rp(i,,)if(tr[].to[i])Q.push(tr[].to[i]);
    while(!Q.empty())
    {
        ri nw=Q.front();Q.pop();tr[nw].flg+=tr[tr[nw].fail].flg;//printf("nw=%d flg=%d\n",nw,tr[nw].flg);
        rp(i,,)
            if(tr[nw].to[i])Q.push(tr[nw].to[i]),tr[tr[nw].to[i]].fail=tr[tr[nw].fail].to[i];
            else tr[nw].to[i]=tr[tr[nw].fail].to[i];
    }
}

int main()
{
//    freopen("3041.in","r",stdin);freopen("3041.out","w",stdout);
    n=read();K=read();while(n--){scanf("%s",ch+);insert(ch);}bfs();
    rp(i,,K)rp(j,,nod_cnt)f[i][j]=-inf;
    rp(i,,K)
    {
        rp(j,,nod_cnt)
        {
//            printf("f[%d][%d]=%d\n",i,j,f[i][j]);
            rp(k,,)f[i][tr[j].to[k]]=max(f[i][tr[j].to[k]],f[i-][j]+tr[tr[j].to[k]].flg);
        }
    }
    rp(i,,nod_cnt)as=max(as,f[K][i]);
    printf("%d\n",as);
    return ;
}

这儿是代码鸭QwQ