【做一道算一道】 多多的求和计算

多多的求和计算

多多路上从左到右有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整除。

思路

第一反应是把情况全部列举出来，两个for循环嵌套,滑动窗口，只有长度为1的特殊处理一下就行，不过数据很大的情况下超时。
搞不出来，查了一下，都说是前缀和，取余的情况来算哈希。有个佬的解释是：
map的作用 map 里面是sum 跟m取余之后的余数相同值的次数 ， 如果sum在累加过程中两次取余相等，那么第一次取余的sum和第二次取余的sum中间加的数列 一定是m的倍数，那么这个序列的和就一定是m的整数倍。

关于里面map[0]=1，我的理解是取余为0意为整除，那么不需要两次取余相等就可以有和谐区间，所以第一次访问map[0]的时候就可以计数，所以初始值是1
而map的其他索引对应的就是其他取余的结果的数量，起始值为0，所以第一次访问时为0，之后第二次访问才加1（有两次取余相同，存在和谐区间），第三次则跟着加1（离第二次访问的地方构成和谐区间，离第一次构成不了，多一次取余）。由此类推。
不确定这个想法是不是对的，欢迎指教。

代码

超时版本：

#include <iostream>
#include <vector>
using namespace std;

int main() {
    int n, m;
    cin >> n >> m;
    int loop = n;
    vector<long long> num;
    while (loop--) {
        long long temp;
        cin >> temp;
        num.push_back(temp);
    }
    int cnt = 0;
    long long sum = 0;
    for (int i = 0; i < n; i++) {
        sum=0;
        for (int left = 0, right = 0; left <= right && left >= 0 && right < n;) {
            sum += num[right];
            if (i == 0 && right - left == i) {
                if (sum % m == 0) {
                    cnt++;
                }
                sum -= num[left];
                left++;
                right++;
            } else if (i != 0 && right - left == i) {
                if (sum % m == 0) {
                        cnt++;
                }
                    sum -= num[left];
                    left++;
                }
                if (right - left < i)
                    right++;
            }
        }
        cout<<cnt;
    }
// 64 位输出请用 printf("%lld")

前缀和版本

#include <iostream>
#include <vector>
using namespace std;
int main(){
    int n, m;
    cin>>n>>m;
    int num[n];
    for (int i = 0; i < n; ++i){
        cin>>num[i];
    }
    vector<int> map(m,0);//下标表示前缀和 % m后的值，map[i]表示取余后值为i的个数；
    map[0] = 1; //能整除的情况，不需要第二次碰到才有和谐区间，出现即可cnt++，所以第一次访问就加1
    long sum = 0;
    long cnt = 0;
    for (int i = 0; i < n; ++i){
        sum += num[i]; 
        int index = sum % m;
        cnt += map[index]++;
    }
    cout<<cnt<<endl;
    return 0;
}