神题啊……搞了我一下午= =(其实第233个提交也是我的)

  我一开始的思路:这个找kpm串的过程,其实就跟在AC自动机上沿fail倒着往下走是差不多的(看当前是哪些点的后缀,如果某个串的后缀是当前串,那它的fail就会指向这里)所以就在fail树上bfs一遍,然后找到所有的编号,排序后输出第k小……

  然后顺利地WA了 QAQ

  只好取膜拜题解……哦原来只要把每个串倒过来,建Trie树的时候,就相当于找当前这个串的子树!

  在子树中统计啊……so easy啦~dfs一遍,把儿子中的编号传给父亲啦~然后就会发现光荣TLE了>_>

  shen me gui!为什么呢?很简单啦,结点那么多,最坏情况下要把n个编号上传n次(比如有10W个相同的串,每个串都为 "aaa....aa"(10W个a))妥妥的TLE了。

  等等……刚刚好像说是子树查询第K大?dfs序不是很好吗!瞬间转成求区间第K大!可持久化线段树!

  ok,终于顺利AC……

  1 /**************************************************************
  2     Problem: 3439
  3     User: Tunix
  4     Language: C++
  5     Result: Accepted
  6     Time:1240 ms
  7     Memory:51688 kb
  8 ****************************************************************/
  9  
 10 //BZOJ 3439
 11 #include<vector>
 12 #include<cstdio>
 13 #include<cstring>
 14 #include<cstdlib>
 15 #include<iostream>
 16 #include<algorithm>
 17 #define rep(i,n) for(int i=0;i<n;++i)
 18 #define F(i,j,n) for(int i=j;i<=n;++i)
 19 #define D(i,j,n) for(int i=j;i>=n;--i)
 20 #define pb push_back
 21 using namespace std;
 22 inline int getint(){
 23     int v=0,sign=1; char ch=getchar();
 24     while(ch<'0'||ch>'9'){ if (ch=='-') sign=-1; ch=getchar();}
 25     while(ch>='0'&&ch<='9'){ v=v*10+ch-'0'; ch=getchar();}
 26     return v*sign;
 27 }
 28 const int N=1e5+10,INF=~0u>>2;
 29 typedef long long LL;
 30 /******************tamplate*********************/
 31 struct Trie{
 32     int ch[26],num;
 33     bool sign;
 34 }T[N];
 35 int cnt=0,n,tot,pos[N];
 36 inline int id(char c){return c-'a';}
 37 vector<int>v[N];
 38 void Insert(char *s,int num){
 39     int x=0,y;
 40     D(i,strlen(s)-1,0){
 41         y=id(s[i]);
 42         if (T[x].ch[y]==0)
 43             T[x].ch[y]=++cnt;
 44         x=T[x].ch[y];
 45     }
 46     T[x].sign=1;
 47     v[x].pb(num);
 48     pos[num]=x;
 49 }
 50 int st[N],ed[N],a[N];
 51 void dfs(int x){
 52     st[x]=tot+1;
 53     rep(i,v[x].size()) a[++tot]=v[x][i];
 54     rep(i,26)
 55         if (T[x].ch[i]!=0)
 56             dfs(T[x].ch[i]);
 57     ed[x]=tot;
 58 }
 59 struct Tree{
 60     int cnt,l,r;
 61 }t[N*30];
 62 int root[N],num=0;
 63 #define mid (l+r>>1)
 64 void update(int &o,int l,int r,int pos){
 65     t[++num]=t[o], o=num, ++t[o].cnt;
 66     if (l==r) return;
 67     if (pos<=mid) update(t[o].l,l,mid,pos);
 68     else update(t[o].r,mid+1,r,pos);
 69 }
 70 int query(int i,int j,int rank){
 71     i=root[i]; j=root[j];
 72     int l=1,r=n;
 73     while(l!=r){
 74         if (t[t[j].l].cnt-t[t[i].l].cnt>=rank)
 75             r=mid,i=t[i].l,j=t[j].l;
 76         else{
 77             rank-=t[t[j].l].cnt-t[t[i].l].cnt;
 78             l=mid+1,i=t[i].r,j=t[j].r;
 79         }
 80     }
 81     if (t[j].cnt-t[i].cnt<rank) return -1;
 82     return l;
 83 }
 84 #undef mid
 85 char s[N];
 86 int main(){
 87 #ifndef ONLINE_JUDGE
 88     freopen("3439.in","r",stdin);
 89     freopen("3439.out","w",stdout);
 90 #endif
 91     n=getint();
 92     F(i,1,n){
 93         scanf("%s",s);
 94         Insert(s,i);
 95     }
 96     dfs(0);
 97 #ifdef debug
 98     F(i,1,tot) printf("%d ",a[i]);puts("");
 99     F(i,1,n) printf("%d %d\n",st[pos[i]],ed[pos[i]]);
100 #endif
101     F(i,1,tot){
102         root[i]=root[i-1];
103         update(root[i],1,n,a[i]);
104     }
105 #ifdef debug
106     F(i,1,num) printf("%d %d %d %d\n",i,t[i].l,t[i].r,t[i].cnt);
107 #endif
108     int k;
109     F(i,1,n){
110         k=getint();
111         int ans=query(st[pos[i]]-1,ed[pos[i]],k);
112 //      printf("%d\n",ans>n ? -1 : ans);
113         printf("%d\n",ans);
114     }
115     return 0;
116 }
View Code

相关文章:

  • 2021-11-02
  • 2021-06-14
  • 2021-11-09
  • 2022-01-17
  • 2021-12-22
  • 2021-11-03
  • 2021-05-29
猜你喜欢
  • 2021-09-30
  • 2021-11-18
  • 2021-09-04
  • 2021-11-11
  • 2021-09-26
  • 2021-10-09
  • 2021-09-01
相关资源
相似解决方案