牛客2020年七夕节比赛 F、牛妹的考验 （AC自动机）

题目链接

题面：

题意：
给定一些字符串的权值（可能有重复的字符串，重复计算即可）。
要求构造一个长度为 L 的字符串，给定的字符串在这个长度为L的字符串中每出现一次就会得到一次相应的权值。考虑包含的情况，即串 abc 既包含 abc 也包含 ab 。

题解：
AC自动机。在处理fail指针的时候考虑包含串的情况。
$dp[x][len]$ 到AC自动机上节点x且此时串的长度为len所能获得的最大权值。

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<string>
#include<queue>
#include<bitset>
#include<map>
#include<unordered_map>
#include<unordered_set>
#include<set>
#define ui unsigned int
#define ll long long
#define llu unsigned ll
#define ld long double
#define pr make_pair
#define pb push_back
#define lc (cnt<<1)
#define rc (cnt<<1|1)
#define len(x)  (t[(x)].r-t[(x)].l+1)
#define tmid ((l+r)>>1)
#define fhead(x) for(int i=head[(x)];i;i=nt[i])
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)>(y)?(y):(x))
using namespace std;

const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const double dnf=1e18;
const int mod=1e9+7;
const double eps=1e-8;
const double pi=acos(-1.0);
const int hp=13331;
const int maxn=2100;
const int maxm=100100;
const int maxp=100100;
const int up=30;

int t[maxn][26],sum[maxn],fail[maxn],cnt=0;
char str[maxn];
int dp[maxn][1100];//dp[x][len] 到AC自动机上节点x且此时串的长度为len所能获得的最大权值。

void _insert(int val)
{
    int p=0;
    int k;
    int len=strlen(str);
    for(int i=0;i<len;i++)
    {
        k=str[i]-'a';
        if(!t[p][k]) t[p][k]=++cnt;
        p=t[p][k];
    }
    sum[p]+=val;

}

void getfail(void)
{
    queue<int>q;
    for(int i=0;i<26;i++)
    {
        if(t[0][i])
        {
            fail[t[0][i]]=0;
            q.push(t[0][i]);
        }
    }
    while(q.size())
    {
        int now=q.front();
        q.pop();

        sum[now]+=sum[fail[now]];
        for(int i=0;i<26;i++)
        {
            int p=t[now][i];
            if(p)
            {
                fail[p]=t[fail[now]][i];
                q.push(p);
            }
            else
            {
                t[now][i]=t[fail[now]][i];
            }
        }
    }
}


int main(void)
{
    int n,l;
    scanf("%d%d",&n,&l);
    int val;
    for(int i=1;i<=n;i++)
    {
        scanf("%s%d",str,&val);
        _insert(val);
    }
    getfail();
    for(int i=0;i<=cnt;i++)
    {
        for(int j=0;j<=l;j++)
            dp[i][j]=-inf;
    }
    dp[0][0]=0;
    int maxx=-inf;
    for(int len=0;len<l;len++)
    {
        for(int x=0;x<=cnt;x++)
        {
            for(int i=0;i<26;i++)
            {
                int now=t[x][i];
                dp[now][len+1]=max(dp[now][len+1],dp[x][len]+sum[now]);
                if(len+1==l)
                    maxx=max(maxx,dp[now][len+1]);
            }
        }
    }
    printf("%d\n",maxx);
    return 0;

}