• BZOJ3998: [TJOI2015]弦论(后缀自动机,Parent树)


    Description

    对于一个给定长度为N的字符串,求它的第K小子串是什么。

    Input

     第一行是一个仅由小写英文字母构成的字符串S

    第二行为两个整数T和K,T为0则表示不同位置的相同子串算作一个。T=1则表示不同位置的相同子串算作多个。K的意义如题所述。

    Output

    输出仅一行,为一个数字串,为第K小的子串。如果子串数目不足K个,则输出-1

    Sample Input

    aabc
    0 3

    Sample Output

    aab

    解题思路:

    在后缀自动机Parent树上的每个节点所代表的串都是以祖先节点为后缀的逆序子串。

    利用这一性质我们可以很方便地求解一个子串出现多少次的问题(其子树内实点数)

    那么这道题是求解排名的问题。

    一个后缀自动机可以识别一个串所有后缀。

    若按前缀查询,就是所有字串,字串出现次数和就是其母串次数和的累加。

    t=0 时,认为其实点只有自己记录。

    而 t=1时,认为其子节点被记录。

    累加其sum值作为以此值为前缀的串个数。

    最后相减逼近答案输出即可。

    代码:

      1 #include<cstdio>
      2 #include<cstring>
      3 #include<algorithm>
      4 typedef long long lnt;
      5 struct sant{
      6     int tranc[26];
      7     int len;
      8     int pre;
      9 }s[1000000];
     10 struct pnt{
     11     lnt sum;
     12     lnt size;
     13 }p[1000000];
     14 int siz;
     15 int fin;
     16 int n,t;
     17 lnt k;
     18 char tmp[1000000];
     19 int has[1000000];
     20 int topo[1000000];
     21 void Insert(int c)
     22 {
     23     int nwp,nwq,lsp,lsq;
     24     nwp=++siz;
     25     s[nwp].len=s[fin].len+1;
     26     p[nwp].size=1;
     27     for(lsp=fin;lsp&&!s[lsp].tranc[c];lsp=s[lsp].pre)
     28         s[lsp].tranc[c]=nwp;
     29     if(!s[lsp].tranc[c])
     30         s[nwp].pre=1;
     31     else{
     32         lsq=s[lsp].tranc[c];
     33         if(s[lsq].len==s[lsp].len+1)
     34             s[nwp].pre=lsq;
     35         else{
     36             nwq=++siz;
     37             s[nwq]=s[lsq];
     38             s[nwq].len=s[lsp].len+1;
     39             s[nwp].pre=s[lsq].pre=nwq;
     40             while(s[lsp].tranc[c]==lsq)
     41             {
     42                 s[lsp].tranc[c]=nwq;
     43                 lsp=s[lsp].pre;
     44             }
     45         }
     46     }
     47     fin=nwp;
     48     return ;
     49 }
     50 int main()
     51 {
     52     fin=siz=1;
     53     scanf("%s",tmp+1);
     54     scanf("%d%lld",&t,&k);
     55     n=strlen(tmp+1);
     56     for(int i=1;i<=n;i++)
     57         Insert(tmp[i]-'a');
     58     for(int i=1;i<=siz;i++)
     59         has[s[i].len]++;
     60     for(int i=1;i<=siz;i++)
     61         has[i]+=has[i-1];
     62     for(int i=1;i<=siz;i++)
     63         topo[has[s[i].len]--]=i;
     64     for(int i=siz;i;i--)
     65         if(t)
     66             p[s[topo[i]].pre].size+=p[topo[i]].size;
     67         else
     68             p[topo[i]].size=1;
     69     p[1].size=0;
     70     for(int i=siz;i;i--)
     71     {
     72         int h=topo[i];
     73         p[h].sum=p[h].size;
     74         for(int c=0;c<26;c++)
     75             if(s[h].tranc[c])
     76                 p[h].sum+=p[s[h].tranc[c]].sum;
     77     }
     78     if(k>p[1].sum)
     79     {
     80         puts("-1");
     81         return 0;
     82     }
     83     int root=1;
     84     while(k>0)
     85     {
     86         for(int c=0;c<26;c++)
     87         {
     88             int l=s[root].tranc[c];
     89             if(!l)continue;
     90             if(k>p[l].sum)
     91                 k-=p[l].sum;
     92             else{
     93                 putchar('a'+c);
     94                 root=s[root].tranc[c];
     95                 k-=p[root].size;
     96                 break;
     97             }
     98         }
     99     }
    100     return 0;
    101 }
     
  • 相关阅读:
    (一)linux 系统命令与文件
    python(1)- 初识python
    四、计算机硬件知识整理
    三、计算机硬件历史
    二、网络基础之网络协议
    jmeter while循环使用
    gitlab分支管理
    gitlab基本操作
    JAVA多种向influxDB中插入数据方式
    JAVA类变量和实例变量
  • 原文地址:https://www.cnblogs.com/blog-Dr-J/p/10035400.html
Copyright © 2020-2023  润新知