#2033. 「SDOI2016」生成魔咒
题目描述
魔咒串由许多魔咒字符组成,魔咒字符可以用数字表示。例如可以将魔咒字符 1 11、2 22 拼凑起来形成一个魔咒串 [1,2] [1, 2][1,2]。
一个魔咒串 S SS 的非空子串被称为魔咒串 S SS 的生成魔咒。
例如 S=[1,2,1] S = [1, 2, 1]S=[1,2,1] 时,它的生成魔咒有 [1] [1][1]、[2] [2][2]、[1,2] [1, 2][1,2]、[2,1] [2, 1][2,1]、[1,2,1] [1, 2, 1][1,2,1] 五种。S=[1,1,1] S = [1, 1, 1]S=[1,1,1] 时,它的生成魔咒有 [1] [1][1]、[1,1] [1, 1][1,1]、[1,1,1] [1, 1, 1][1,1,1] 三种。
最初 S SS 为空串。共进行 n nn 次操作,每次操作是在 S SS 的结尾加入一个魔咒字符。每次操作后都需要求出,当前的魔咒串 S SS 共有多少种生成魔咒。
输入格式
第一行一个整数 n nn。
第二行 n nn 个数,第 i ii 个数表示第 i ii 次操作加入的魔咒字符。
输出格式
输出 n nn 行,每行一个数。第 i ii 行的数表示第 i ii 次操作后 S SS 的生成魔咒数量。
样例
样例输入
7
1 2 3 3 3 1 2
样例输出
1
3
6
9
12
17
22
数据范围与提示
对于 10% 10\%10% 的数据,1≤n≤10 1 \leq n \leq 101≤n≤10;
对于 30% 30\%30% 的数据,1≤n≤100 1 \leq n \leq 1001≤n≤100;
对于 60% 60\%60% 的数据,1≤n≤1000 1 \leq n \leq 10001≤n≤1000;
对于 100% 100\%100% 的数据,1≤n≤100000 1 \leq n \leq 1000001≤n≤100000。
用来表示魔咒字符的数字 x xx 满足 1≤x≤109 1 \leq x \leq 10 ^ 91≤x≤109。
题解:
离线,将插入过程变化为删除过程
那就最开始就是一个长度为n的字符串让你求不重复字串个数
利用后缀数组height[i]值可以求解
那么每次删除的时候, 将位置为i的字符从 sa中删除,找到前一个存在的,和后一个存在的字符后缀串,fi,se
那么答案更新就是
ans = ans + lcp(fi,rank[i]) + lcp(ran[k],se) - lcp(fi,se);
可以用set删,存位置
#include<bits/stdc++.h> using namespace std; #pragma comment(linker, "/STACK:102400000,102400000") #define ls i<<1 #define rs ls | 1 #define mid ((ll+rr)>>1) #define pii pair<int,double> #define MP make_pair typedef long long LL; typedef unsigned long long ULL; const long long INF = 1e18+1LL; const double pi = acos(-1.0); const int N = 1e5+10, M = 1e3+20,inf = 2e9; int *ran,r[N],sa[N],height[N],wa[N],wb[N],wm[N]; bool cmp(int *r,int a,int b,int l) { return r[a] == r[b] && r[a+l] == r[b+l]; } void SA(int *r,int *sa,int n,int m) { int *x=wa,*y=wb,*t; for(int i=0;i<m;++i)wm[i]=0; for(int i=0;i<n;++i)wm[x[i]=r[i]]++; for(int i=1;i<m;++i)wm[i]+=wm[i-1]; for(int i=n-1;i>=0;--i)sa[--wm[x[i]]]=i; for(int i=0,j=1,p=0;p<n;j=j*2,m=p){ for(p=0,i=n-j;i<n;++i)y[p++]=i; for(i=0;i<n;++i)if(sa[i]>=j)y[p++]=sa[i]-j; for(i=0;i<m;++i)wm[i]=0; for(i=0;i<n;++i)wm[x[y[i]]]++; for(i=1;i<m;++i)wm[i]+=wm[i-1]; for(i=n-1;i>=0;--i)sa[--wm[x[y[i]]]]=y[i]; for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;++i) { x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++; } } ran=x; } void Height(int *r,int *sa,int n) { for(int i=0,j=0,k=0;i<n;height[ran[i++]]=k) for(k?--k:0,j=sa[ran[i]-1];r[i+k] == r[j+k];++k); } int n,a[N],san[N]; LL ans; vector<LL > an; int dp[N][30]; void Lcp_init() { for(int i = 1; i <= n; ++i) dp[i][0] = height[i]; for(int j = 1; (1<<j) <= n; ++j) { for(int i = 1; i + (1<<j) - 1 <= n; ++i) { dp[i][j] = min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]); } } } int lcp(int l,int r) { l++; if(l > r) swap(l,r); int len = r - l + 1; int k = 0; while((1<<(k+1)) <= len) k++; return min(dp[l][k],dp[r - (1<<k) + 1][k]); } int main() { scanf("%d",&n); for(int i = 1; i <= n; ++i) scanf("%d",&a[i]); for(int i = 1; i <= n; ++i) san[i] = a[i]; sort(san+1,san+n+1); int SAs = unique(san+1,san+n+1) - san - 1; for(int i = 1; i <= n; ++i) a[i] = lower_bound(san+1,san+SAs+1,a[i]) - san; int len = 0; for(int i = 1; i <= n; ++i) r[len++] = a[n - i + 1]; r[len] = 0; SA(r,sa,len+1,n+1); Height(r,sa,n); Lcp_init(); for(int i = 1; i <= len; ++i) ans = ans + i - height[i]; set<int > s; s.clear(); s.insert(-1); s.insert(inf); for(int i = 1; i <= len; ++i) s.insert(i); an.push_back(ans); for(int i = 1; i < n; ++i) { s.erase(ran[i-1]); int fi = *(--s.lower_bound(ran[i-1])); int se = *(s.lower_bound(ran[i-1])); ans = ans - (n - i + 1); if(fi != -1) ans += lcp(fi,ran[i-1]); if(se != inf) ans += lcp(ran[i-1],se); if(fi != -1 && se != inf) ans -= lcp(fi,se); an.push_back(ans); } for(int i = an.size()-1; i >= 0; --i) printf("%lld\n",an[i]); return 0; }