Submission #684804

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
684804	2023-01-22T13:35:57 Z	US3RN4M3	Vudu (COCI15_vudu)	C++17	126 / 140	988 ms	65536 KB

vudu

#include<ext/pb_ds/assoc_container.hpp>
#include<bits/stdc++.h>
using namespace std;
using namespace __gnu_pbds;
using ll = long long;
typedef tree<int, null_type, less<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;
	vector<pair<ll, int>> vals(n);
	ll sum = 0;
	ll ans = 0;
	for(int i = 0; i < n; i++) {
		sum += nums[i];
		ll goal = p*(i + 1);
		vals[i] = {sum - goal, i};
		if(sum >= goal) ans++;
	}
	sort(vals.begin(), vals.end());
	ost t;
	for(int i = 0; i < n; i++) {
		ans += t.order_of_key(vals[i].second);
		t.insert(vals[i].second);
	}
	cout << ans << endl;
}

Compilation message

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

Subtask #1

126.0 / 140.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	5 ms	724 KB	Output is correct
2	Correct	4 ms	596 KB	Output is correct
3	Correct	3 ms	596 KB	Output is correct
4	Correct	988 ms	65536 KB	Output is correct
5	Correct	505 ms	36556 KB	Output is correct
6	Correct	875 ms	57104 KB	Output is correct
7	Correct	836 ms	59440 KB	Output is correct
8	Correct	729 ms	51568 KB	Output is correct
9	Runtime error	879 ms	65536 KB	Execution killed with signal 9
10	Correct	823 ms	57964 KB	Output is correct