uva1502 - GRE Words

链接

https://vjudge.net/problem/UVA-1502

题解

这题数据好像是出了点锅(SPOJ上这么说的)，所以用除了scanf之外的读入方式可能会造成莫名其妙的$TLE$或者$WA$
网上有很多用$AC$自动机做的大佬，没有仔细研究过他们的做法，我的做法是用后缀数组
我可以倒着做$dp$
我只需要看下当前串是哪些串的子串，然后用哪些串的$dp$值去算出当前串的$dp$值
以当前串为子串的串就是后缀数组中连续的一段，直接往上往下分别二分就可以确定范围，然后直接用线段树查询最小值即可
算出来之后，当前串的所有后缀的$dp$值会发生改变，用线段树修改下

代码

#include <bits/stdc++.h>
#define maxn 400000
#define maxk 19
#define iinf 0x3f3f3f3f
#define cl(x) memset(x,0,sizeof(x))
using namespace std;
struct SuffixArray
{
    int sa[maxn], rank[maxn], ws[maxn], wv[maxn], wa[maxn], wb[maxn], height[maxn], st[maxk+2][maxn], N;
    bool cmp(int *r, int a, int b, int l){return r[a]==r[b] and r[a+l]==r[b+l];}
    void clear()
    {
        cl(sa), cl(rank), cl(ws), cl(wv), cl(wa), cl(wb), cl(height);
    }
    void build(int *r, int n, int m)
    {
        N=n;
        n++;
        int i, j, k=0, p, *x=wa, *y=wb, *t;
        for(i=0;i<m;i++)ws[i]=0;
        for(i=0;i<n;i++)ws[x[i]=r[i]]++;
        for(i=1;i<m;i++)ws[i]+=ws[i-1];
        for(i=n-1;i>=0;i--)sa[--ws[x[i]]]=i;
        for(p=j=1;p<n;j<<=1,m=p)
        {
            for(p=0,i=n-j;i<n;i++)y[p++]=i;
            for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
            for(i=0;i<n;i++)wv[i]=x[y[i]];
            for(i=0;i<m;i++)ws[i]=0;
            for(i=0;i<n;i++)ws[wv[i]]++;
            for(i=1;i<m;i++)ws[i]+=ws[i-1];
            for(i=n-1;i>=0;i--)sa[--ws[wv[i]]]=y[i];
            for(t=x,x=y,y=t,p=1,i=1,x[sa[0]]=0;i<n;i++)
                x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        }
        
        for(i=0;i<n;i++)rank[sa[i]]=i;
        
        for(i=0;i<n-1;height[rank[i++]]=k)
            for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
    }
    void build_st()     //st表
    {
        int i, k;
        for(i=1;i<=N;i++)st[0][i]=height[i];
        for(k=1;k<=maxk;k++)
            for(i=1;i+(1<<k)-1<=N;i++)
                st[k][i]=min(st[k-1][i],st[k-1][i+(1<<k-1)]);
    }
    int lcp(int x, int y)   //最长公共前缀
    {
        int l=rank[x], r=rank[y];
        if(l>r)swap(l,r);
        if(l==r)return N-sa[l];
        int t=log2(r-l);
        return min(st[t][l+1],st[t][r-(1<<t)+1]);
    }
}SA;
struct SegmentTree
{
    int Max[maxn<<2], l[maxn<<2], r[maxn<<2];
    void build(int pos, int L, int R)
    {
        int mid(L+R>>1);
        l[pos]=L, r[pos]=R, Max[pos]=-iinf;
        if(L==R)return;
        build(pos<<1,L,mid);
        build(pos<<1|1,mid+1,R);
    }
    int segmax(int pos, int L, int R)
    {
        int ans(-iinf), mid(l[pos]+r[pos]>>1);
        if(L<=l[pos] and R>=r[pos])return Max[pos];
        if(L<=mid)ans=max(ans,segmax(pos<<1,L,R));
        if(R>mid)ans=max(ans,segmax(pos<<1|1,L,R));
        return ans;
    }
    void insert(int pos, int tar, int val)
    {
        int mid(l[pos]+r[pos]>>1);
        if(l[pos]==r[pos])
        {
            Max[pos]=max(Max[pos],val);
            return;
        }
        if(tar<=mid)insert(pos<<1,tar,val);
        else insert(pos<<1|1,tar,val);
        Max[pos]=max(Max[pos<<1],Max[pos<<1|1]);
    }
}segtree;
char s[maxn];
int r[maxn], lens[maxn], start[maxn], w[maxn];
pair<int,int> get_range(int x)
{
    pair<int,int> ans;
    int l, r, mid, pos(SA.rank[start[x]]);
    l=0, r=pos, mid=l+r>>1;
    while(l<r)
    {
        if(SA.lcp(SA.sa[mid],SA.sa[pos])<lens[x])l=mid+1;
        else r=mid;
        mid=l+r>>1;
    }
    ans.first=l;
    l=pos, r=SA.N, mid=l+r+1>>1;
    while(l<r)
    {
        if(SA.lcp(SA.sa[mid],SA.sa[pos])<lens[x])r=mid-1;
        else l=mid;
        mid=l+r+1>>1;
    }
    ans.second=l;
    return ans;
}
int main()
{
    int T, N, len, ans;
    scanf("%d",&T);
    for(auto kase=1;kase<=T;kase++)
    {
        len=0;
        scanf("%d",&N);
        for(auto i=1;i<=N;i++)
        {
            scanf("%s %d",s,w+i);
            lens[i]=strlen(s);
            start[i]=len;
            for(auto j=0;j<lens[i];j++)r[len++]=s[j];
            r[len++]=i+300;
        }
        SA.clear();
        r[len]=0;
        SA.build(r,len,N+500);
        SA.build_st();
        segtree.build(1,0,SA.N);
        ans=0;
        for(auto i=N;i;i--)
        {
            auto range=get_range(i);
            auto f=max( 0, segtree.segmax(1,range.first,range.second) )+w[i];
            for(auto j=0;j<lens[i];j++)segtree.insert(1,SA.rank[start[i]+j],f);
            ans=max(ans,f);
        }
        printf("Case #%d: %d\n",kase,ans);
    }
    return 0;
}