Submission #256575

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
256575	2020-08-02T21:11:10 Z	ElyesChaabouni	Vudu (COCI15_vudu)	C++17	112 / 140	1000 ms	65540 KB

vudu

#pragma GCC optimize("O3")
#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
#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;
	//scanf("%d", &n);
	int a[n];
	ordered_set s;
	//cout << n << '\n';
	for(int i = 0; i < n; i++)
		//scanf("%d", &a[i]);
		cin >> a[i];
	int p;
	//scanf("%d", &p);
	cin >> p;
	long long sum=0, cu=0, ans=0;
	for(int i = n-1; i >= 0; i--)
	{
		sum+=a[i];
		s.insert(make_pair(sum-cu, i));
		//cout << here[i] << ' ';
		cu+=p;
	}
	//cout << '\n';
	cu=0;
	sum=0;
	for(int i = n-1; i >= 0; i--)
	{
		ordered_set::iterator it=s.lower_bound(make_pair(p+sum-cu, -1));
		if(it!=s.end())
			ans+=i+1-s.order_of_key(*it);
		//cout << s.order_of_key(*s.lower_bound(make_pair(p+pr-cu, -1))) << '\n';
		sum+=a[i];
		s.erase(make_pair(sum-cu, i));
		cu+=p;
	}
	cout << ans << '\n';
	//printf("%lld", ans);
}

Compilation message

vudu.cpp:3:0: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
 #pragma GCC optimization ("O3")
 
vudu.cpp:4:0: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
 #pragma GCC optimization ("unroll-loops")

Subtask #1

112.0 / 140.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	7 ms	896 KB	Output is correct
2	Correct	5 ms	768 KB	Output is correct
3	Correct	4 ms	768 KB	Output is correct
4	Execution timed out	1077 ms	65536 KB	Time limit exceeded
5	Correct	589 ms	36736 KB	Output is correct
6	Correct	963 ms	57336 KB	Output is correct
7	Correct	852 ms	59616 KB	Output is correct
8	Correct	914 ms	51832 KB	Output is correct
9	Runtime error	527 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
10	Correct	978 ms	58152 KB	Output is correct