Submission #684800

#TimeUsernameProblemLanguageResultExecution timeMemory
684800US3RN4M3Vudu (COCI15_vudu)C++17
126 / 140
891 ms65536 KiB
#include<ext/pb_ds/assoc_container.hpp> #include<bits/stdc++.h> using namespace std; using namespace __gnu_pbds; using ll = long long; typedef tree<pair<ll, int>, null_type, less<pair<ll, int>>, rb_tree_tag, tree_order_statistics_node_update> ost; main() { int n; cin >> n; vector<int> nums(n); for(int & i : nums) cin >> i; ll p; cin >> p; ost t; ll sum = 0; ll ans = 0; t.insert({0, -1}); for(int i = 0; i < n; i++) { sum += nums[i]; ll goal = p * (i + 1); ans += i + 1 - t.order_of_key({goal - sum, -1}); t.insert({goal - sum, i}); } cout << ans << endl; }

Compilation message (stderr)

vudu.cpp:8:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
    8 | main() {
      | ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...