fork download
copy 
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3. #define ll long long
  4.  
  5. ll dp[100004], a[100004];
  6.  
  7. void solve() {
  8.     ll n, k;
  9.     cin >> n >> k;
  10.     for (int i = 1; i <= n; i++){
  11.         cin >> a[i];
  12.     }
  13.     dp[0] = 0;
  14.     for (int i = 1; i <= n; i++){
  15.         dp[i] = dp[i - 1] + a[i];
  16.     }
  17.     ll tong = 0;
  18.     for (int i = 0; i < n; i++) {
  19.         ll dau = dp[i] + 1;
  20.         ll cuoi = dp[i] + k;
  21.         ll lo = lower_bound(dp + 1, dp + n + 1, dau) - dp;
  22.         ll hi = upper_bound(dp + 1, dp + n + 1, cuoi) - dp;
  23.         tong += hi - lo;
  24.     }
  25.     cout << tong;
  26. }
  27.  
  28. int main() {
  29.     solve();
  30. }
  31.  
Success #stdin #stdout 0.01s 5288KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
Standard output is empty
https://ideone.com/4tKJgg
language:
C++ (gcc 8.3)
created:
2 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!