【2017年浙江工业大学大学生程序设计迎新赛决赛】D 序列 【离散化+分块+莫队】

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld
题目描述
读入一个长度为n的整数数列a1,a2,…,an，以及一个整数K。
q组询问。
每组询问包含一个二元组(l, r), 其中1≤l≤r≤ n,
求所有满足以下条件的二元组(l2, r2)的数目:
1: 1≤l≤l2≤r2≤r≤n,
2: 是k的倍数。

输入描述:
第一行一个整数T，表示有T组数据。
对于每组数据，
第一行输入3个整数, n, q, k，
接下来一行输入n个整数，a1,a2,…,an，
接下来q行，每行输入2个整数 l, r
输出描述:
对于每一个询问，输出满足条件的二元组(l2,r2)的数目。
示例1
输入

1
5 2 4
4 1 4 8 8
2 2
2 4
输出

0
3
说明

当(l, r)为(2, 4)时，
有
(l2,r2)=(3,4),a3+a4=12,是4的倍数
(l2,r2)=(3,3),a3=4,是4的倍数
(l2,r2)=(4,4),a4=8,是4的倍数
备注:
1≤ n,q ≤5×104,
1≤ l ≤ r ≤n,
1≤ ai, k ≤109,
保证∑n ≤6×104,∑q ≤6×104

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN = 5e4+11;

int sum[MAXN];
int HASH[MAXN],sz;    //  离散
int  limit;  //分块

struct Query{
    int l,r,id;
    bool operator <(const Query &b){
        if( (l-1)/limit != (b.l-1)/limit )
            return (l-1)/limit < (b.l-1)/limit;
        return r<b.r;
    }
}Q[MAXN+1];
LL Cnt[MAXN+1],Ans[MAXN]; LL ans;
void UpDate(int x){
    int y=sum[x];
    ans-=Cnt[y]*Cnt[y]-Cnt[y];
    Cnt[y]++;
    ans+=Cnt[y]*Cnt[y]-Cnt[y];
}
void Delete(int x){
    int y=sum[x];
    ans-=Cnt[y]*Cnt[y]-Cnt[y];
    Cnt[y]--;
    ans+=Cnt[y]*Cnt[y]-Cnt[y];
}

void solve(int m){
    for(int i=1;i<=m;i++){ scanf("%d%d",&Q[i].l,&Q[i].r);Q[i].id=i;  }
    sort(Q+1,Q+1+m);
    memset(Cnt,0,sizeof(Cnt));
    ans=0; int L=1,R=0;
    for(int i=1;i<=m;i++){
        int id=Q[i].id;
        while(R<Q[i].r) UpDate(++R);
        while(L>Q[i].l) UpDate(--L);
        while(R>Q[i].r) Delete(R--);
        while(L<Q[i].l) Delete(L++);
        UpDate(--L);
        Ans[id]=ans;
        Delete(L++);
    }
    for(int i=1;i<=m;i++) printf("%lld\n",Ans[i]/2);
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        int n,m,k;scanf("%d%d%d",&n,&m,&k);
        limit=ceil(sqrt(n*1.0));  sz=0;
        HASH[sz++]=0; sum[0]=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&sum[i]);
            sum[i]=sum[i-1]+sum[i];
            sum[i]%=k;  HASH[sz++]=sum[i];
        }
        sort(HASH,HASH+sz); sz=unique(HASH,HASH+sz)-HASH;
        for(int i=0;i<=n;i++) sum[i]=lower_bound(HASH,HASH+sz,sum[i])-HASH+1;
        //for(int i=0;i<=n;i++) printf("%d ",sum[i]);
        solve(m);  // 莫队
    }
return 0;
}