HDU2896 && 多校2009 病毒侵袭(AC自动机)

 

利用 ac自动机 将源码字符串中出现的特征码找出来,用 set 存储即可

const int N=1e5+5;

    int n,m;
    int i,j,k;
    char s[N];
    int tot=0,t[N][128];
    int vis[N],f[N];
    set<int> all;

void insert(char *s,int id)
{
    int rt=0;
    for(int i=0;s[i];i++){
        int x=s[i];
        if(!t[rt][x]) t[rt][x]=++tot;
        rt=t[rt][x];
    }
    vis[rt]=id;
}

void getfail()
{
    queue<int> q;
    for(int i=0;i<128;i++) if(t[0][i]) f[t[0][i]]=0,q.push(t[0][i]);
    while(q.size()){
        int rt=q.front(); q.pop();
        for(int i=0;i<128;i++){
            if(t[rt][i]) f[t[rt][i]]=t[f[rt]][i],q.push(t[rt][i]);
            else t[rt][i]=t[f[rt]][i];
        }
    }
}

void query(char *s)
{
    int rt=0;
    all.clear();
    for(int i=0;s[i];i++){
        int x=s[i];
        rt=t[rt][x];
        for(int j=rt;j;j=f[j]) if(vis[j]) all.insert(vis[j]);
    }
}

int main()
{
    //IOS;
    while(~sd(n)){
        for(int i=1;i<=n;i++) ss(s),insert(s,i);
        getfail();
        sd(m);
        int ans=0;
        for(int i=1;i<=m;i++){
            ss(s);
            query(s);
            if(all.size()){
                printf("web %d:",i);
                for(set<int>::iterator it=all.begin();it!=all.end();it++) printf(" %d",*it);
                puts("");
                ans++;
            }
        }
        printf("total: %d\n",ans);
    }
    //PAUSE;
    return 0;
}

 

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 酷酷鲨 设计师:CSDN官方博客 返回首页