EOJ Monthly 2018.12 F. 日落轨迹

Posted 2021-01-25 wang9897

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了EOJ Monthly 2018.12 F. 日落轨迹相关的知识，希望对你有一定的参考价值。

题解: 对于任何一个串的前x字符内的本质不同子串我们可以直接在SAM树上得到然后我们考虑循环串的性质 (设循环节长度为l ) 则大于2*l的位置为等差数列即每增加一个字符则增加l个本质不同的子串所以对于2*l我们在后缀树上处理处理对于x>2*l我们通过等差求得

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
#include <set>
#include <map>
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define link(x) for(edge *j=h[x];j;j=j->next)
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
const int MAXN=4e5+10;
const double eps=1e-8;
const int mod1=1e9+7;
const int mod2=1e9+9;
#define ll long long
using namespace std;
struct edge{int t,v;edge*next;}e[MAXN<<1],*h[MAXN],*o=e;
void add(int x,int y,int vul){o->t=y;o->v=vul;o->next=h[x];h[x]=o++;}
ll read(){
    ll x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch==‘-‘)f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-‘0‘,ch=getchar();
    return x*f;
}
char str[MAXN];
ll ans[MAXN],sum;
int dis[MAXN],fa[MAXN],ch[MAXN][26];
int cnt,cur,rt;
ll ksm(ll a,ll b,ll c){
    ll ans=1;
    while(b){
	if(b&1)ans=ans*a%c;
	a=a*a%c;b=b>>1;
    }
    return ans;
}
int nxt[MAXN];
void Next(char str[],int len){
    int i=0,j=-1;nxt[0]=-1;
    while(i<len){
	if(j==-1||str[i]==str[j])nxt[++i]=++j;
	else j=nxt[j];
    }
}
void built(int x){
    int last=cur;cur=++cnt;dis[cur]=dis[last]+1;int p=last;
    for(;p&&!ch[p][x];p=fa[p])ch[p][x]=cur;
    if(!p)fa[cur]=rt,sum+=dis[cur];
    else{
	int q=ch[p][x];
	if(dis[q]==dis[p]+1)fa[cur]=q,sum+=dis[cur]-dis[fa[cur]];
	else{
	    int nt=++cnt;dis[nt]=dis[p]+1;
	    memcpy(ch[nt],ch[q],sizeof(ch[q]));
	    fa[nt]=fa[q];sum+=dis[nt]-dis[fa[nt]];
	    sum-=dis[q]-dis[fa[q]];fa[q]=fa[cur]=nt;
	    sum+=dis[q]-dis[fa[q]];sum+=dis[cur]-dis[fa[cur]];
	    for(;ch[p][x]==q;p=fa[p])ch[p][x]=nt;
	}
    }
}


int main(){
    int n=read(),q=read();
    inc(i,0,n-1)scanf(" %c",&str[i]),str[i+n]=str[i];
    Next(str,n);
    int t1=nxt[n];
    int m=n-t1;
    if(n%m!=0)m=n;
    cnt=rt=cur=1;
    inc(i,0,2*n-1)built(str[i]-‘a‘),ans[i]=sum;
    while(q--){
	int t=read();
	if(t<=2*m)printf("%lld
",ans[t-1]);
	else{
	    t-=2*m;ll ans1=1ll*t*m+ans[2*m-1];
	    printf("%lld
",ans1);
	}
    }
    return 0;
}

单测试点时限: 2.5 秒

内存限制: 512 MB