51nod1472 cf549F 取余最大值 单调栈+可持久化线段树

Description

---

有一个长度为n的数组a，现在要找一个长度至少为2的子段，求出这一子段的和，然后减去最大值，然后对k取余结果为0。

问这样的子段有多少个。

1 ≤ n ≤ 300 000, 1 ≤ k ≤ 1 000 000
1 ≤ ai ≤ 10^9

Solution

---

等快递的时候找题写写，结果看完就会，写完已经下午了( ╯□╰ )

注意到这个最大值很烦，考虑正反两次单调栈搞出每个数控制的区间，顺便求模k意义下的前缀和
我们枚举now作为最大值求答案。注意到now影响的区间被分成左右两份，显然只需要枚举较小的区间。这样做是nlogn的
设当前枚举左区间的下标为i，那么我们需要在右区间内找到一个j使得s[j]-s[i]-a[now]=0，然后变成了求区间内一个数出现的次数。用桶是肯定挂的，写可持久化线段树就行了
实现起来细节比较多，必须有两个数的限制非常膈应人

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const int INF=0x3f3f3f3f;
const int N=300005;

struct treeNode {int l,r,sum;} t[N*21];

int l[N],r[N],a[N],q[N],s[N];
int root[N],tot;

int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}

void modify(int pre,int &now,int tl,int tr,int x) {
    t[now=++tot]=t[pre]; t[now].sum++;
    if (tl==tr) return ;
    int mid=(tl+tr)>>1;
    if (x<=mid) modify(t[pre].l,t[now].l,tl,mid,x);
    else modify(t[pre].r,t[now].r,mid+1,tr,x);
}

int query(int pre,int now,int tl,int tr,int x) {
    if (tl==tr) return t[now].sum-t[pre].sum;
    int mid=(tl+tr)>>1;
    if (x<=mid) return query(t[pre].l,t[now].l,tl,mid,x);
    return query(t[pre].r,t[now].r,mid+1,tr,x);
}

int main(void) {
    int n=read(),k=read(); modify(0,root[0],0,k,0);
    rep(i,1,n) {
        a[i]=read();
        s[i]=s[i-1]+a[i]; s[i]%=k;
        modify(root[i-1],root[i],0,k,s[i]);
    }
    q[1]=1; l[1]=1;
    for (int i=2,t=1;i<=n;i++) {
        while (t&&a[i]>a[q[t]]) t--;
        l[i]=q[t]+1;
        q[++t]=i;
    }
    q[1]=n; r[n]=n; q[0]=n+1;
    for (int i=n-1,t=1;i;--i) {
        while (t&&a[i]>=a[q[t]]) t--;
        r[i]=q[t]-1;
        q[++t]=i;
    } LL ans=0; int x,y;
    rep(i,1,n) {
        if (i-l[i]<=r[i]-i) rep(j,l[i],i) {
            x=(j==i)?root[j]:root[i-1];
            y=root[r[i]];
            ans+=query(x,y,0,k,(k+(a[i]+s[j-1])%k)%k);
        } else rep(j,i,r[i]) {
            x=(l[i]>=2)?root[l[i]-2]:0;
            y=root[std:: min(j-2,i-1)];
            ans+=query(x,y,0,k,(k+(-a[i]+s[j])%k)%k);
        }
    }
    printf("%lld\n", ans);
    return 0;
}