Codeforces Round #542 Morse Code（字典树-dp）

题目链接：Morse Code

题意：

给出一个01串，除"0011", "0101", "1110", and "1111".外，其中每长度不超过4的01串可以表示一个字母，给你一个长度为m的串。问该01串的全部子串可以表示多少种不同的字母序列。

m <= 3000

例如"111" 的全部子串有三种"111", "11", "1"
"111"可以表示为:"1\1\1", "1\11", "11\1", "111"
"11"可以表示为"1\1", "11"
"1"可以表示为"1",
答案为7

又例如串"0011"，的全部子串: "0011", "001", "011", "00", "01", "11", "0", "1"
表示的字母序列可能为:
"0\011", "00\11", "001\1", "0\0\11", "0\01\1", "00\1\1", "0\0\1\1", 
"001", "0\01", "00\1", "0\0\1",       "011", "0\11", "01\1", "0\1\1",
"00", "0\0", "01", "0\1", "11", "1\1", "0", "1", 
答案为23

题目分析：

可以考虑对原串的每个本质不同子串进行计算答案贡献，用 cur[j] 表示以j处为起点的子串当前在trie树上的节点，再在cur[j]节点后加入一个字符x后，如果trie[cur[j]][x]原本节点已经存在，说明从j处起点到当前加入的字符x构成的子串已经存在，不用计算答案，否则添加新节点，则该节点的答案贡献为 sum(dp[k])，k为当前节点之前的4个祖先，第四个祖先需要判断从那个位置到当前位置是否为除外的四个串，计算完当前节点答案后将cur[j] = trie[cur[j]][x] 。

依此加入全部字符，每个字符都枚举起点j，复杂度O(n^2)。

代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const ll mod = 1e9+7;
const int maxn = 3000*1500;
int trie[maxn][2], tot, dp[maxn], fa[maxn];
int cur[3010], ans;
int n, a[3010];
bool flag;
bool check(int i){
    if(a[i-3] == 0 && a[i-2] == 0 && a[i-1] == 1 && a[i] == 1)return 1;
    if(a[i-3] == 0 && a[i-2] == 1 && a[i-1] == 0 && a[i] == 1)return 1;
    if(a[i-3] == 1 && a[i-2] == 1 && a[i-1] == 1 && a[i] == 0)return 1;
    if(a[i-3] == 1 && a[i-2] == 1 && a[i-1] == 1 && a[i] == 1)return 1;
    return 0;
}
void add(int j,int x){
    int o = cur[j];
    if(trie[o][x]){
        cur[j] = trie[o][x];
        return;
    }
    cur[j] = trie[o][x] = ++tot;
    fa[tot] = o;
    for(int i = 1;i <= 4;i++){
        if(i == 4 && flag)break;
        dp[tot] = (dp[tot] + dp[o])%mod;
        o = fa[o];
        if(o == -1)break;
    }
    ans = (ans + dp[tot])%mod;
}
int main(){
    scanf("%d",&n);
    dp[tot] = 1; fa[tot] = -1; flag = 0; ans = 0;
    for(int i = 0;i<=n;i++)cur[i] = 0;
    for(int i = 1;i<=n;i++){
        scanf("%d",&a[i]);
        if(i>=4)flag = check(i);
        for(int j = 0;j<i;j++){
            add(j, a[i]);
        }
        printf("%d\n",ans);
    }
}