답안 #684804

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
684804 2023-01-22T13:35:57 Z US3RN4M3 Vudu (COCI15_vudu) C++17
126 / 140
988 ms 65536 KB
#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() {
      | ^~~~
# 결과 실행 시간 메모리 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