Advertisement
yicongli

LG4070

Mar 7th, 2019
133
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.05 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. #define gc c=getchar()
  6. #define r(x) read(x)
  7. #define ll long long
  8.  
  9. template<typename T>
  10. inline void read(T&x){
  11.     x=0;T k=1;char gc;
  12.     while(!isdigit(c)){if(c=='-')k=-1;gc;}
  13.     while(isdigit(c)){x=x*10+c-'0';gc;}x*=k;
  14. }
  15.  
  16. const int N=2e5+7;
  17.  
  18. int Max[N];
  19. int Min[N];
  20. map<int,int>to[N];
  21. int fa[N];
  22. int siz[N];
  23.  
  24. int tot;
  25.  
  26. inline int NewNode(int mx,int mi,int su){
  27.     ++tot;
  28.     Max[tot]=mx;
  29.     Min[tot]=mi;
  30.     fa[tot]=su;
  31.     return tot;
  32. }
  33.  
  34. inline int extend(int u,int c){
  35.     int t=NewNode(Max[u]+1,0,0);
  36.     siz[t]=1;
  37.     for(;u&&!to[u][c];u=fa[u])to[u][c]=t;
  38.     if(!u){
  39.         Min[t]=1;
  40.         fa[t]=1;
  41.         return t;
  42.     }
  43.     int v=to[u][c];
  44.     if(Max[v]==Max[u]+1){
  45.         Min[t]=Max[v]+1;
  46.         fa[t]=v;
  47.         return t;
  48.     }
  49.     int x=NewNode(Max[u]+1,0,fa[v]);
  50.     to[x]=to[v];
  51.     Min[t]=Min[v]=Max[x]+1;
  52.     fa[t]=fa[v]=x;
  53.     Min[x]=Min[fa[x]]+1;
  54.     for(;u&&to[u][c]==v;u=fa[u])to[u][c]=x;
  55.     return t;
  56. }
  57.  
  58. char s[N];
  59.  
  60. int main(){
  61.     int n;r(n);
  62.     int lst=NewNode(0,0,0);
  63.     ll ans=0;
  64.     for(int i=0;i<n;++i){
  65.         int t;r(t);
  66.         lst=extend(lst,t);
  67.         printf("%lld\n",ans+=Max[lst]-Max[fa[lst]]);
  68.     }
  69.     return 0;
  70. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement