bzoj2754&JZOJ2834【SCOI2012】喵星球上的点名 AC自动机+STL

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_35866453/article/details/55522367

题意比较复杂,就不说了。

感觉这题目拿来恶心人的。。连输入都要花费好大功夫。表示这种题目经典套路啊,一般来说直接上AC自动机,暴力把大串往小串上贴,然后直接计算答案就行了,但是这数据大小简直了。。算了算发现离散也没用,表示不会用STL,于是去网上膜了一波做法,一脸懵逼,勉强看懂,如果考试肯定切不了这题。

听说正解是SA+主席树。

code(超丑。。):

#include<iostream>
#include<set>
#include<map>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#include<vector>
#include<queue>
#include<algorithm>
#include<cmath>
#define inf 1000000000
#define pa pair<int,int>
#define ll long long 
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
int n,m;
const int maxn=1e5+5,maxm=2e4+5;
int ans1[maxn],ans2[maxn];
vector<int> a[maxm],st[maxn],V,M;
map<int,int>to[maxn];
bool vis[maxn],mark[3*maxm];
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct data
{
    int cnt,ans;
    int fail[maxn],q[maxn];
    data()
    {
        cnt=1;
        fo(i,-1,10000)
        to[0][i]=1;
        fail[1]=0;
    }
    void insert(int id)
    {
        int L=read(),x;
        int now=1;
        fo(i,1,L)
        {
            x=read();
            if (!to[now][x])to[now][x]=++cnt;
            now=to[now][x];
        }
        st[now].push_back(id);
    }
    void buildfail()
    {
        int head=0,tail=1;
        q[0]=1;
        while (head!=tail)
        {
            int now=q[head++];
            for(map<int,int>::iterator i=to[now].begin();i!=to[now].end();i++)
            {
                int t=i->first,k=fail[now];
                while (!to[k][t])k=fail[k];
                fail[i->second]=to[k][t];
                q[tail++]=i->second;
            }
        }
    }
    void get(int id,int x){
        for(int i=x;i;i=fail[i])
            if(!vis[i])
            {
                vis[i]=1;V.push_back(i);
                for(int j=0;j<st[i].size();j++)
                    if(!mark[st[i][j]])
                    {
                        mark[st[i][j]]=1;M.push_back(st[i][j]);
                        ans1[st[i][j]]++;
                        ans2[id]++;
                    }
            }
            else break;
    }
    void solve(int x){
        int now=1;
        for(int i=0;i<a[x].size();i++)
        {
            int t=a[x][i];
            while(!to[now][t])now=fail[now];
            now=to[now][t];get(x,now);
        }
        for(int i=0;i<V.size();i++)vis[V[i]]=0;
        for(int i=0;i<M.size();i++)mark[M[i]]=0;
        V.clear();M.clear();
    }
}trie;
int main()
{
    n=read(),m=read();
    int L,x;
    fo(i,1,n)
    {
        L=read();
        while(L--)x=read(),a[i].push_back(x);
        a[i].push_back(-1);
        L=read();
        while(L--)x=read(),a[i].push_back(x);
    }
    fo(i,1,m)
    trie.insert(i);
    trie.buildfail();
    fo(i,1,n)
    trie.solve(i);
    fo(i,1,m)printf("%d\n",ans1[i]);
    fo(i,1,n)
    {
        printf("%d",ans2[i]);
        if (i!=n)printf(" ");
    }
    return 0;
}
展开阅读全文

没有更多推荐了,返回首页