多多的求和计算

多多的求和计算
多多路上从左到右有N棵树（编号1～N），其中第i个颗树有和谐值Ai。
多多鸡认为，如果一段连续的树，它们的和谐值之和可以被M整除，那么这个区间整体看起来就是和谐的。
现在多多鸡想请你帮忙计算一下，满足和谐条件的区间的数量。
时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 256M，其他语言512M
输入描述：

第一行，有2个整数N和M，表示树的数量以及计算和谐值的参数。
（ 1 <= N <= 100,000, 1 <= M <= 100 ）
第二行，有N个整数Ai, 分别表示第i个颗树的和谐值。
（ 0 <= Ai <= 1,000,000,000 ）

输出描述：

共1行，每行1个整数，表示满足整体是和谐的区间的数量。

示例1
输入例子：

5 2
1 2 3 4 5

输出例子：

6

例子说明：

长度为1: [2], [4]
长度为2: 无
长度为3: [1,2,3], [3,4,5]
长度为4: [1,2,3,4], [2,3,4,5]
长度为5: 无
共6个区间的和谐值之和可以被2整除。

#include<bits/stdc++.h>
#include <vector>
using namespace std;

int main() {
    long long  n, data;
    while (cin >> n >> data) { // 注意 while 处理多个 case
        vector<long long > sumAi(data+1,0);
        long long  temp;
        long long  sum =0 ;
        sumAi[0]=1;// 数值未加入前，前缀和为0，统计起点为0的情况 //用于匹配只取一个数的情况？ //不取数的情况
        for(int i =1 ;i<=n;i++){
            cin>>temp;
            sum =(sum+temp)%data; //前缀和 
            sumAi[sum]++;

        }
        // for(int i =0 ;i<data;i++)
        //     cout<<sumAi[i]<<" ";
        // cout<<endl;
        long long  res=0;
        for(int i = 0;i<data;i++){
           res += sumAi[i]*(sumAi[i]-1)/2 ; //这里利用的是高斯公式 Cn2,在n个里面取2个；
        }
        cout<<res<<endl;
    }
}
// 64 位输出请用 printf("%lld")