HDU6208The Dominator of Strings 【AC自动机】

传送门


显然只有长度最长的串才可能是答案
若有2个串长度最长 且不相同 无解

建立ac自动机,若存在>=2个深度最大的节点 无解
用最长的串跑一遍查询,如果能匹配到的串恰好有n个 答案就是该串

#include<stdio.h>
#include<bits/stdc++.h>
#define ll long long
#define pii pair<int,int>
#define pll pair<ll,ll>
#define MEM(a,x) memset(a,x,sizeof(a))
#define lowbit(x) ((x)&-(x))

using namespace std;

const int INF = 1e9+7;
const int inf=INF;
const int N = 100000 + 5;

char str[N];
bool vis[N];

struct Trie{
    const static int maxn=26e5+5;
    const static int BASE='a';
    const static int N = 26;
    int next[maxn][N],end[maxn];
    int father[maxn],dept[maxn];
    char ch[maxn];
    int root,L;
    //bool vis[N];

    int newNode(){
        MEM(next[L],-1);
        end[L++]=0;
        return L-1;
    }
    void init(){
        L=0;
        root=newNode();
        dept[root]=0;
    }
    void insert(char buf[]){
        int now=root;
        for(int i=0;buf[i];++i){
            int j=buf[i]-BASE;
            if(next[now][j]==-1){
                int x=newNode();
                dept[x]=dept[now]+1;
                ch[x]=buf[i];
                father[x]=now;
                next[now][j]=x;

            }
            now=next[now][j];
        }
        end[now]++;
    }
    //ac自动机
    int fail[maxn];
    void build(){
        queue<int>que;
        fail[root]=root;
        for(int i=0;i<N;++i){
            if(next[root][i]==-1){
                next[root][i]=root;
            }
            else{
                fail[next[root][i]]=root;
                que.push(next[root][i]);
            }
        }
        while(!que.empty()){
            int now=que.front();
            que.pop();
            for(int i=0;i<N;++i){
                if(next[now][i]==-1){
                    next[now][i]=next[fail[now]][i];
                }
                else{
                    fail[next[now][i]]=next[fail[now]][i];
                    que.push(next[now][i]);
                }
            }
        }
    }

    bool check1(int&idx){//是否只有一个长度最长的串 idx=该串最后一个元素在自动机中下标
        int maxDept=0;
        for(int i=0;i<L;++i){
            maxDept=max(maxDept,dept[i]);
        }
        int deptNum=0;
        for(int i=0;i<L;++i){
            if(dept[i]==maxDept){
                idx=i;
                deptNum+=1;
            }
        }
        return deptNum==1;
    }

    ll getNum(int now){
        ll ans=0;
        while(now!=root&&!vis[now]){
            vis[now]=true;
            ans+=end[now];
            end[now]=0;
            now=fail[now];
        }
        return ans;
    }
    void rev(char*str){
        int n=strlen(str);
        int i=0,j=n-1;
        while(i<j){
            swap(str[i],str[j]);
            ++i;
            --j;
        }
    }

    void slove(int idx,int n){
        fill(vis,vis+L+1,0);
        int now=idx;
        int sum=0;
        int ansIdx=0;
        while(now!=root){
            sum+=getNum(now);
            str[ansIdx++]=ch[now];
            now=father[now];
        }
        if(sum==n){
            str[ansIdx++]=0;
            rev(str);
            puts(str);
        }
        else{
            puts("No");
        }
    }
}trie;

int main(){
    //freopen("/home/lu/code/r.txt","r",stdin);
    //freopen("/home/lu/code/w.txt","w",stdout);
    int T;
    scanf("%d",&T);
    while(T--){
        int n;
        scanf("%d",&n);
        trie.init();
        for(int i=0;i<n;++i){
            scanf("%s",str);
            trie.insert(str);
        }
        int ed=0;
        if(!trie.check1(ed)){
            puts("No");
            continue;
        }
        trie.build();
        trie.slove(ed,n);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值