洛谷试炼日记（trie树和AC自动机）（解题报告）

Trie树

P4407 [JSOI2009]电子字典

题目链接
第一次近距离接触树上dfs，之前可能接触过但是没仔细了解，就是用dfs枚举,因为题目说目标串可以改动之后变为字典树里的串
所以对目标串进行dfs就行啦
细节看代码就好
AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 1e5 + 10;
int ch[N * 20][26],cnt;
char s[22];
bool bo[N * 20],vis[N * 20],isword;
int n,m,tot;

void insert(char s[]){
    int u = 0;
    for(int i = 0;s[i];i ++){
        int v = s[i] - 'a';
        if(!ch[u][v]) ch[u][v] = ++ cnt;
        u = ch[u][v];
    }
    bo[u] = 1;
}
int len;
void dfs(int rt,int l,int f){
    if(l == len && bo[rt] && !f){
        isword = true;return;
    }
    if(l == len && bo[rt] && f){
        if(!vis[rt]){
            vis[rt] = 1;
            tot ++;//更新答案
        }
        return;
    }
    int c = s[l] - 'a';
    if(!f){
        if(l < len) dfs(rt,l + 1,1);//删除0
        for(int i = 0;i < 26;i ++){
            if(ch[rt][i]){
                dfs(ch[rt][i],l,1);//添加
                if(i != c) dfs(ch[rt][i],l + 1,1); // 替换
            }
        }
    }
    if(l >= len) return;
    if(ch[rt][c]) dfs(ch[rt][c],l + 1,f);//直接查
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i = 0;i < n;i ++){
        scanf("%s",s);
        insert(s);
    }
    for(int i = 0;i < m;i ++){
        scanf("%s",s);
        len = strlen(s);
        dfs(0,0,0);
        if(isword) printf("-1\n");
        else printf("%d\n",tot);
        memset(vis,0,sizeof(vis));
        tot = 0;
        isword = 0;
    }
    return 0;
}

P4551 最长异或路径

题目链接
一道01Trie树半裸题
由题意中的异或我们可以想到将数转为二进制数
那么对于两个结点的异或和呢 会等于根到每个结点路径上的异或和的和（比较绕）
所以我们可以预处理出每个结点到根的路径上的异或和
再以每个结点到根的异或和为数值建立01Trie树
再用每个结点到根的异或和去匹配一次找最大值
由于是由高位到低位匹配，所以如果某个位置存在与该位置原有数值相反的数的话，结果肯定更优 比如 0111 和 1000 肯定后者比较大
AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 2e6 + 10;
struct E{
    int next,to,w;
}e[N];
int head[N],tot = -1;
void add(int u,int v,int w){
    e[++tot].next = head[u];
    head[u] = tot;
    e[tot].to = v;
    e[tot].w = w;
}
int sum[N];
void dfs(int u,int fa){
    for(int i = head[u];~i;i = e[i].next){
        int v = e[i].to,w = e[i].w;
        if(v != fa){
            sum[v] = sum[u] ^ w;
            dfs(v,u);
        }
    }
}

int ch[N][2],cnt;
void insert(int x){
    int u = 0;
    for(int i = (1 << 30);i;i >>= 1){
        bool c = x & i;
        if(!ch[u][c]) ch[u][c] = ++cnt;
        u = ch[u][c];
    }
}

int query(int x){
    int u = 0;
    int ans = 0;
    for(int i = (1 << 30);i;i >>= 1){
        bool c = x & i;
        if(ch[u][!c]){
            ans += i;
            u = ch[u][!c];
        }
        else u = ch[u][c];
    }
    return ans;
}
int main(){
    memset(head,-1,sizeof(head));
    int n;
    scanf("%d",&n);
    for(int i = 1;i < n;i ++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    dfs(1,-1);
    for(int i = 1;i <= n;i ++) insert(sum[i]);
    int ans = 0;
    for(int i = 1;i <= n;i ++){
        ans = max(ans,query(sum[i]));
    }
    printf("%d\n",ans);
    return 0;
}

AC自动机

P5357 【模板】AC自动机（二次加强版）

题目链接
就是一道模板紫题而已啦
多的不说直接看怎么优化AC自动机，告别呆呆的暴力跳fail
看了许多大佬题解，大多用fail树，也有拓扑排序的，其实本质都差不多
这里看看fail树的代码就行啦 我相信大家都看得懂
看不懂自行画个fail树再理解一下也差不多了

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>

using namespace std;
const int N = 2e6 + 10;
char s[N];
int head[N],tot;
int ch[N][26],fail[N],cnt,fl[N],sum[N];
struct E{
    int next,to;
}e[N];
void insert(char s[],int x){
    int u = 0;
    for(int i = 0;s[i];i ++){
        int v = s[i] - 'a';
        if(!ch[u][v]) ch[u][v] = ++ cnt;
        u = ch[u][v];
    }
    fl[x] = u;
}

void getfail(){
    queue<int> q;
    for(int i = 0;i < 26;i ++) if(ch[0][i]) q.push(ch[0][i]);
    while(!q.empty()){
        int u = q.front();q.pop();
        for(int i = 0;i < 26;i ++){
            if(ch[u][i]){
                fail[ch[u][i]] = ch[fail[u]][i];
                q.push(ch[u][i]);
            }
            else ch[u][i] = ch[fail[u]][i];
        }
    }
}

void query(char s[]){
    int u = 0;
    for(int i = 0;s[i];i ++){
        u = ch[u][s[i] - 'a'];
        sum[u] ++;
    }
}

void add(int u,int v){
    e[++ tot].next = head[u];
    head[u] = tot;
    e[tot].to = v;
}

void dfs(int u){
    for(int i = head[u];i;i = e[i].next){
        int v = e[i].to;
        dfs(v);
        sum[u] += sum[v];
    }
}
int n;
int main(){
    scanf("%d",&n);
    for(int i = 1;i <= n;i ++){
        scanf("%s",s);
        insert(s,i);
    }
    getfail();
    scanf("%s",s);
    query(s);
    for(int i = 1;i <= cnt;i ++) add(fail[i],i);
    dfs(0);
    for(int i = 1;i <= n;i ++) printf("%d\n",sum[fl[i]]);
    return 0;
}

P3121 [USACO15FEB]Censoring G

题目链接
个人很喜欢这道题，代码精简思路简单
就AC自动机+两个栈愉快搞定
代码不难看懂
AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>

using namespace std;
const int N = 1e5 + 10;
char s[N],ss[N];
int n,cnt,top;
int ch[N][26],fail[N],fl[N];
int stk1[N],stk2[N];

void insert(char s[]){
    int u = 0;
    for(int i = 0;s[i];i ++){
        int v = s[i] - 'a';
        if(!ch[u][v]) ch[u][v] = ++ cnt;
        u = ch[u][v];
    }
    fl[u] = strlen(s);
}

void getfail(){
    queue<int> q;
    for(int i = 0;i < 26;i ++) if(ch[0][i]) q.push(ch[0][i]);
    while(!q.empty()){
        int u = q.front();q.pop();
        for(int i = 0;i < 26;i ++){
            if(ch[u][i]){
                fail[ch[u][i]] = ch[fail[u]][i];
                q.push(ch[u][i]);
            }
            else ch[u][i] = ch[fail[u]][i];
        }
    }
}

void solve(char s[]){
    int u = 0;
    for(int i = 0;s[i];i ++){
        int v = s[i] - 'a';
        u = ch[u][v];
        stk2[++ top] = u;
        stk1[top] = i;
        if(fl[u]){
            top -= fl[u];
            if(!top) u = 0;
            else u = stk2[top];
        }
    }
}

int main(){
    scanf("%s",s);
    scanf("%d",&n);
    for(int i = 0;i < n;i ++){
        scanf("%s",ss);
        insert(ss);
    }
    getfail();
    solve(s);
    for(int i = 1;i <= top;i ++) printf("%c",s[stk1[i]]);
    return 0;
}