快轉到主要內容

CSES-1661 Subarray Sums II

標籤: 前綴和
目錄

題目連結:https://cses.fi/problemset/task/1661

題意
#

在長度為 n 的陣列中
找出有多少個總和為 x 的子陣列
相較於 I,II 多了負值,因此不具備單調性,所以不可用雙指針

思路
#

(待補)

程式碼
#

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

#define pb push_back
#define fi first
#define se second
#define INF LONG_LONG_MAX/1000
#define WA() cin.tie(0)->sync_with_stdio(0)
#define all(x) (x).begin(), (x).end()
#define int long long
#define PII pair<int, int>

signed main() { WA();
    int n, x; cin >> n >> x;
    vector<int> v(n);
    for (auto &i : v) cin >> i;
    int ans = 0;
    map<int, int> mp; mp[0] = 1;
    vector<int> prefix(n+1);
    for (int i = 1; i <= n; i++) prefix[i] = v[i-1]+prefix[i-1];
    for (int i = 1; i <= n; i++) {
        ans += mp[prefix[i]-x];
        mp[prefix[i]]++;
    }
    cout << ans << '\n';
}
Piau 的筆記本
作者
Piau 的筆記本
希望我寫下來的東西能夠長久的記在我的腦中