oj.uz

Submission #734367

# Submission time Handle Problem Language Result Execution time Memory
734367 2023-05-02T10:31:43 Z Magikarp4000 Horses (IOI15_horses) C++17
17 / 100
 1500 ms 51984 KB 
#include "horses.h"
#include <bits/stdc++.h>
using namespace std;
#define OPTM ios_base::sync_with_stdio(0); cin.tie(0);
#define INF int64_t(1e9+7)
#define ln '\n' 
#define ll long long
#define ull unsigned long long
#define ui unsigned int
#define us unsigned short
#define FOR(i,s,n) for (int i = s; i < n; i++)
#define FORR(i,n,s) for (int i = n; i > s; i--)
#define FORX(u, arr) for (auto u : arr)
#define PB push_back
#define in(v,x) (v.find(x) != v.end())
#define F first
#define S second
#define PII pair<int, int>
#define PLL pair<ll, ll>
#define UM unordered_map
#define US unordered_set
#define PQ priority_queue
#define ALL(v) v.begin(), v.end()
const ll LLINF = 1e18+1;

const int MAXN = 5e5+1, LOG = 31;
const ll MAXX = 1e9;
int n;
ll x[MAXN], y[MAXN];
set<int> a;
ll st[MAXN*4], st1[MAXN*4];

void update(int s, int tl, int tr, int idx, ll val) {
	if (tl == tr) st[s] = val;
	else {
		int tm = (tl+tr)/2;
		if (idx <= tm) update(s*2,tl,tm,idx,val);
		else update(s*2+1,tm+1,tr,idx,val);
		st[s] = max(st[s*2], st[s*2+1]);
	}
}

ll query(int s, int tl, int tr, int l, int r) {
	if (tl == tr) return st[s];
	else {
		int tm = (tl+tr)/2;
		ll cur = -INF;
		if (l <= tm) cur = max(cur, query(s*2,tl,tm,l,r));
		if (r > tm) cur = max(cur, query(s*2+1,tm+1,tr,l,r));
		return cur;
	}
}

void update1(int s, int tl, int tr, int idx, ll val) {
	if (tl == tr) st1[s] = val;
	else {
		int tm = (tl+tr)/2;
		if (idx <= tm) update1(s*2,tl,tm,idx,val);
		else update1(s*2+1,tm+1,tr,idx,val);
		st1[s] = (st1[s*2]*st1[s*2+1])%INF;
	}
}

ll query1(int s, int tl, int tr, int l, int r) {
	if (tl == tr) return st1[s];
	else {
		int tm = (tl+tr)/2;
		ll cur = 1ll;
		if (l <= tm) cur = (cur*query1(s*2,tl,tm,l,r))%INF;
		if (r > tm) cur = (cur*query1(s*2+1,tm+1,tr,l,r))%INF;
		return cur;
	}
}

int num = 0;

int calc() {
	if (a.size() == 0) return query(1,0,n-1,0,n-1);
	vector<int> v;
	int cnt = 0;
	for (auto it = a.rbegin(); it != a.rend(); it++) {
		v.PB(*it);
		cnt++;
		if (cnt >= LOG) break;
	}
	reverse(ALL(v));
	int vn = v.size();
	ll res = 0, xval = 1, pos = 0, prev = -1, yval = 0, ygood = 0, yprev = -1;
	FOR(i,0,vn) {
		int en = i == vn-1 ? n-1 : v[i+1]-1;
		xval *= x[v[i]];
		if (xval > MAXX) {
			prev = pos;
			yprev = yval;
			xval = x[v[i]];
			yval = query(1,0,n-1,v[i],en);
			res = xval*yval;
			pos = v[i];
			ygood = yval;
		}
		else {
			yval = query(1,0,n-1,v[i],en);
			ll cur = xval*yval;
			if (cur > res) {
				res = cur;
				pos = v[i];
				ygood = yval;
			}
		}
	}
	num++;
	//cout << prev << ' ' << pos << ln;
	if (prev == -1 || x[pos]*y[pos] > y[prev]) {
		ll ans = ygood;
		ans = (ans*query1(1,0,n-1,0,pos))%INF;
		//cout << "ans: " << num << ' ' << ans << ln;
		return ans;
	}
	else {
		ll ans = yprev;
		ans = (ans*query1(1,0,n-1,0,prev))%INF;
		return ans;
	}
}

int init(int N, int X[], int Y[]) {
	n = N;
	FOR(i,0,n) {
		x[i] = X[i];
		y[i] = Y[i];
	}
	a.insert(0);
	FOR(i,0,n) {
		update(1,0,n-1,i,y[i]);
		update1(1,0,n-1,i,x[i]);
		if (x[i] > 1) a.insert(i);
	}
	return calc();
}

int updateX(int pos, int val) {	
	update1(1,0,n-1,pos,val);
	x[pos] = val;
	if (val == 1 && a.count(pos) && pos != 0) a.erase(pos);
	if (val > 1) a.insert(pos);
	return calc();
}

int updateY(int pos, int val) {
	update(1,0,n-1,pos,val);
	y[pos] = val;
	return calc();
}

Compilation message

horses.cpp: In function 'int calc()':
horses.cpp:78:33: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   78 |  if (a.size() == 0) return query(1,0,n-1,0,n-1);
      |                            ~~~~~^~~~~~~~~~~~~~~
horses.cpp:87:17: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   87 |  int vn = v.size();
      |           ~~~~~~^~
horses.cpp:115:31: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  115 |   ans = (ans*query1(1,0,n-1,0,pos))%INF;
      |                               ^~~
horses.cpp:117:10: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  117 |   return ans;
      |          ^~~
horses.cpp:121:31: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  121 |   ans = (ans*query1(1,0,n-1,0,prev))%INF;
      |                               ^~~~
horses.cpp:122:10: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  122 |   return ans;
      |          ^~~

Subtask #1
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 1 ms 212 KB Output is correct
9 Correct 1 ms 212 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 0 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 0 ms 212 KB Output is correct
17 Correct 0 ms 212 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 0 ms 212 KB Output is correct
20 Correct 0 ms 212 KB Output is correct

Subtask #2
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 1 ms 212 KB Output is correct
9 Correct 0 ms 212 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 1 ms 212 KB Output is correct
16 Correct 1 ms 212 KB Output is correct
17 Correct 1 ms 212 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 0 ms 212 KB Output is correct
20 Correct 0 ms 212 KB Output is correct
21 Correct 0 ms 212 KB Output is correct
22 Incorrect 1 ms 212 KB Output isn't correct
23 Halted 0 ms 0 KB -

Subtask #3
0 / 20.0

# Verdict Execution time Memory Grader output
1 Execution timed out 1570 ms 51984 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 1 ms 212 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 0 ms 212 KB Output is correct
13 Correct 1 ms 212 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 1 ms 212 KB Output is correct
17 Correct 1 ms 212 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 0 ms 212 KB Output is correct
20 Correct 0 ms 212 KB Output is correct
21 Correct 0 ms 212 KB Output is correct
22 Incorrect 1 ms 212 KB Output isn't correct
23 Halted 0 ms 0 KB -

Subtask #5
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 212 KB Output is correct
2 Correct 1 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 1 ms 212 KB Output is correct
6 Correct 1 ms 212 KB Output is correct
7 Correct 1 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 1 ms 212 KB Output is correct
10 Correct 1 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 0 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 1 ms 212 KB Output is correct
17 Correct 1 ms 212 KB Output is correct
18 Correct 1 ms 212 KB Output is correct
19 Correct 1 ms 212 KB Output is correct
20 Correct 1 ms 212 KB Output is correct
21 Correct 1 ms 212 KB Output is correct
22 Incorrect 1 ms 212 KB Output isn't correct
23 Halted 0 ms 0 KB -