JSOI2012 玄武密码

在美丽的玄武湖畔,鸡鸣寺边,鸡笼山前,有一块富饶而秀美的土地,人们唤作进香河。相传一日,一缕紫气从天而至,只一瞬间便消失在了进香河中。老人们说,这是玄武神灵将天书藏匿在此。
很多年后,人们终于在进香河地区发现了带有玄武密码的文字。更加神奇的是,这份带有玄武密码的文字,与玄武湖南岸台城的结构有微妙的关联。于是,漫长的破译工作开始了。
经过分析,我们可以用东南西北四个方向来描述台城城砖的摆放,不妨用一个长度为N的序列来描述,序列中的元素分别是‘E’,‘S’,‘W’,‘N’,代表了东南西北四向,我们称之为母串。而神秘的玄武密码是由四象的图案描述而成的M段文字。这里的四象,分别是东之青龙,西之白虎,南之朱雀,北之玄武,对东南西北四向相对应。
现在,考古工作者遇到了一个难题。对于每一段文字,其前缀在母串上的最大匹配长度是多少呢?
Input
第一行有两个整数,N和M,分别表示母串的长度和文字段的个数。
第二行是一个长度为N的字符串,所有字符都满足是E,S,W和N中的一个。
之后M行,每行有一个字符串,描述了一段带有玄武密码的文字。依然满足,所有字符都满足是E,S,W和N中的一个。
Output
输出有M行,对应M段文字。 N<=10^7,M<=10^5,每一段文字的长度<=100。
每一行输出一个数,表示这一段文字的前缀与母串的最大匹配串长度。
Sample Input
7 3
SNNSSNS
NNSS
NNN
WSEE
Sample Output
4
2
0

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=10000005,maxm=100005;
int n,m,cnt=1,trie[maxn][4],mp[130],nxt[maxn],vis[maxn],f[maxn];
char s[maxn],str[maxm][105];
bool note[maxn];
void insert(char *str,int x)
{
    int len=strlen(str),u=1;
    for(int i=0;i<len;i++)
    {
        int c=mp[(int)str[i]];
        if(!trie[u][c])
        {
            trie[u][c]=++cnt;
            f[cnt]=u; //第cnt个点的父亲点是U 
        }
        u=trie[u][c];
    }
    vis[x]=u; //第x个单词的结束位置是U 
}
void bfs()
{
    queue<int> q;
    q.push(1);
    for(int i=0;i<4;i++) 
	    trie[0][i]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0;i<4;i++)
            if(!trie[u][i]) 
			    trie[u][i]=trie[nxt[u]][i];
            else
            {
                q.push(trie[u][i]);
                nxt[trie[u][i]]=trie[nxt[u]][i];
            }
    }
}
void find()
{
    int u=1;
    for(int i=0;i<n;i++)//将母串在TRIE上跑一次,看哪些点是可以走到的 
    {
        u=trie[u][mp[(int)s[i]]];
        for(int j=u;j>1;j=nxt[j])
            if(note[j]) 
			    break;
            else 
			    note[j]=true;
    }
}
int calc(char *str,int i)
{
    int ans=strlen(str);//取出字符串的长度 
    for(int u=vis[i];u;u=f[u])
    //找出第I个字符串最后出现的位置
    //如果它在母串遍历时被找到过,则说明当前字符串的[1,i]这一段可以在母串中找到 
    {
        if(note[u]) 
		     return ans;
		//如果这个点没找到过,则说明当前字符串的前缀的右边界要左移 
        ans--;
    }
    return 0;
}
int main()
{
    mp['E']=0;
    mp['S']=1;
    mp['W']=2;
    mp['N']=3;
    scanf("%d%d%s",&n,&m,s);
    for(int i=1;i<=m;i++) //将M个单词加入到TRIE中 
    {
        scanf("%s",str[i]);
        insert(str[i],i);
    }
    bfs();
    find();
    for(int i=1;i<=m;i++) 
	     printf("%d\n",calc(str[i],i));
    return 0;
}

  

猜你喜欢

转载自www.cnblogs.com/cutemush/p/12582045.html
今日推荐