BZOJ 4212: 神牛的养成计划

4212: 神牛的养成计划

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 148  Solved: 33
[Submit][Status][Discuss]

Description

Hzwer成功培育出神牛细胞，可最终培育出的生物体却让他大失所望......

 

后来，他从某同校女神 牛处知道，原来他培育的细胞发生了基因突变，原先决定神牛特征的基因序列都被破坏了，神牛hzwer很生气，但他知道基因突变的低频性，说不定还有以下优秀基因没有突变，那么他就可以用限制性核酸内切酶把它们切出来，然后再构建基因表达载体什么的，后面你懂的......

 

黄学长现在知道了N个细胞的DNA序列，它们是若干个由小写字母组成的字符串。一个优秀的基因是两个字符串s1和s2,当且仅当s1是某序列的前缀的同时，s2是这个序列的后缀时，hzwer认为这个序列拥有这个优秀基因。

 

现在黄学长知道了M个优秀基因s1和s2，它们想知道对于给定的优秀基因，有多少个细胞的DNA序列拥有它。

 

 

Input

第一行：N，表示序列数

接下来N行，每行一个字符串，代表N个DNA序列，它们的总长为L1

接下来一个M，表示询问数

接下来M行，每行两个字符串s1和s2，由一个空格隔开，hzwer希望你能在线回答询问，所以s1等于“s1”的所有字符按字母表的顺序向后移动ans位（字母表是一个环），ans为上一个询问的答案，s2同理。例如ans=2  “s1”=qz

则s1=sb。对于第一个询问，ans=0

s1和s2的总长度为L2

 

 

Output

输出M行，每行一个数，第i行的数表示有多少个序列拥有第i个优秀基因。

 

Sample Input

10
emikuqihgokuhsywlmqemihhpgijkxdukjfmlqlwrpzgwrwozkmlixyxniutssasrriafu
emikuqihgokuookbqaaoyiorpfdetaeduogebnolonaoehthfaypbeiutssasrriafu
emikuqihgokuorocifwwymkcyqevdtglszfzgycbgnpomvlzppwrigowekufjwiiaxniutssasrriafu
emikuqihgokuorociysgfkzpgnotajcfjctjqgjeeiheqrepbpakmlixyxniutssasrriafu
emikuqihgokuorociysgfrhulymdxsqirjrfbngwszuyibuixyxniutssasrriafu
emikuqihgokuorguowwiozcgjetmyokqdrqxzigohiutssasrriafu
emikuqihgokuorociysgsczejjmlbwhandxqwknutzgdmxtiutssasrriafu
emikuqihgokuorociysgvzfcdxdiwdztolopdnboxfvqzfzxtpecxcbrklvtyxniutssasrriafu
emikuqihgokuorocsbtlyuosppxuzkjafbhsayenxsdmkmlixyxniutssasrriafu
emikuqihgokuorociysgfjvaikktsixmhaasbvnsvmkntgmoygfxypktjxjdkliixyxniutssasrriafu
10
emikuqihgokuorociysg yxniutssasrriafu
aiegqmedckgqknky eqpoowonnewbq
xfbdnjbazhdnhkhvb qrqgbnmlltlkkbtyn
bjfhrnfedlhrlolzfv qppxpoofxcr
zhdfpldcbjf stsidponnvnmmdvap
zhdfpldcbjfpjmjxdt gdstsidponnvnmmdvap
dlhjtphgfnjtnqnbhxr wxwmhtsrrzrqqhzet
bjfhrnfedlhrlolzfv frqppxpoofxcr
zhdfpldcbjf dponnvnmmdvap
ucyakgyxweakehes nondykjiiqihhyqvk

Sample Output

4
7
3
5
5
1
3
5
10
4

HINT

 

N<=2000

L1<=2000000

M<=100000

L2<=2000000

 

 

Solution

先将序列串正序插入trie树，S1匹配到trie树一个节点X，那S1为其子树所有节点表示串的前缀。X的子树满足S1为其前缀。每个节点记录其包含的序列尾节点区间。

然后用DFS序转换为区间，在区间上逆序插入可持久化trie树，来匹配S2的逆序。

trie树+可持久化trie树：

复杂度O(L1+L2)。

#include<cstdio>
#include<string>
#include<vector>
#include<cstring>
const int len(2000),lem(2000000);
struct Trie{int nx[26],tag;}trie[lem+10];
int tot=1,tp,sz[len+10];std::vector<int>line[len+10];
int Eu[lem+10],size[lem+10],cnt;
struct CMTrie{int nx[26],sum;}tree[lem+10];
int stot,root[len+10];
int n,m,x,ans,leng[len+10],l1,l2;
char tmp[lem+10],s1[lem+10],s2[lem+10];std::string D[len+10];
void insert(int k,int x,int l)
{
    if(l==leng[x])
    {
        if(!trie[k].tag)trie[k].tag=++tp;
        line[trie[k].tag].push_back(x);
        sz[trie[k].tag]++;
        return ;
    }
    int next=(D[x][l]-'a');
    if(!trie[k].nx[next])trie[k].nx[next]=++tot;
    insert(trie[k].nx[next],x,l+1);
}
int ins(int k,int x,int l)
{
    if(!l)
    {
        tree[++stot]=tree[k];
        tree[stot].sum++;
        return stot;
    }
    int next=(D[x][l]-'a');
    int tmp=ins(tree[k].nx[next],x,l-1);
    tree[++stot]=tree[k];
    tree[stot].sum-=tree[tree[stot].nx[next]].sum;
    tree[stot].nx[next]=tmp;
    tree[stot].sum+=tree[tree[stot].nx[next]].sum;
    return stot;
}
void dfs(int k)
{
    if(trie[k].tag)
    {
        Eu[k]=++cnt;root[cnt]=root[cnt-1];size[k]=1;
        for(int v=0;v<sz[trie[k].tag];v++)
            root[cnt]=ins(root[cnt],line[trie[k].tag][v],leng[line[trie[k].tag][v]]);
    }else Eu[k]=cnt+1;
    for(int v=0;v<26;v++)
    if(trie[k].nx[v])
    {
        dfs(trie[k].nx[v]);
        size[k]+=size[trie[k].nx[v]];
    }
}
int match(int k)
{
    int l=0;
    for(;l<=l1;l++)
        k=trie[k].nx[s1[l]-'a'];
    return k;
}
int match(int k1,int k2)
{
    int l=l2;
    for(;l>=0;l--)
    k1=tree[k1].nx[s2[l]-'a'],k2=tree[k2].nx[s2[l]-'a'];
    return tree[k1].sum-tree[k2].sum;
}
int main()
{
//    freopen("C.in","r",stdin);
//    freopen("C.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",tmp);
        D[i]=tmp;leng[i]=D[i].length()-1;
    }
    for(int i=1;i<=n;i++)insert(1,i,0);
//    fprintf(stderr,"tot%d\n",tot);
    root[0]=++stot;
    dfs(1);
//    fprintf(stderr,"stot%d\n",stot);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%s%s",s1,s2);
        l1=strlen(s1);l2=strlen(s2);l1--;l2--;
        for(int i=0;i<=l1;i++)
        s1[i]=char((int(s1[i]-'a')+ans)%26+(int)'a');
        for(int i=0;i<=l2;i++)
        s2[i]=char((int(s2[i]-'a')+ans)%26+(int)'a');
        x=match(1);
        ans=match(root[Eu[x]+size[x]-1],root[Eu[x]-1]);
        printf("%d\n",ans);
    }
    return 0;
}