求助SA
查看原帖
求助SA
130387
_Anchor楼主2020/12/30 14:16

#1#2 WA了...

不知道为什么...

求差错/kel

#include<bits/stdc++.h>
using namespace std;
template <typename T>
inline void read(T &x){
	x=0;char ch=getchar();bool f=false;
	while(!isdigit(ch)){if(ch=='-'){f=true;}ch=getchar();}
	while(isdigit(ch)){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
	x=f?-x:x;
	return ;
}
template <typename T,typename... Args> inline void read(T& t, Args&... args){read(t);read(args...);}
template <typename T>
inline void write(T x){
	if(x<0) putchar('-'),x=-x;
	if(x>9) write(x/10);
	putchar(x%10^48);
	return ;
}
template <typename T>
inline void print(T x){write(x),putchar(' ');}
#define ll long long
#define ull unsigned long long
#define inc(x,y) (x+y>=MOD?x+y-MOD:x+y)
#define dec(x,y) (x-y<0?x-y+MOD:x-y)
#define min(x,y) (x<y?x:y)
#define max(x,y) (x>y?x:y)
const int N=1e6+5,M=1e6+5,MOD=1e9+7;
char str[N];
ll ans;
int n,m=127,rk[N],sa[N],he[N],tmp[N],cnt[N];
void SA(){
	for(int i=1;i<=n;i++) cnt[rk[i]=str[i]]++;
	for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
	for(int i=n;i;i--) sa[cnt[str[i]]--]=i;
	for(int k=1;k<n;k<<=1){
		int q=0;
		for(int i=n-k+1;i<=n;i++) tmp[++q]=i;
		for(int i=1;i<=n;i++) if(sa[i]-k>0) tmp[++q]=sa[i]-k;
		for(int i=0;i<=m;i++) cnt[i]=0;
		for(int i=1;i<=n;i++) cnt[rk[i]]++;
		for(int i=1;i<=n;i++) cnt[i]+=cnt[i-1];
		for(int i=n;i;i--) sa[cnt[rk[tmp[i]]]--]=tmp[i],tmp[i]=0;
		swap(rk,tmp);m=rk[sa[1]]=1;
		for(int i=2;i<=n;i++){
			if(tmp[sa[i]]!=tmp[sa[i-1]] or tmp[sa[i]+k]!=tmp[sa[i-1]+k]) m++;
			rk[sa[i]]=m;
		}
	}
	for(int i=1,k=0;i<=n;i++){
		if(k) k--;
		while(str[i+k]==str[sa[rk[i]-1]+k]) k++;
		he[rk[i]]=k;
	}
	return ;
}
int main(){
	read(n);
	scanf("%s",str+1);
	SA();
	for(int i=1;i<=n;i++) ans+=he[rk[i]];
	write(1ll*n*(n+1)/2-ans);
	system("Pause");
	return 0;
}
2020/12/30 14:16
加载中...