Submission #256546

# Submission time Handle Problem Language Result Execution time Memory
256546 2020-08-02T20:19:59 Z ElyesChaabouni Vudu (COCI15_vudu) C++14
70 / 140
876 ms 65540 KB
#include<bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp> // Common file
#include <ext/pb_ds/tree_policy.hpp> // Including tree_order_statistics_node_update#define eps 1e-9
#define MOD1 998244353
#define MOD2 1000000007
#define INV_10 299473306
#define INF 1000000001
#define PI 3.14159265358979323846
#define ordered_set tree<pair<long long, int>, null_type,less<pair<long long, int> >, rb_tree_tag,tree_order_statistics_node_update> 
using namespace std;
using namespace __gnu_pbds;
int main()
{
	ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	int n;
	cin >> n;
	long long a[n];
	ordered_set s;
	//cout << n << '\n';
	for(int i = 0; i < n; i++)
		cin >> a[i];
	long long p;
	cin >> p;
	long long sum=0, cu=0, ans=0;
	long long here[n];
	for(int i = n-1; i >= 0; i--)
	{
		sum+=a[i];
		here[i]=sum-cu;
		s.insert(make_pair(here[i], i));
		//cout << here[i] << ' ';
		cu+=p;
	}
	//cout << '\n';
	cu=0;
	long long pr=0;
	for(int i = n-1; i >= 0; i--)
	{
		if(s.lower_bound(make_pair(p+pr-cu, -1))!=s.end())
			ans+=i+1-s.order_of_key(*s.lower_bound(make_pair(p+pr-cu, -1)));
		//cout << s.order_of_key(*s.lower_bound(make_pair(p+pr-cu, -1))) << '\n';
		s.erase(make_pair(here[i], i));
		pr+=a[i];
		cu+=p;
	}
	cout << ans << '\n';
}
# Verdict Execution time Memory Grader output
1 Correct 6 ms 1024 KB Output is correct
2 Correct 5 ms 768 KB Output is correct
3 Correct 4 ms 768 KB Output is correct
4 Runtime error 587 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
5 Correct 641 ms 48632 KB Output is correct
6 Runtime error 567 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
7 Runtime error 472 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
8 Correct 876 ms 65536 KB Output is correct
9 Runtime error 475 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
10 Runtime error 485 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)