[分治] 51Nod1472 Codeforces #549F. Yura and Developers

分治学傻的我只会无脑分治……
题意就是求满足 $sum \equiv max$ 的区间。反正就是按套路搞。
注意计数的数组不能每次 $clear$ ，要开时间戳。

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=300005,maxk=1000005;
int n,K,a[maxn],sum[maxn],_max[maxn],m_s[maxn];
int cnt_sum[maxk],cnt_ms[maxk],vis1[maxk],vis2[maxk],clk;
LL ans;
void Solve(int L,int R){
    if(L>=R) return;
    int mid=(L+R)>>1;
    Solve(L,mid); Solve(mid+1,R);
    clk++; sum[mid]=_max[mid]=m_s[mid]=0;
    for(int i=mid+1;i<=R;i++){
        _max[i]=max(_max[i-1],a[i]);
        sum[i]=(sum[i-1]+a[i])%K; m_s[i]=((_max[i]-sum[i])%K+K)%K;
        if(vis2[m_s[i]]!=clk) vis2[m_s[i]]=clk, cnt_ms[m_s[i]]=0;
        cnt_ms[m_s[i]]++;
    }
    for(int i=mid,sumL=0,maxL=0,p=mid;i>=L;i--){
        maxL=max(maxL,a[i]); sumL=(sumL+a[i])%K;
        while(p<R&&_max[p+1]<=maxL){
            p++; cnt_ms[m_s[p]]--; 
            if(vis1[sum[p]]!=clk) vis1[sum[p]]=clk, cnt_sum[sum[p]]=0;
            cnt_sum[sum[p]]++;
        }
        int t1=((maxL-sumL)%K+K)%K, t2=sumL;
        if(vis1[t1]==clk) ans+=cnt_sum[t1];
        if(vis2[t2]==clk) ans+=cnt_ms[t2];
    } 
}
int main(){
    scanf("%d%d",&n,&K);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    Solve(1,n);
    printf("%I64d\n",ans);
    return 0; 
}