fork download
copy 
  1. #include <bits/stdc++.h>
  2. #define VuDucNam ios_base::sync_with_stdio(false);
  3. #define Lop9a6 cin.tie(NULL);
  4. #define THCSPhamVanDong cout.tie(NULL);
  5. #define ll long long
  6. #define el cout << '\n'
  7. #define sz(a) (ll) a.size()
  8. #define all(a) a.begin(), a.end()
  9. #define fi first
  10. #define se second
  11. #define fill(a, x) memset(a, x, sizeof(a))
  12. #define file(name)                         \
  13.     if (fopen(name ".inp", "r"))           \
  14.     {                                      \
  15.         freopen(name ".inp", "r", stdin);  \
  16.         freopen(name ".out", "w", stdout); \
  17.     }
  18. #define TIME (1.0 * clock() / CLOCKS_PER_SEC)
  19. #define RUNTIME cerr << "\nRuntime: " << TIME << "s.\n"
  20.  
  21. using namespace std;
  22.  
  23. const int N5 = 1e5 + 5;
  24. const ll MOD = 1e9 + 7;
  25. const ll inf = LLONG_MAX;
  26.  
  27. int n;
  28. ll s,pre[N5],cnt;
  29. multiset<ll> si;
  30.  
  31. void solve()
  32. {
  33.     cin >> n >> s;
  34.     for(int i=1;i<=n;++i){
  35.         ll x;
  36.         cin >> x;
  37.         pre[i]=pre[i-1]+x;
  38.         si.insert(pre[i]);
  39.     }
  40.     for(int i=1;i<=n;++i){
  41.         auto it1 = si.lower_bound(pre[i]-s);
  42.         auto it2 = si.upper_bound(pre[i]+s);
  43.         cnt+= distance(si.begin(),it1) + distance(it2,si.end());
  44.     }
  45.     cout << cnt;
  46.     el;
  47. }
  48.  
  49. int main()
  50. {
  51.     VuDucNam Lop9a6 THCSPhamVanDong
  52.     file("code");
  53.     solve();
  54.     RUNTIME;
  55.     return 0;
  56. }
Success #stdin #stdout #stderr 0.01s 5288KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
0
stderr
copy 
Runtime: 0.006122s.
https://ideone.com/c3bhAL
language:
C++ (gcc 8.3)
created:
31 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!