zoukankan      html  css  js  c++  java
  • 【Codeforces163E】e-Government AC自动机fail树 + DFS序 + 树状数组

    E. e-Government

    time limit per test:1 second
    memory limit per test:256 megabytes
    input:standard input
    output:standard output

    The best programmers of Embezzland compete to develop a part of the project called "e-Government" — the system of automated statistic collecting and press analysis.

    We know that any of the k citizens can become a member of the Embezzland government. The citizens' surnames are a1, a2, ..., ak. All surnames are different. Initially all k citizens from this list are members of the government. The system should support the following options:

    • Include citizen ai to the government.
    • Exclude citizen ai from the government.
    • Given a newspaper article text, calculate how politicized it is. To do this, for every active government member the system counts the number of times his surname occurs in the text as a substring. All occurrences are taken into consideration, including the intersecting ones. The degree of politicization of a text is defined as the sum of these values for all active government members.

    Implement this system.

    Input

    The first line contains space-separated integers n and k (1 ≤ n, k ≤ 105) — the number of queries to the system and the number of potential government members.

    Next k lines contain the surnames a1, a2, ..., ak, one per line. All surnames are pairwise different.

    Next n lines contain queries to the system, one per line. Each query consists of a character that determines an operation and the operation argument, written consecutively without a space.

    Operation "include in the government" corresponds to the character "+", operation "exclude" corresponds to "-". An argument of those operations is an integer between 1 and k — the index of the citizen involved in the operation. Any citizen can be included and excluded from the government an arbitrary number of times in any order. Including in the government a citizen who is already there or excluding the citizen who isn't there changes nothing.

    The operation "calculate politicization" corresponds to character "?". Its argument is a text.

    All strings — surnames and texts — are non-empty sequences of lowercase Latin letters. The total length of all surnames doesn't exceed106, the total length of all texts doesn't exceed 106.

    Output

    For any "calculate politicization" operation print on a separate line the degree of the politicization of the given text. Print nothing for other operations.

    Examples

    input
    7 3
    a
    aa
    ab
    ?aaab
    -2
    ?aaab
    -3
    ?aaab
    +2
    ?aabbaa

    output

    6
    4
    3
    6

    Solution

    fail树的经典运用。

    先建出fail树,然后用树状数组维护DFS序即可。

    Code

    #include<iostream>
    #include<cstdio>
    #include<cstring>
    #include<algorithm>
    #include<cmath>
    #include<queue>
    using namespace std;
    #define MAXN 1000100
    int K,N,loc[MAXN],visit[MAXN];
    struct EdgeNode{int next,to;}edge[MAXN<<1];
    int head[MAXN],cnt=1;
    inline void AddEdge(int u,int v) {cnt++; edge[cnt].next=head[u]; head[u]=cnt; edge[cnt].to=v;}
    inline void InsertEdge(int u,int v) {AddEdge(u,v); AddEdge(v,u);}
    char S[MAXN];
    namespace FailTree
    {
        int son[MAXN][27],end[MAXN],sz=1,fail[MAXN];
    #define id(str) str-'a'+1
        inline int Insert(int x,char str[])
        {
            int len=strlen(str+1),now=1;
            for (int i=1; i<=len; i++)
                if (son[now][id(str[i])]) now=son[now][id(str[i])];
                    else son[now][id(str[i])]=++sz,now=sz;
            end[now]=1; loc[x]=now;
        }
        queue<int>q;
        inline void Getfail()
        {
            q.push(1);
            while (!q.empty())
                {
                    int now=q.front(); q.pop();
                    for (int i=1; i<=26; i++)
                        if (son[now][i])
                            {
                                int fa=fail[now];
                                while (fa && !son[fa][i]) fa=fail[fa];
                                fail[son[now][i]]=fa? son[fa][i]:1;
                                q.push(son[now][i]);
                            }
                }
            for (int i=1; i<=sz; i++) InsertEdge(fail[i],i);
        }
    }
    using namespace FailTree;
    namespace Divide
    {
        int pl[MAXN],pr[MAXN],dfn,tree[MAXN<<1];
        inline void DFS(int now,int last)
        {
            pl[now]=++dfn;
            for (int i=head[now]; i; i=edge[i].next)
                if (edge[i].to!=last)
                    DFS(edge[i].to,now);
            pr[now]=++dfn;
        }
        inline int lowbit(int x) {return x&-x;}
        inline void Modify(int pos,int D) {for (int i=pos; i<=dfn; i+=lowbit(i)) tree[i]+=D;}
        inline int Query(int pos) {int re=0; for (int i=pos; i; i-=lowbit(i)) re+=tree[i]; return re;}
        inline int Calc(char str[])
        {
            int len=strlen(str+1),ans=0,now=1;
            for (int i=1; i<=len; i++)
                {
                    while (now && !son[now][id(str[i])]) now=fail[now];
                    now=now? son[now][id(str[i])]:1;
                    ans+=Query(pl[now]);
                }
            return ans;
        }
        inline void Change(int x,int D)
        {
            if (visit[x] && D>0) return;
            if (!visit[x] && D<0) return;
            visit[x]^=1;
            Modify(pl[loc[x]],D); Modify(pr[loc[x]],-D);
        }
    }
    using namespace Divide;
    int main()
    {
        scanf("%d%d",&K,&N);
        for (int i=1; i<=N; i++) scanf("%s",S+1),Insert(i,S);
        Getfail(); DFS(1,0);
        for (int i=1; i<=N; i++) Modify(pl[loc[i]],1),Modify(pr[loc[i]],-1),visit[i]=1;
        while (K--)
            {
                char opt=getchar(); int x; 
                while (opt!='+' && opt!='-' && opt!='?') opt=getchar();
                switch (opt)
                    {
                        case '+' : scanf("%d",&x); Change(x,1); break;
                        case '-' : scanf("%d",&x); Change(x,-1); break;
                        case '?' : scanf("%s",S+1); printf("%d
    ",Calc(S)); break;
                    }
            }
        return 0;
    }
  • 相关阅读:
    编写BinIoDemo.java的Java应用程序,程序完成的功能是:完成1.doc文件的复制,复制以后的文件的名称为自己的学号姓名.doc。
    编写IoDemo.java的Java应用程序,程序完成的功能是:首先读取text.txt文件内容,再通过键盘输入文件的名称为iodemo.txt,把text.txt的内容存入iodemo.txt
    编写TextRw.java的Java应用程序,程序完成的功能是:首先向TextRw.txt中写入自己的学号和姓名,读取TextRw.txt中信息并将其显示在屏幕上。
    事物 视图 与索引
    大家跟我一起涨知识(高级查询与分页)
    涨知识Style
    数据库打印图形
    SQL 编程
    新闻发布系统
    数据库分页储存
  • 原文地址:https://www.cnblogs.com/DaD3zZ-Beyonder/p/6146801.html
Copyright © 2011-2022 走看看