++ 實現 struct ostream -s swa 一個 turn div

Description

魔咒串由許多魔咒字符組成，魔咒字符可以用數字表示。例如可以將魔咒字符 1、2 拼湊起來形成一個魔咒串 [1,2]。 一個魔咒串 S 的非空字串被稱為魔咒串 S 的生成魔咒。 例如 S=[1,2,1] 時，它的生成魔咒有 [1]、[2]、[1,2]、[2,1]、[1,2,1] 五種。S=[1,1,1] 時，它的生成魔咒有 [1]、 [1,1]、[1,1,1] 三種。最初 S 為空串。共進行 n 次操作，每次操作是在 S 的結尾加入一個魔咒字符。每次操作後都 需要求出，當前的魔咒串 S 共有多少種生成魔咒。

Input

第一行一個整數 n。 第二行 n 個數，第 i 個數表示第 i 次操作加入的魔咒字符。 1≤n≤100000。,用來表示魔咒字符的數字 x 滿足 1≤x≤10^9

Output

輸出 n 行，每行一個數。第 i 行的數表示第 i 次操作後 S 的生成魔咒數量

Sample Input

7
1 2 3 3 3 1 2

Sample Output

1
3
6
9
12
17
22

HINT

Source

鳴謝Menci上

考慮添加的話所有後綴都要改變，並不好做，所以可以考慮把序列翻轉，變為每次在前面添加，這樣對以前的後綴就沒有影響了，且答案和原來是等價的，所以我們翻轉後反著做；

首先需要知道如果沒有添加操作該怎麽做。

一個子串是一個後綴的前綴，對於每一個後綴，能產生len-sa[i]+1個前綴，然後有height[i]個前綴與sa[i-1]重合，所以答案是∑ len-sa[i]+1-height[i];

然後考慮添加，那麽在前面加一個字母，相當於加了一個後綴，我們需要知道這個後綴的前綴能產生多少個沒有出現過的子串；

最直接的想法就是求該後綴和前面已有的後綴的lcp的最大值，然後長度-maxlcp就是貢獻。

然後我們又會知道跟一個後綴lcp最大的後綴必然是排名靠近他的，所以我們需要找到找到該後綴排名的前驅和後繼，然後取這兩個中的lcp的最大者即可

求前驅和後繼的話可以直接用set實現。

// MADE BY QT666
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstring>
#include<set>
#define int long long
using namespace std;
typedef long long ll;
const int N=200050;
const int Inf=1e9+7;
set<int> q;
set<int>::iterator it;
int gi(){
  int x=0,flag=1;
  char ch=getchar();
  while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘) flag=-1;ch=getchar();}
  while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();
  return x*flag;
}
int sa[N],len,y[N],rk,rnk[N],height[N],ans[N],pre[N],pre2[N],ST[N][20],qianqu[N],houji[N],a[N];
struct data{
  int fir,sec,id;
}x[N];
bool cmp(const data &a,const data &b){
  if(a.fir==b.fir) return a.sec<b.sec;
  else return a.fir<b.fir;
}
void work2(){
  rk=1;y[x[1].id]=rk;
  for(int i=2;i<=len;i++){
    if(x[i-1].fir!=x[i].fir||x[i-1].sec!=x[i].sec) rk++;
    y[x[i].id]=rk;
  }
}
void work(){
  sort(x+1,x+1+len,cmp);work2();
  for(int i=1;i<=len;i<<=1){
    for(int j=1;j+i<=len;j++) x[j].fir=y[j],x[j].sec=y[j+i],x[j].id=j;
    for(int j=len-i+1;j<=len;j++) x[j].fir=y[j],x[j].sec=0,x[j].id=j;
    sort(x+1,x+1+len,cmp);work2();
    if(rk==len) break;
  }
}
void get_height(){
  int kk=0;for(int i=1;i<=len;i++) rnk[sa[i]]=i;
  for(int i=1;i<=len;i++){
    if(kk) kk--;
    int j=sa[rnk[i]-1];
    while(a[i+kk]==a[j+kk]) kk++;
    height[rnk[i]]=kk;
  }
}
void make_ST(){
  pre[0]=1;for(int i=1;i<=16;i++) pre[i]=pre[i-1]<<1;
  pre2[0]=-1;for(int i=1;i<=len;i++) pre2[i]=pre2[i>>1]+1;
  for(int i=2;i<=len;i++) ST[i][0]=height[i];
  for(int j=1;j<=16;j++)
    for(int i=2;i<=len;i++){
      if(i+pre[j]-1<=len){
	ST[i][j]=min(ST[i][j-1],ST[i+pre[j-1]][j-1]);
      }
    }
}
int query(int l,int r){
  int x=pre2[r-l+1];
  return min(ST[l][x],ST[r-pre[x]+1][x]);
}
int LCP(int l,int r){
  if(l>r) swap(l,r);
  return query(l+1,r);
}
void update(int x,int y,int flag){
  int a=sa[x],b=sa[y];
  if(flag) qianqu[a]=b;
  else houji[a]=b;
}
main(){
  len=gi();for(int i=len;i>=1;i--) a[i]=gi();
  for(int i=1;i<=len;i++) x[i].id=i,x[i].fir=x[i].sec=a[i];
  work();for(int i=1;i<=len;i++) sa[y[i]]=i;
  get_height();make_ST();
  for(int i=len;i>=1;i--){
    q.insert(rnk[i]);it=q.find(rnk[i]);
    if(it!=q.begin()){
      it--;update(rnk[i],*it,1);it++;
    }
    if((++it)!=q.end()) update(rnk[i],*it,0);
    int le=len-i+1,lcp=max(LCP(rnk[i],rnk[qianqu[i]]),LCP(rnk[i],rnk[houji[i]]));
    ans[le]=ans[le-1]+le-lcp;
  }
  for(int i=1;i<=len;i++) printf("%lld\n",ans[i]);
  return 0;
}

bzoj 4516: [Sdoi2016]生成魔咒