AT_abc233_d .Count Interval

题目链接：D - Count Interval

---

思路：

前缀和，根据所求的 k = 区间和 sum[r] - sum[l-1] 得

sum[l-1] = sum[r] - k 

从左到右遍历 1~n ，当遍历到点 i 时，要累加在 i 的左边有多少个满足 sum[l-1] = sum[r] - k 的 前缀和

用 map 记录即可

---

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<unordered_set>
#include<unordered_map>
#include<bitset>
#include<tuple>
#define inf 72340172838076673
#define int long long
#define endl '\n'
#define F first
#define S second
#define  mst(a,x) memset(a,x,sizeof (a))
using namespace std;
typedef pair<int, int> pii;

const int N = 200086, mod = 998244353;

int n, k;
int a[N];
int res = 0;

void solve() {

    cin >> n >> k;
    map<int, int> mp;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        a[i] += a[i - 1];
    }
    
    for (int i = 1; i <= n; i++) {
        mp[a[i - 1]]++;
        res += mp[a[i] - k];
    }
    
    cout << res << endl;
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    
    int T = 1;
// cin >> T;
    while (T--) solve();
    
    return 0;
}