CF1129C Morse Code

Posted 2021-02-10 creed-qwq

pro:
维护一个01字符串,支持在结尾动态加字符。
每一个长度<=4的01串可以对应一个字母(有几个特例除外)
每次操作后询问，这个字符串的所有子串一共可以对应出多少种本质不同的字符串。

sol:
考虑dp。
dp[i][j]表示从i这个字符开始到当前倒数第j位这个01串，可以划分为多少不同的字符串。
每次只需要更新一下从1到i的每一个dp数组即可。
但时考虑本质不同这个问题。
显然每一种字母串对应唯一的01串。
因此我们只需要对相同的01子串只统计一次即可。
这里用SAM进行判重即可。

#include<bits/stdc++.h>
#define N 22000
#define L 20000
#define eps 1e-7
#define inf 1e9+7
#define db double
#define ll long long
#define ldb long double
using namespace std;
inline int read()
{
    char ch=0;
    int x=0,flag=1;
    while(!isdigit(ch)){ch=getchar();if(ch=='-')flag=-1;}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*flag;
}
const ll mo=1e9+7;
int f[N],id[N],dp[N][8];
int root=1,size=1,last=1;
struct node{int pos,len,lnk,nxt[2];}s[N];
void build(int k)
{
    int cur=++size,p=last;last=cur;
    s[cur].len=s[p].len+1;s[cur].pos=s[cur].len;
    while(p&&!s[p].nxt[k])s[p].nxt[k]=cur,p=s[p].lnk;
    if(!p){s[cur].lnk=root;return;}
    int q=s[p].nxt[k];
    if(s[q].len==s[p].len+1){s[cur].lnk=q;return;}
    else
    {
        int clone=++size;
        s[clone]=s[q];s[clone].len=s[p].len+1;
        while(p&&s[p].nxt[k]==q)s[p].nxt[k]=clone,p=s[p].lnk;
        s[q].lnk=s[cur].lnk=clone;
    }
}
int main()
{
    srand(time(0));
    int n=read(),ans=0;
    for(int i=1;i<=n;i++)f[i]=read(),build(f[i]);
    for(register int o=1;o<=n;o++)
    {
        id[o]=root;
        for(register int i=1;i<o;i++)
        {
            for(register int j=4;j>=1;j--)dp[i][j]=dp[i][j-1];dp[i][0]=0;
            for(register int j=1;j<=4;j++)if(o-i+1>=j)
            {
                if(j==4)
                {
                    if(f[o-3]==0&&f[o-2]==0&&f[o-1]==1&&f[o]==1)continue;
                    if(f[o-3]==0&&f[o-2]==1&&f[o-1]==0&&f[o]==1)continue;
                    if(f[o-3]==1&&f[o-2]==1&&f[o-1]==1&&f[o]==0)continue;
                    if(f[o-3]==1&&f[o-2]==1&&f[o-1]==1&&f[o]==1)continue;       
                }
                dp[i][0]=(dp[i][0]+dp[i][j])%mo;
            }
            int x=id[i];
            if(x!=-1){if(s[x].nxt[f[o]])x=s[x].nxt[f[o]];else x=-1;}
            if(x==-1||o==s[x].pos)ans=(ans+dp[i][0])%mo;id[i]=x;
        }
        dp[o][0]=dp[o][1]=1;
        int x=id[o];
        if(x!=-1){if(s[x].nxt[f[o]])x=s[x].nxt[f[o]];else x=-1;}
        if(x==-1||o==s[x].pos)ans=(ans+dp[o][0])%mo;id[o]=x;
        printf("%d
",ans);
    }
    return 0;
}