bzoj2119 股市的预测

传送门

感觉智商莫名其妙的就变低了……写这题的时候死活想不出来……

做法其实不难……

题目要求形如ABA的串的个数，我们可以枚举A的长度，利用标记关键点的方法统计答案。设枚举到的答案为k，每k个点标记一个关键点，显然每个A只会恰好包含一个关键点，那么我们枚举每个关键点$i$，求一下它和$i+m+l$前后各能匹配到哪儿，显然答案应该加上能匹配的长度。注意为了防止重复计算需要把LCP和LCS(最长公共后缀)对$k$取min。

求LCP和LCS我是用hash写的，一开始base选的2147483647然后WA了，换成123456789和233333333倒是都能过……什么鬼……

/**************************************************************
    Problem: 2119
    User: hzoier
    Language: C++
    Result: Accepted
    Time:3664 ms
    Memory:1800 kb
****************************************************************/
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50010;
const unsigned long long base=233333333ll;
unsigned long long gethash(int,int);
int LCP(int,int);
int LCS(int,int);
unsigned long long h[maxn],pw[maxn];
long long ans=0;
int n,m,a[maxn];
int main(){
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)scanf("%d",&a[i]);
    for(int i=n;i;i--)a[i]-=a[i-1];
    n--;
    for(int i=n;i;i--)h[i]=h[i+1]*base+a[i]+base+1ull;
    pw[0]=1;
    for(int i=1;i<=n;i++)pw[i]=pw[i-1]*base;
    for(int l=1;m+(l<<1)<=n;l++){
            for(int i=l;i+m+l<=n;i+=l){
            int tmpa=min(LCS(i,i+m+l),l),tmpb=min(LCP(i,i+m+l),l);
            if(tmpa+tmpb>l)ans+=tmpa+tmpb-l;
        }
    }
    printf("%lld",ans);
    return 0;
}
inline unsigned long long gethash(int x,int l){return h[x]-h[x+l]*pw[l];}
int LCP(int x,int y){
    if(x>y)swap(x,y);
    int L=1,R=n-y+1;
    while(L<=R){
        int M=(L+R)>>1;
        if(gethash(x,M)==gethash(y,M))L=M+1;
        else R=M-1;
    }
    return R;
}
int LCS(int x,int y){
    if(x>y)swap(x,y);
    int L=1,R=x;
    while(L<=R){
        int M=(L+R)>>1;
        if(gethash(x-M+1,M)==gethash(y-M+1,M))L=M+1;
        else R=M-1;
    }
    return R;
}