蓝桥杯复习之前缀和

题目链接：https://www.luogu.com.cn/problem/P8649

思路：
    看到区间和，第一反应肯定是前缀和，我们求出前缀和后对前缀和数组每一个值模k，然后对一个数组的值查看前面有几个相同的，举个例子：

    样例中前缀和数组a取模后为：1    1    0    0    1

    下标依次为1 2 3 4 5

    当题目所求子序列以a5结尾时，已知a5=1且a5前面与a5相同的值有2个，则以a5结尾可以获得2个满足题目要求的区间
    
    如果以a3结尾则是一个，因为实际上有一个a0=0。
 

#include<bits/stdc++.h>using namespace std;using ll = long long;
const ll N = 3e5 + 5;
const ll mod = 1e9 + 7;
int gcd(int a, int b) {return b ? gcd(b, a % b) : a;
}void solve() {int n; cin >> n;ll k; cin >> k;vector<ll>a(n + 1);vector<ll>pre(n + 1);vector<ll>cnt(n + 1);for (int i = 1; i <= n; i++) {cin >> a[i];pre[i] = pre[i - 1] + a[i];}ll ans = 0;cnt[0] = 1;for (int i = 1; i <= n; i++) {ans += cnt[pre[i] % k];cnt[pre[i] % k]++;}cout << ans << '\n';
}signed main() {ios::sync_with_stdio(false);cin.tie(0);std::cout.tie(0);int t1 = 1;//cin >> t1;while (t1--)solve();return 0;
}