fork download
copy 
  1. #include <iostream>
  2. #include <bits/stdc++.h>
  3. #define Yosry ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  4. #define ll long long
  5. #define int long long
  6. using namespace std;
  7.  
  8.  
  9.  
  10. void solve() {
  11.     int n ;
  12.     cin >> n ;
  13.     vector<int> v(n);
  14.     int N = 3e5;
  15.     vector<int> freq(N);
  16.     freq[0]=1;
  17.     int ans{};
  18.     for (int i = 0; i < n; ++i) {
  19.         cin >> v[i];
  20.         if (i)v[i]+=v[i-1];
  21.         v[i]%=n;
  22.         if (v[i]<0)v[i]+=n;
  23.         ans+=freq[v[i]];
  24.         freq[v[i]]++;
  25.     }
  26.     cout << ans ;
  27.  
  28.  
  29.  
  30.  
  31. }
  32.  
  33. signed main() {
  34.     Yosry
  35.     int t=1;
  36. //    cin >> t;
  37.     while (t--) {
  38.         solve();
  39.     }
  40. }
  41.  
Success #stdin #stdout 0.01s 5552KB
comments (?)
stdin
copy 
Standard input is empty
stdout
copy 
146070
https://ideone.com/VWpkEk
language:
C++ (gcc 8.3)
created:
1 day 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!