BZOJ3926 ZJOI2015 诸神眷顾的幻想乡 后缀自动机+DFS

题意:给定一颗字符树,求树中路径所构成的不同的字符串的数量,其中AB和BA视作不同的字符串

题解:

题目里有这样一句话:太阳花田的结构比较特殊,只与一个空地相邻的空地数量不超过20个。

一共有10W个点,却只有20个叶子……因此树上所有的字串就是以叶子为起点搜索出的所有字串,丽洁姐真的好善良啊- -(无雾)

这样从每个点开始就能跑出来一颗Trie树,对Trie构造广义后缀自动机——每个节点看成是一个根,在后面加字符的时候和普通的SAM一样。

然后在SAM上用DFS统计不同字串的数量即可

#include <queue>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <climits>
#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long

const int MAXC=10+2;
const int MAXN=100000+2;
struct HASH{
    int u;
    HASH *next;
    HASH(){}
    HASH(int _u,HASH *_next):u(_u),next(_next){}
}*table[MAXN],mem[2*MAXN];
struct SAM{
    int v;
    SAM *child[MAXC],*f;
    bool flag;
    SAM(){}
    SAM(int _v):v(_v),f(0),flag(0){ memset(child,0,sizeof(child));}
}*root,*last=root=new SAM(0),*node[MAXN];
int N,C,c[MAXN],d[MAXN],cnt;
queue<SAM *> q;

void Insert(int u,int v){ table[u]=&(mem[cnt++]=HASH(v,table[u]));}

SAM *Extend(SAM *&x,int c){
    if(x->child[c] && x->child[c]->v==x->v+1) return x->child[c];

    SAM *p=x,*np=new SAM(p->v+1);
    while(p && !p->child[c]) p->child[c]=np,p=p->f;

    if(!p) np->f=root;
    else{
        SAM *q=p->child[c];
        if(q->v==p->v+1) np->f=q;
        else{
            SAM *nq=new SAM(p->v+1);
            memcpy(nq->child,q->child,sizeof(q->child));
            nq->f=q->f,q->f=np->f=nq;
            while(p && p->child[c]==q) p->child[c]=nq,p=p->f;
        }
    }
    return np;
}

void DFS(int x,int f){
    node[x]=Extend(node[f],c[x]);
    for(HASH *p=table[x];p;p=p->next)
        if(p->u!=f) DFS(p->u,x);
}

ll BFS(SAM *x){
    ll ret=0;
    q.push(x);

    SAM *t;
    while(!q.empty()){
        t=q.front(),q.pop();
        if(t!=x) ret+=t->v-t->f->v;
        for(int i=0;i<C;i++)
            if(t->child[i] && !t->child[i]->flag)
                t->child[i]->flag=1,q.push(t->child[i]);
    }
    return ret;
}

int main(){
    scanf("%d %d",&N,&C);
    for(int i=1;i<=N;i++) scanf("%d",c+i);
    for(int i=1,a,b;i<N;i++){
        scanf("%d %d",&a,&b);
        d[a]++,d[b]++;
        Insert(a,b),Insert(b,a);
    }

    node[0]=root;
    for(int i=1;i<=N;i++)
        if(d[i]==1) DFS(i,0);

    printf("%lld\n",BFS(root));

    return 0;
}

时间: 2024-12-16 20:10:50

BZOJ3926 ZJOI2015 诸神眷顾的幻想乡 后缀自动机+DFS的相关文章

[BZOJ3926] [ZJOI2015]诸神眷顾的幻想乡|后缀自动机

唔..过去了一个多月才敢动一下一试的题呢..毕竟好神啊好神啊 看右边→就写着Orz yu990601 那不如就从他A掉的T3开始吧... 昨天晚上看了一下SAM..大概还是比较亲切的吧 毕竟去年暑假还是听过的...虽然并没有听懂 但是很开心的是现在再看CLJ的讲稿的时候大部分都能看懂了呢 直接这道题吧...首先这棵树的叶子节点不超过20个 然后一个很神奇的性质就是所有的子串都会在以某个叶子节点为根的树上以一条直线的姿态出现... 这么说不利于解题...也就是必定会作为叶子节点到以它为根的树上的某

bzoj3926: [Zjoi2015]诸神眷顾的幻想乡 后缀自动机在tire树上拓展

题意:有棵树每个点有个颜色(不超过10种),每个节点不超过20个儿子,问你每两点之间的颜色序列不同的有多少种 题解:先建出树,对于每个叶子节点,bfs一遍建在sam上,每次保留当前点在sam上的位置,拓展时用父亲节点在sam上的位置当成last即可.然后统计sam本质不同的字符串有多少个 注:dfs建树复杂度是错的,但是也能过这题 /************************************************************** Problem: 3926 User:

BZOJ 3926 Zjoi2015 诸神眷顾的幻想乡 后缀自动机

题目大意:给定一棵树,每个节点有一个字符,求从一个节点出发沿最短路径走到另一个节点所构成的字符串一共有多少种 此生无悔入东方,来世愿生幻想乡 题目戳这里 注意一句话:太阳花田的结构比较特殊,只与一个空地相邻的空地的数量不超过20个 有奖问答:↑你看到这句话的第一反应是啥? 1.度数<=20 2.叶节点数<=20 仔细看几遍就能找到答案~ [捂脸熊]陈老师真是语文高手.... 叶节点数<=20还做啥了... 直接从每个叶节点DFS一遍,然后构建广义后缀自动机,最终答案就是每个节点的深度-p

bzoj3926: [Zjoi2015]诸神眷顾的幻想乡

1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <cmath> 5 #include <algorithm> 6 #define maxn 100005 7 #define maxl 200005 8 #define maxm 4000005 9 using namespace std; 10 11 typedef long long ll; 12

bzoj3926: [Zjoi2015]诸神眷顾的幻想乡 对广义后缀自动机的一些理解

先说一下对后缀自动机的理解,主要是对构造过程的理解. 构造中,我们已经得到了前L个字符的后缀自动机,现在我们要得到L+1个字符的后缀自动机,什么需要改变呢? 首先,子串$[0,L+1)$对应的状态不存在,应当建立一个状态来表示这个串,显然,这个状态(np)的right集合是{L+1},max=L+1. 现在新建立了一个状态,我们还有两件事要干:找出能转移到这个状态的状态,建立链接:确定这个状态的min,即找到它在parent树上的父亲. 能转移到$np$的状态显然都是right集合包含L的状态,

【BZOJ3926】[Zjoi2015]诸神眷顾的幻想乡 广义后缀自动机

[BZOJ3926][Zjoi2015]诸神眷顾的幻想乡 Description 幽香是全幻想乡里最受人欢迎的萌妹子,这天,是幽香的2600岁生日,无数幽香的粉丝到了幽香家门前的太阳花田上来为幽香庆祝生日. 粉丝们非常热情,自发组织表演了一系列节目给幽香看.幽香当然也非常高兴啦. 这时幽香发现了一件非常有趣的事情,太阳花田有n块空地.在过去,幽香为了方便,在这n块空地之间修建了n-1条边将它们连通起来.也就是说,这n块空地形成了一个树的结构. 有n个粉丝们来到了太阳花田上.为了表达对幽香生日的祝

[Zjoi2015]诸神眷顾的幻想乡

[Zjoi2015]诸神眷顾的幻想乡 Time Limit: 10 Sec  Memory Limit: 512 MBSubmit: 1537  Solved: 892 Description 幽香是全幻想乡里最受人欢迎的萌妹子,这天,是幽香的2600岁生日,无数幽香的粉丝到了幽香家门前的太阳花田上来为幽香庆祝生日. 粉丝们非常热情,自发组织表演了一系列节目给幽香看.幽香当然也非常高兴啦. 这时幽香发现了一件非常有趣的事情,太阳花田有n块空地.在过去,幽香为了方便,在这n块空地之间修建了n-1条

字符串(广义后缀自动机):BZOJ 3926 [Zjoi2015]诸神眷顾的幻想乡

3926: [Zjoi2015]诸神眷顾的幻想乡 Time Limit: 10 Sec  Memory Limit: 512 MBSubmit: 843  Solved: 510[Submit][Status][Discuss] Description 幽香是全幻想乡里最受人欢迎的萌妹子,这天,是幽香的2600岁生日,无数幽香的粉丝到了幽香家门前的太阳花田上来为幽香庆祝生日. 粉丝们非常热情,自发组织表演了一系列节目给幽香看.幽香当然也非常高兴啦. 这时幽香发现了一件非常有趣的事情,太阳花田有n

BZOJ 3926: [Zjoi2015]诸神眷顾的幻想乡

3926: [Zjoi2015]诸神眷顾的幻想乡 Time Limit: 10 Sec  Memory Limit: 512 MBSubmit: 1017  Solved: 599[Submit][Status][Discuss] Description 幽香是全幻想乡里最受人欢迎的萌妹子,这天,是幽香的2600岁生日,无数幽香的粉丝到了幽香家门前的太阳花田上来为幽香庆祝生日. 粉丝们非常热情,自发组织表演了一系列节目给幽香看.幽香当然也非常高兴啦. 这时幽香发现了一件非常有趣的事情,太阳花田有