51nod1600(LCT,树链剖分,后缀自动机)

传送门:http://www.51nod.com/Challenge/Problem.html#!#problemId=1600
题面描述:
对于一个字符串|S|,我们定义fail[i],表示最大的x使得S[1…x]=S[i-x+1…i],满足(x<i)
显然对于一个字符串,如果我们将每个0<=i<=|S|看成一个结点,除了i=0以外i向fail[i]连边,这是一颗树的形状,根是0
我们定义这棵树是G(S),设f(S)是G(S)中除了0号点以外所有点的深度之和,其中0号点的深度为-1
定义key(S)等于S的所有非空子串S’的f(S’)之和
给定一个字符串S,现在你要实现以下几种操作:
1.在S最后面加一个字符
2.询问key(S)

善良的出题人不希望你的答案比long long大,所以你需要将答案对1e9+7取模

∣ S ∣ ≤ 1 0 5 |S|\le10^5 S105


#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define ll long long
#define pb push_back
#define SZ(x) ((int)(x.size()))
const int p = 1e9+7;
int n,poss[201000],ans[201000];
char s[201000];
inline int calc(int a,int b){return (a+b)%p;}
inline int del(int a,int b){return (a-b+p)%p;}
inline int mul(int a,int b){return 1ll*a*b%p;}
namespace SAM{
	int cnt,last;
	int tr[401000][26],mx[401000],par[401000];
	void extand(int x,int id){
	    int np = ++cnt , p = last;
	    mx[np] = mx[p] + 1;last = np; poss[id] = np;
	    while(p && !tr[p][x]) tr[p][x] = np , p = par[p];
	    if(!p) par[np] = 1;
	    else{
	        int q = tr[p][x];
	        if(mx[q] == mx[p] + 1) par[np] = q;
	        else{
	            int nq = ++cnt;mx[nq] = mx[p] + 1;
	            rep(i,0,25) tr[nq][i] = tr[q][i];
	            par[nq] = par[q];par[q] = par[np] = nq;
	            while(p && tr[p][x] == q) tr[p][x] = nq , p = par[p];
	        }
	    }
    }	
}using namespace SAM;
namespace Tree{
	#define ls (rt<<1)
	#define rs (rt<<1|1)
	vector<int>G[401000];
	int sum1[1601000],sum2[1601000],det[1601000];
	int fa[401000],dep[401000],son[401000],sz[401000];
	int pos[401000],top[401000],tot,tmp[401000];
	void insert(int x,int y){G[x].pb(y);G[y].pb(x);}
	void dfs1(int x,int f){
		dep[x] = dep[f]+1; fa[x] = f;
		rep(i,0,SZ(G[x])-1) if(G[x][i] != f){
			int y = G[x][i];
			dfs1(y,x);
			sz[x] += sz[y];
			if(sz[y] > sz[son[x]]) son[x] = y;
		}
		sz[x]++;
	}
	void dfs2(int x){
		top[x] = son[fa[x]] == x ? top[fa[x]] : x;
		pos[x] = ++tot; tmp[tot] = x;
		if(son[x]) dfs2(son[x]);
		rep(i,0,SZ(G[x])-1) if(G[x][i] != fa[x] && G[x][i] != son[x]) 
		    dfs2(G[x][i]);
	}
	void pushdown(int rt){
		if(det[rt]){
			sum1[ls] = calc(sum1[ls] ,  mul( det[rt] , sum2[ls] ) );
			det[ls] = calc(det[ls],det[rt]);
			sum1[rs] = calc(sum1[rs] ,  mul( det[rt] , sum2[rs] ) );
			det[rs] = calc(det[rs],det[rt]);
			det[rt] = 0;
		}
	}
	void update(int rt){sum1[rt] = calc(sum1[ls],sum1[rs]);}
	void build(int rt,int l,int r){
		if(l == r){
			sum2[rt] = SAM::mx[tmp[l]] - SAM::mx[SAM::par[tmp[l]]];
			return;
		}
		int mid = (l+r)>>1;
		build(ls,l,mid); build(rs,mid+1,r);
		sum2[rt] = calc(sum2[ls],sum2[rs]);
	}
	int query(int rt,int l,int r,int x,int y){
		if(x <= l && r <= y) return sum1[rt];
		pushdown(rt);
		int mid = (l+r)>>1,ans = 0;
		if(mid >= x) ans = calc(ans,query( ls , l , mid , x , y ) );
		if(mid <  y) ans = calc(ans,query( rs , mid + 1 , r , x , y ));
		return ans;
	}
	int ask(int x,int y){
		int ans = 0;
		while(top[x] != top[y]){
			if(dep[top[x]] < dep[top[y]]) swap(x,y);
			ans = calc( ans , query( 1 , 1 , tot , pos[top[x]] , pos[x] ) );
			x = fa[top[x]];
		}
		if(dep[x] > dep[y]) swap(x,y);
		ans = calc( ans , query(1,1,tot,pos[x] , pos[y]) );
		return ans;
	}
	void add(int rt,int l,int r,int x,int y){
		if(x <= l && r <= y){
			sum1[rt] = calc(sum1[rt],sum2[rt]);
			det[rt]++;
			return;
		}
		pushdown(rt);
		int mid = (l+r)>>1;
		if(mid >= x) add(ls,l,mid,x,y);
		if(mid <  y) add(rs,mid+1,r,x,y);
		update(rt);
	}
	void add(int x,int y){
		while(top[x] != top[y]){
			if(dep[top[x]] < dep[top[y]]) swap(x,y);
			add(1,1,tot,pos[top[x]] ,pos[x]);
			x = fa[top[x]];
		}
		if(dep[x] > dep[y]) swap(x,y);
		add(1,1,tot,pos[x],pos[y]);
	}
}using namespace Tree;
int main(){
	freopen("string.in","r",stdin);
	freopen("string.out","w",stdout);
	scanf("%d",&n); scanf("%s",s+1);
	SAM::cnt = SAM::last = 1;
	rep(i,1,n) SAM::extand(s[i]-'a',i);
	rep(i,2,SAM::cnt) Tree::insert(i,SAM::par[i]);
	Tree::dfs1(1,0); Tree::dfs2(1);
	Tree::build(1,1,Tree::tot);
	rep(i,1,n){
		ans[i] = Tree::ask( 1 , poss[i] );
		ans[i] = calc(ans[i] , del( mul(2,ans[i-1]) , ans[i-2] ) );
		Tree::add( 1 , poss[i] );
		printf("%d\n",ans[i]);
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值