AC自動機-題目集合

 AC自動機-題目集合 模板

  如果你想要學習AC自動機,推薦一些學習資料.

學習可以看這篇博客 http://blog.csdn.net/niushuai666/article/details/7002823

或者看看大佬的視頻講解 http://www.bilibili.com/video/av6295004/

下面是一些AC自動機的一些入門題目.和AC代碼,先學再做練習.

題目鏈接: hdu-2222

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<set>
#include<map>
#include<list>
#include<queue>
#include<deque>
#include<stack>
#include<string>
#include<vector>
#include<iostream>
#include<algorithm>
#include<stdlib.h>
#include<time.h>

using namespace std;
typedef long long LL;
const int INF=2e9+1e8;
const int MOD=1e9+7;
const int MAXSIZE=1e6+5;
const double eps=0.0000000001;
void fre()
{
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
}
#define memst(a,b) memset(a,b,sizeof(a))
#define fr(i,a,n) for(int i=a;i<n;i++)

const int MAXNode=5e5+5;
struct Aho
{
    int ch[MAXNode][26]; // 多模式字符串 需要構建的 Tire(字典樹)
    int last[MAXNode];   //  統計某一個模式串 (在Tire中MaxNode位置結尾的字符串) 個數,
    int fail[MAXNode]; //失配指針
    int sz,root;
    int newnode()
    {
        memset(ch[sz],-1,sizeof(ch[sz]));
        last[sz++]=0;
        return sz-1;
    }
    void init()
    {
        sz=0;
        root=newnode();
    }
    void insert(char *s)
    {
        int len=strlen(s);
        int now=root;
        for(int i=0; i<len; i++)
        {
            if(ch[now][s[i]-'a']==-1) ch[now][s[i]-'a']=newnode();
            now=ch[now][s[i]-'a'];
        }
        last[now]++;
    }
    void GetFail()// 如何構建失敗指針,還是不太會.所以套 kuangbin 模板
    {
        queue<int>Q;
        fail[root]=root;
        for(int i = 0; i < 26; i++)
        {
            if(ch[root][i]==-1)
                ch[root][i]=root;
            else
            {
                fail[ch[root][i]]=root;
                Q.push(ch[root][i]);
            }
        }
        while(!Q.empty())
        {
            int now=Q.front();
            Q.pop();
            for(int i=0; i<26; i++)
            {
                if(ch[now][i]==-1)
                    ch[now][i]=ch[fail[now]][i];
                else
                {
                    fail[ch[now][i]]=ch[fail[now]][i];
                    Q.push(ch[now][i]);
                }
            }
        }
    }
    int match(char *s)
    {
        int len=strlen(s);
        int now=root;
        int res=0;
        for(int i=0; i<len; i++)
        {
            now=ch[now][s[i]-'a'];
            int temp=now;
            while(temp!=root)
            {
                res+=last[temp];
                last[temp]=0;
                temp=fail[temp];
            }
        }
        return res;
    }
} ac;
char str[MAXSIZE];
inline void read(int& ret) //適用於正整數
{
    char c;
    ret=0;
    while((c=getchar())<'0'||c>'9');
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
}
int main(int argc,char *argv[])
{
    int ncase;
    read(ncase);
    while(ncase--)
    {
        int n;
        ac.init();
        scanf("%d",&n);
        for(int i=0; i<n; i++)
        {
            scanf("%s",str);
            ac.insert(str);
        }
        ac.GetFail();
        scanf("%s",str);
        printf("%d\n",ac.match(str));
    }
    return 0;
}

/**  http://blog.csdn.net/wr_technology  */

鄙人只是存模板,失配指針怎麼構建,目前正在學習之中。若想要學習 AC自動機。請點這篇博文。總結的詳細。

------------------------------------------  分割線  ---------------------------------------------------------

上面是kuangbin模板,不太會。因爲他在構建fail指針的的時候不是最簡單的方法。所以我又找了一另一個代碼來學習,這一份就是樸素的方法,不像kuangbin模板優化過,不適合初學者。

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<set>
#include<map>
#include<list>
#include<queue>
#include<deque>
#include<stack>
#include<string>
#include<vector>
#include<iostream>
#include<algorithm>
#include<stdlib.h>
#include<time.h>

using namespace std;
typedef long long LL;
const int INF=2e9+1e8;
const int MOD=1e9+7;
const int MAXSIZE=1e6+5;
const double eps=0.0000000001;
void fre()
{
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
}
#define memst(a,b) memset(a,b,sizeof(a))
#define fr(i,a,n) for(int i=a;i<n;i++)

const int MAXNode=5e5+5;
struct Aho
{
    int ch[MAXNode][26]; // 多模式字符串 需要構建的 Tire(字典樹)
    int last[MAXNode];   //  統計某一個模式串 (在Tire中MaxNode位置結尾的字符串) 個數,
    int fail[MAXNode]; //失配指針
    int sz,root;
    int newnode()
    {
        memset(ch[sz],-1,sizeof(ch[sz]));
        last[sz++]=0;
        return sz-1;
    }
    void init()
    {
        sz=0;
        root=newnode();
    }
    void insert(char *s)
    {
        int len=strlen(s);
        int now=root;
        for(int i=0; i<len; i++)
        {
            if(ch[now][s[i]-'a']==-1) ch[now][s[i]-'a']=newnode();
            now=ch[now][s[i]-'a'];
        }
        last[now]++;
    }
    void GetFail()//
    {
        queue<int>Q;
        Q.push(root);
        while(!Q.empty())
        {
            int now=Q.front();
            Q.pop();
            for(int i=0;i<26;i++)
            {
                if(ch[now][i]!=-1)
                {
                    if(now==root) fail[ch[now][i]]=root;
                    else
                    {
                        int temp=fail[now];
                        while(temp&&ch[temp][i]==-1) temp=fail[temp];
                        fail[ch[now][i]]=ch[temp][i];
                    }
                    Q.push(ch[now][i]);
                }
            }
        }
        return ;
    }
    int match(char *s)
    {
        int len=strlen(s);
        int now=root,temp;
        int res=0;
        for(int i=0; i<len; i++)
        {

            while(now&&ch[now][s[i]-'a']==-1) 
                now=fail[now];temp=now=ch[now][s[i]-'a'];
            while(temp!=root)
            {
                res+=last[temp];
                last[temp]=0;
                temp=fail[temp];
            }
        }
        return res;
    }
} ac;
char str[MAXSIZE];
inline void read(int& ret)
{
    char c;
    ret=0;
    while((c=getchar())<'0'||c>'9');
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
}
int main(int argc,char *argv[])
{
    int ncase;
    read(ncase);
    while(ncase--)
    {
        int n;
        ac.init();
        scanf("%d",&n);
        for(int i=0; i<n; i++)
        {
            scanf("%s",str);
            ac.insert(str);
        }
        ac.GetFail();
        scanf("%s",str);
        printf("%d\n",ac.match(str));
    }
    return 0;
}

/**  http://blog.csdn.net/wr_technology  */

------------------------------------------------------------------------------------------------------------------------------------------------------------

下面是 hdu-2896 

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<set>
#include<map>
#include<list>
#include<queue>
#include<deque>
#include<stack>
#include<string>
#include<vector>
#include<iostream>
#include<algorithm>
#include<stdlib.h>
#include<time.h>
using namespace std;
typedef long long LL;
const int INF=2e9+1e8;
const int MOD=1e9+7;
const int MAXSIZE=1e6+5;
const double eps=0.0000000001;
void fre()
{
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
}
#define memst(a,b) memset(a,b,sizeof(a))
#define fr(i,a,n) for(int i=a;i<n;i++)
const int MAXNode=5e5+5;


int ans[10005][505],cnt[505];
struct Aho
{
    int ch[100000][128]; // 多模式字符串 需要構建的 Tire(字典樹)
    int last[MAXNode];   //  統計某一個模式串 (在Tire中MaxNode位置結尾的字符串) 個數,
    int fail[MAXNode]; //失配指針
    int sz,root;
    int newnode()
    {
        memset(ch[sz],-1,sizeof(ch[sz]));
        last[sz++]=0;
        return sz-1;
    }
    void init()
    {
        sz=0;
        root=newnode();
    }
    void insert(char *s,int id)
    {
        int now=root;
        for(int i=0;  s[i]!='\0' ; i++)
        {
            if(ch[now][(int)s[i]]==-1) ch[now][(int)s[i]]=newnode();
            now=ch[now][(int)s[i]];
        }
        last[now]=id;
    }
    void GetFail()
    {
        queue<int>Q;
        Q.push(root);
        while(!Q.empty())
        {
            int now=Q.front();
            Q.pop();
            for(int i=0; i<128; i++)
            {
                if(ch[now][i]!=-1)
                {
                    if(now==root) fail[ch[now][i]]=root;
                    else
                    {
                        int temp=fail[now];
                        while(temp&&ch[temp][i]==-1) temp=fail[temp];
                        fail[ch[now][i]]=ch[temp][i];
                    }
                    Q.push(ch[now][i]);
                }
            }
        }
        return ;
    }
    void match(char *s,int id)
    {
        int  flag[550];
        memset(flag,0,sizeof(flag));
        int now=root,temp;
        for(int i=0; s[i]!='\0'; i++)
        {

            while(now&&ch[now][(int)s[i]]==-1)
                now=fail[now];
            temp=now=ch[now][(int)s[i]];
            while(temp!=root&&flag[last[temp]]==0)
            {
                if(last[temp])
                {
                    ans[id][cnt[id]++]=last[temp];
                    flag[last[temp]]=1;
                }
                temp=fail[temp];
            }
            if(cnt[id]>=3) break;
        }
        return ;
    }
} ac;
char str[MAXSIZE];
void print(int m)
{
    int sum=0;
    for(int i=1; i<=m; i++)
    {
        if(!cnt[i]) continue;
        printf("web %d:",i);
        for(int j=0; j<cnt[i]; j++) printf(" %d",ans[i][j]);
        printf("\n");
        sum++;
    }
    printf("total: %d\n",sum);
}

int main(int argc,char *argv[])
{
    int n,m;
    scanf("%d",&n);
    getchar();
    memset(cnt,0,sizeof(cnt));
    ac.init();
    for(int i=1; i<=n; i++)
    {
        gets(str);
        ac.insert(str,i);
    }
    ac.GetFail();
    scanf("%d",&m);
    getchar();
    for(int j=1; j<=m; j++)
    {
        scanf("%s",str);
        ac.match(str,j);
        sort(ans[j],ans[j]+cnt[j]);
    }
    print(m);
    return 0;
}

-----------------------------------------------------------------------------------------------------------------------

hihocoder-1036

#include<cstdio>
#include<cstring>
#include<cctype>
#include<cmath>
#include<set>
#include<map>
#include<list>
#include<queue>
#include<deque>
#include<stack>
#include<string>
#include<vector>
#include<iostream>
#include<algorithm>
#include<stdlib.h>
#include<time.h>

using namespace std;
typedef long long LL;
const int INF=2e9+1e8;
const int MOD=1e9+7;
const int MAXSIZE=1e6+5;
const double eps=0.0000000001;
void fre()
{
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
}
#define memst(a,b) memset(a,b,sizeof(a))
#define fr(i,a,n) for(int i=a;i<n;i++)

const int MAXNode=5e6+5;

struct Aho
{
    int ch[MAXNode][26],fail[MAXNode],last[MAXNode],root,sz;
    int newnode()
    {
        memset(ch[sz],-1,sizeof(ch[sz]));
        last[sz++]=0;
        return sz-1;
    }
    void init()
    {
        sz=0;
        root=newnode();
    }
    void insert(char *s)
    {
        int now=root;
        for(int i=0; s[i]!='\0'; i++)
        {
            int id=(int)(s[i]-'a');
            if(ch[now][id]==-1) ch[now][id]=newnode();
            now=ch[now][id];
        }
        last[now]=1;
    }
    void GetFail()
    {
        queue<int>Q;
        fail[root]=root;
        Q.push(root);
        while(!Q.empty())
        {
            int now=Q.front();
            Q.pop();
            for(int i=0; i<26; i++)
            {
                if(now==root)
                {
                    if(ch[now][i]==-1) ch[now][i]=root;
                    else fail[ch[now][i]]=root,Q.push(ch[now][i]);
                }
                else
                {
                    if(ch[now][i]==-1) ch[now][i]=ch[fail[now]][i];
                    else fail[ch[now][i]]=ch[fail[now]][i],Q.push(ch[now][i]);
                }
            }
        }
    }
    bool match(char *s)
    {
        int now=root;
        for(int i=0; s[i]!='\0'; i++)
        {
            now=ch[now][(int)(s[i]-'a')];
            if(last[now]) return true;
        }
        return false;
    }
} ac;
char str[MAXSIZE];
int main()
{
    int n;
    cin>>n;
    ac.init();
    for(int i=0; i<n; i++)
    {
        scanf("%s",str);
        ac.insert(str);
    }
    scanf("%s",str);
    ac.GetFail();
    if(ac.match(str)) cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
    return 0;
}