HDU2896 HDU3065(AC自动机)

HDU2896

题意

求目标串中出现了几个给定的子串.

思路

ac自动机模板题().

代码

#include<bits/stdc++.h>
using namespace std;
struct Trie{
    int next[105010][130],fail[105010],end[105010];
    int root,L;
    int newnode(){
        for(int i=0;i<130;i++) next[L][i]=-1;
        end[L++]=-1;
        return L-1;
    }
    void init(){
        L=0;
        root=newnode();
    }
    void insert(char buf[],int id){
        int len=strlen(buf);
        int now=root;
        for(int i=0;i<len;i++){
            if(next[now][buf[i]]==-1)
                next[now][buf[i]]=newnode();
            now=next[now][buf[i]];
        }
        end[now]=id;
    }
    void build(){
        queue<int> Q;
        fail[root]=root;
        for(int i=0;i<130;i++){
            if(next[root][i]==-1)
                next[root][i]=root;
            else {
                fail[next[root][i]]=root;
                Q.push(next[root][i]);
            }
        }
        while(!Q.empty()){
            int now=Q.front();
            Q.pop();
            for(int i=0;i<130;i++){
                if(next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];
                else {
                    fail[next[now][i]]=next[fail[now]][i];
                    Q.push(next[now][i]);
                }
            }
        }
    }
    bool used[510];
    bool query(char buf[],int id,int n){
        // printf("id=%d\n",id);
        int len=strlen(buf);
        int now=root;
        memset(used,false,sizeof(used));
        bool flag=false;
        for(int i=0;i<len;i++){
            // printf("now=%d\n",now);
            now=next[now][buf[i]];
            int tmp=now;
            while(tmp!=root){
                if(end[tmp]!=-1){
                    used[end[tmp]]=true;
                    flag=true;
                }
                tmp=fail[tmp];
            }
        }
        if(!flag) return false;
        printf("web %d:",id);
        for(int i=1;i<=n;i++)
            if(used[i]) printf(" %d",i);
        printf("\n");
        return true;
    }
};
Trie ac;
char buf[1000010];
int main(){
    int n,m;
    scanf("%d",&n);
    ac.init();
    for(int i=1;i<=n;i++){
        scanf("%s",buf);
        ac.insert(buf,i);
    }
    ac.build();
    int ans=0;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",buf);
        if(ac.query(buf,i,n)) ans++;
    }
    printf("total: %d\n",ans);
    //system("pause");
    return 0;
}

HDU3065

题意

在模式串中寻找给定串中出现了几次.

思路

ac自动机模板题().

代码

#include<bits/stdc++.h>
using namespace std;
char s[1010][60];
struct Trie{
    int next[50010][128],fail[50010],end[50010];
    int root,L;
    int used[1010];
    int newnode(){
        memset(used,0,sizeof(used));
        for(int i=0;i<128;i++) next[L][i]=-1;
        end[L++]=-1;
        return L-1;
    }
    void init(){
        L=0;
        root=newnode();
    }
    void insert(char buf[],int id){
        int len=strlen(buf);
        int now=root;
        for(int i=0;i<len;i++){
            if(next[now][buf[i]]==-1)
                next[now][buf[i]]=newnode();
            now=next[now][buf[i]];
        }
        end[now]=id;
    }
    void build(){
        queue<int> Q;
        fail[root]=root;
        for(int i=0;i<128;i++){
            if(next[root][i]==-1)
                next[root][i]=root;
            else {
                fail[next[root][i]]=root;
                Q.push(next[root][i]);
            }
        }
        while(!Q.empty()){
            int now=Q.front();
            Q.pop();
            for(int i=0;i<128;i++){
                if(next[now][i]==-1)
                    next[now][i]=next[fail[now]][i];
                else {
                    fail[next[now][i]]=next[fail[now]][i];
                    Q.push(next[now][i]);
                }
            }
        }
    }
    void query(char buf[],int n){
        // printf("id=%d\n",id);
        int len=strlen(buf);
        int now=root;
        for(int i=0;i<len;i++){
            // if(buf[i]>'Z'&&buf[i]<'A') continue;
            now=next[now][buf[i]];
            int tmp=now;
            while(tmp!=root){
                if(end[tmp]!=-1){
                    used[end[tmp]]++;
                }
                tmp=fail[tmp];
            }
        }
        for(int i=1;i<=n;i++){
            if(used[i]) printf("%s: %d\n",s[i],used[i]);
        }
        // return true;
    }
};
Trie ac;
char buf[2000010];
int main(){
    int n,m;
    while(~scanf("%d",&n)){    
        ac.init();
        for(int i=1;i<=n;i++){
            scanf("%s",s[i]);
            ac.insert(s[i],i);
        }
        ac.build();
        int ans=0;
        scanf("%s",buf);
        ac.query(buf,n);
    }
    //system("pause");
    return 0;
}