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

来源：互联网发布：头影测量软件编辑：程序博客网时间：2024/04/25 07:27

分治学傻的我只会无脑分治……
题意就是求满足 sum≡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; }

阅读全文

0 0