KGB成功破解特朗普的秘密消息

P2922 [USACO08DEC]秘密消息Secret Message

题面

题目分析

很容易看出来是一个\(Trie\)计数问题.

和一般的\(Trie\)计数不同,在这道题的题意中,文本串是可以作为模式串的前缀的.

比如说文本串是\(0001\),而模式串是\(00010000000\),这种情况也要算上.

区别于一般的\(Trie\)中给一个文本串的末尾的结点做一个标记,同志们得把这个标记改成计数器.

当模式串在\(Trie\)树上跑,如果遇到了计数器,就累加入答案.表明有文本串在中途夭折了.

模式串匹配到尾时,加上该结点被多少个文本串经过的计数器.

理一下两个计数器的作用:

• 第一个计数器的作用是作为模式串前缀的文本串累加上去,在文本串的末尾结点累加,在模式串匹配的过程中统计.

• 第二个计数器表示有多少个文本串的前缀中有模式串,在文本串的每一个结点上都要累加,在模式串匹配完之后统计.

代码

为了减小一点常数,我把Trie过程写在了循环里,边读入边做,请各位同志谅解.

#include "cstdio"
#include "cstring"
#include "algorithm"
#include "iostream"

#define debug(x) printf("debug:%d\n",x)
#define I_copy_this_anwser return

using namespace std;

int m,n,tot=1,ans;
int trie[500010][3],CharCnt[500010],EndCnt[500010];
bool end[500010];

int main()
{
    scanf("%d%d",&m,&n);
    for(int i=1;i<=m;i++)
    {
        int p=1,num,len;
        scanf("%d",&len);
        for(int j=1;j<=len;j++)
        {
            scanf("%d",&num);
            if(trie[p][num]==0)
            {
                trie[p][num]=++tot;
            }
            p=trie[p][num];
            if(j!=len)//这里注意一下,两个计数器不能加重复了
            {
                CharCnt[p]++;//记录文本串前缀的计数器
            }
        }
        EndCnt[p]++;//记录文本串末尾的计数器
    }
    for(int i=1;i<=n;i++)
    {
        int p=1,cnt=0,num,len;
        scanf("%d",&len);
        bool flag=false;
        for(int j=1;j<=len;j++)
        {
            scanf("%d",&num);
            if(flag==true)
            {
                continue;
            }
            p=trie[p][num];
            if(EndCnt[p])
            {
                cnt+=EndCnt[p];
            }
            if(p==0)//后面再也没有能够匹配该模式串的文本串了,输出能够作为文本串前缀的模式串个数即可(科技为了我)
            {
                flag=true;
                printf("%d\n",cnt);
            }
        }
        if(flag==false)
        {
            cnt+=CharCnt[p];//统计模式串可以作为多少个文本串的前缀
            printf("%d\n",cnt);
        }
    }
return 0;
}

转载于:https://www.cnblogs.com/Lemir3/p/11176156.html