[51nod1472/CF549F]取余最大值

题目大意

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

笛卡尔树

首先有一个想法，就是建立笛卡尔树。
那么对于一个节点，其作为最大值的区间是它的子树。
接下来我们考虑如何计算包含i的有多少个合法区间。
对于包含i的[l,r]，sum[r]-sum[l-1]-a[i]是k的倍数。
我们全部模k，就会变成sum[r]=sum[l-1]+a[i]或者sum[l-1]=sum[r]-a[i]
考虑枚举一个l或者r，例如枚举l，然后在右边区间查找对应数值的位置有多少个。
区间查找可以随便做，因为无修改可以考虑对每个权值建vector，然后用lower_bound和upper_bound。
现在考虑枚举的复杂度，对于i的左右子树，枚举size较小的一方，那么复杂度应该和启发式合并一致，枚举的复杂度是n log n。

#include<cstdio>
#include<algorithm>
#include<vector>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int maxn=300000+10;
vector<int> b[1000000+10];
int a[maxn],left[maxn],right[maxn],s[maxn],sum[maxn];
int i,j,k,l,t,n,m,top;
ll ans;
int read(){
    int x=0;
    char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x;
}
int find(int l,int r,int v){
    int fi=upper_bound(b[v].begin(),b[v].end(),r)-b[v].begin()-1;
    int se=lower_bound(b[v].begin(),b[v].end(),l)-b[v].begin()-1;
    return fi-se;
}
int main(){
    n=read();k=read();
    fo(i,1,n){
        a[i]=read();
        sum[i]=(sum[i-1]+a[i])%k;
    }
    fo(i,1,n){
        while (top&&a[i]>=a[s[top]]) top--;
        left[i]=s[top];
        s[++top]=i;
    }
    top=0;
    s[0]=n+1;
    fd(i,n,1){
        while (top&&a[i]>a[s[top]]) top--;
        right[i]=s[top];
        s[++top]=i;
    }
    fo(i,0,n)
        b[sum[i]].push_back(i);
    fo(i,1,n)
        if (i-left[i]<right[i]-i){
            fo(j,left[i]+1,i){
                t=(sum[j-1]+a[i]%k)%k;
                ans+=(ll)find(i,right[i]-1,t);
            }
        }
        else{
            fo(j,i,right[i]-1){
                t=((sum[j]-a[i]%k)%k+k)%k;
                ans+=(ll)find(left[i],i-1,t);
            }
        }
    printf("%I64d\n",ans-n);
}