The 2024 ICPC Asia East Continent Online Contest (II) C. Prefix of Suffixes（kmp border性质）

题目

思路来源

daydreamers群友

kmp的正确用法，或者说border理论及其应用 - ShanLunjiaJian's Blog

https://www.luogu.com/article/e08ebmz6 ix35 字符串好文

题解

离线的话可以考虑用exkmp（Z函数），但是本题强制在线，所以只能考虑用kmp了

根据题目要求的式子，假设当前匹配到了i，也就是从i-1转移到i的时候，

是需要考虑增量贡献的，也就是新增了一个a[i]的时候，

考虑哪些数组b的位置能与其形成(i,j)对的贡献

推导不难发现，满足[x,i]与[1,i-x+1]完全相同的x，是有贡献的x

暴力的做法是将这些x找到，然后将这些b[x]相加得到sum，增量贡献就是sum*a[i]

但是因为i是动态变化的，x的集合也是在动态变化的，现算复杂度难以满足需求

所以考虑用总的[1,i]的b之和，减去已经没贡献的，也就是[x,i]与前缀的lcp小于i-x+1的，

没贡献的x，一定会在某个x<=j<=i的位置变得没有贡献，只被减一遍之后，后面就不用减了

然后考虑怎么找到这样的x，

一开始想的做法就是只考虑s[i]!=s[j]的时候回跳，也就是如下图代码

但是这样是不对的，群友给了一个反例，后来我想了想确实是这样，

例如：

S=abab

T=abac

S与T的两个前缀aba是始终匹配上的，

但是S[1]=b实际与T[0]是不能匹配的，

应该也从sum里减掉，这里没有减

所以就是即使s[i]=s[j]求出nex数组之后，也需要继续回跳

但是暴力这样做的复杂度是不对的，极限情况考虑aaaaaaa就可以了

所以需要引入border的性质，以下直接粘ix35老师的博客了，

自己再写一遍的话说不准哪还会写错了…

总之性质就是，

字符串的border形成了O(logn)个值域不交的等差数列，

形如[l1,l1+d1,l1+2*d1,...][l2,l2+d2,l2+2*d2,...],...,[lx,lx+dx,lx+2*dx,...]……

利用这个性质，再结合周期与border的定义，不难发现

一个next位置对应的border与下一个字符匹配上了，

那么和它在同一个等差数列内的都与下一个字符匹配上了，

所以，暴力回跳next数组的时候就应该直接跳到等差数列的链头，

否则会被aaaaaaaa这种数据卡TLE

具体可以看代码，

用top数组维护当前所在的等差数列的链头，

而nex数组仍然是kmp fail的含义

要么失配跳过至少一段等差数列，

要么匹配，匹配又分当前是否在链头，

在的话就跳到前一个链的链尾，否则跳到当前链的链头

所以复杂度O(nlogn)

代码

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
const int N=3e5+10,mod=998244353;
int n,j,nex[N],fa[N],top[N];
ll sum,ans,s[N],a[N],b[N];
bool used[N];
int main(){
    sci(n);
    j=nex[0]=top[0]=-1;
    rep(i,0,n-1){
        scanf("%lld%lld%lld",&s[i],&a[i],&b[i]);
        int x=j;
        s[i]=(s[i]+ans)%n;
        while(j!=-1 && s[i]!=s[j]){//[0,j)是已经和[i-j,i)完全匹配了
            j=nex[j];
        }
		nex[i+1]=++j;
        if(i+1-nex[i+1]==nex[i+1]-nex[nex[i+1]])top[i+1]=top[nex[i+1]];
        else top[i+1]=nex[i+1];

        sum+=b[i];
        while(x!=-1){
            if(s[i]==s[x]){
                if(s[nex[x]]==s[i])x=top[x];//直接跳到这个等差数列的链头
                else x=nex[x];//等差数列临界，跳到前一个等差数列
            }
            else{//至少跳了一个等差数列
                sum-=b[i-x];
                x=nex[x];
            }
        }
        ans+=sum*a[i];
        printf("%lld\n",ans);
    }
    return 0;
}