codeforces 100792B Banana Brain's Bracelet Z-Box算法 - 线段树

传送门
题目大意：给一个循环串A，和一个字符串C，从A中截下一段B，并将B首尾相接成一个循环串使得C是B的一段。求使|B|最大。
题解：首先把A翻倍拼接得到S，前面放C，求一遍Z-Box，然后把S和C翻过来再求一遍，分别得到z1和z2。然后有两种情况，要么C是S的子串，输出|A|，要么C=C1+C2，然后S中出现了C2，隔了一段出现了C1，即C2….C1，需要满足总长度<=|A|，使这样的长度最大。这样枚举C1的起点i，把[i,i+z1[i]-1]看作C1，然后C2的终点的下标 j 以及z2[ j ]都是由取值限制的，例如z2[j]>=|C|-z1[i]。这就是一个二维数点，但是维护的是下标最值（要求最小的j），所以一种办法是二分+主席树，或者注意到如果按照权值为第一维建主席树那么就不用相减了，但是要用BIT维护。但是如果注意到j的下标递增，那么直接用线段树维护一个队列即可。

Upd：注意到这个题的最后目标是要两段的长>=|A|而位置的差小于等于什么东西，维护的使坐标的最值。因此上述按照枚举位置方法维护长度不如排序长度然后按照枚举长度的方法维护下标，后者用一个set可以很快的实现。
代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<climits>
#include<algorithm>
#define N 2000010
#define INF (INT_MAX/10-10)
using namespace std;
int n,m,z1[N],z2[N];char s[N],t[N],p[N];
inline int solve(int *z)
{
    int len=0;
    for(int i=1;i<=m;i++) p[++len]=t[i];
    p[++len]='^';
    for(int i=1;i<=n;i++) p[++len]=s[i];
    for(int i=2,l=0,r=0;i<=len;i++)
    {
        if(i>r) z[i]=(p[i]==p[1]);
        else z[i]=min(z[i-l+1],r-i+1);
        while(i+z[i]<=len&&p[i+z[i]]==p[1+z[i]]) z[i]++;
        if(i+z[i]>r&&z[i]) l=i,r=i+z[i]-1;
    }
    return 0;
}
inline int Reverse()
{
    for(int i=1;i<=n/2;i++) swap(s[i],s[n-i+1]);
    for(int i=1;i<=m/2;i++) swap(t[i],t[m-i+1]);
    return 0;
}
vector<int> q[N];int qc[N];
struct segment{
    int l,r,id;
    segment *ch[2];
}*rt;
int build(segment* &rt,int l,int r)
{
    rt=new segment;rt->l=l,rt->r=r;
    int mid=(l+r)>>1;if(l==r) return rt->id=INF;
    build(rt->ch[0],l,mid),build(rt->ch[1],mid+1,r);
    return rt->id=INF;
}
int update(segment* &rt,int p,int v,int id)
{
    int l=rt->l,r=rt->r,mid=(l+r)>>1;
    if(l==r)
    {
        if(v==1) return q[l].push_back(id),rt->id=q[l][qc[l]];
        else return qc[l]++,rt->id=((qc[l]<(int)q[l].size())?q[l][qc[l]]:INF);
    }
    if(p<=mid) update(rt->ch[0],p,v,id);
    else update(rt->ch[1],p,v,id);
    return rt->id=min(rt->ch[0]->id,rt->ch[1]->id);
}
int query(segment* &rt,int s,int t)
{
    int l=rt->l,r=rt->r,mid=(l+r)>>1,ans=INF;
    if(s<=l&&r<=t) return rt->id;
    if(s<=mid) ans=min(ans,query(rt->ch[0],s,t));
    if(mid<t) ans=min(ans,query(rt->ch[1],s,t));
    return ans;
}
int main()
{
    scanf("%s%s",s+1,t+1);
    n=(int)strlen(s+1),m=(int)strlen(t+1);
    if(n<m) return !printf("-1\n");
    for(int i=n+1;i<2*n;i++) s[i]=s[i-n];
    n=2*n-1,solve(z1),Reverse(),solve(z2);
    for(int i=1,j=m+2;i<=n;i++,j++) z1[i]=z1[j];
    for(int i=1,j=m+2;i<=n;i++,j++) z2[i]=z2[j];
    for(int i=1;i<=n/2;i++) swap(z2[i],z2[n-i+1]);
    for(int i=1;i<=n;i++)
        if(z1[i]>=m) return !printf("%d\n",n/2+1);
    int ans=-1;build(rt,1,n+1);
    for(int i=1,t,k;i<=n;i++)
    {
        k=i-(n+1)/2+m-2;if(k>0&&z2[k]) update(rt,z2[k],-1,i);
        t=query(rt,m-z1[i],n+1);if(z2[i]) update(rt,z2[i],1,i);
        if(t<INF) ans=max(ans,i+z1[i]-1-t+min(z2[t],m-z1[i]));
    }
    printf("%d\n",ans);return 0;
}