TakahiroNakamori

TakahiroNakamori

中森崇博(ナカモリタカヒロ)

12 /
26
2021
 

Count Interval

問題

D – Count Interval
プログラミング初級者から上級者まで楽しめる、競技プログラミングコンテストサイト「AtCoder」。オンラインで毎週開催プログラミングコンテストを開催しています。競技…

記録

解法

コード例

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using P = pair<int, int>;
#define rep(i, n) for(int i = 0; i < (n); ++i)
// using mint = modint998244353;
// using mint = modint1000000007;
// const int mod = 1000000007;
// const ll INF = 1LL << 60;
// const int INF = 1001001001;

int main() {
  ll N, K;
  cin >> N >> K;

  vector<ll> A(N);
  rep(i, N) { cin >> A[i]; }

  vector<ll> sum(N + 1);
  rep(i, N) { sum[i + 1] = sum[i] + A[i]; }

  map<ll, ll> mp;
  ll ans = 0;
  for(int i = 1; i <= N; ++i) {
    mp[sum[i - 1]]++;
    ans += mp[sum[i] - K];
  }

  cout << ans << endl;
  return 0;
}