Submission #583782

#	Time	Username	Problem	Language	Result	Execution time	Memory
583782		M_W	Horses (IOI15_horses)	C++17	17 / 100	549 ms	33428 KiB

horses

#include <bits/stdc++.h>
using namespace std;
long long tp[500001 << 2], tmd[500001 << 2];
const long long md = 1e9 + 7;
int N;

void upd_product(int v, int tl, int tr, int pos, int val){
	if(tl == pos && tl == tr){
		tp[v] = tmd[v] = val * 1ll;
		return;
	}
	int tm = (tl + tr) >> 1;
	if(pos <= tm) upd_product(v * 2, tl, tm, pos, val);
	else upd_product(v * 2 + 1, tm + 1, tr, pos, val);
	
	tmd[v] = (tmd[v * 2] * tmd[v * 2 + 1]) % md;
	tp[v] = tp[v * 2] * tp[v * 2 + 1];
	if(tp[v] > 1e9 || tp[v * 2] == -1 || tp[v * 2 + 1] == -1) tp[v] = -1;
}

long long query_product(int v, int tl, int tr, int l, int r){
	if(l > r) return 1ll;
	if(tl == l && tr == r) return tp[v];
	int tm = (tl + tr) >> 1;
	
	long long left = query_product(v * 2, tl, tm, l, min(r, tm));
	long long right = query_product(v * 2 + 1, tm + 1, tr, max(l, tm + 1), r);
	
	if(left == -1 || right == -1 || left * right > 1e9) return -1;
	return left * right;
}


long long query_md(int v, int tl, int tr, int l, int r){
	if(l > r) return 1ll;
	if(tl == l && tr == r) return tmd[v];
	int tm = (tl + tr) >> 1;
	
	long long left = query_product(v * 2, tl, tm, l, min(r, tm));
	long long right = query_product(v * 2 + 1, tm + 1, tr, max(l, tm + 1), r);
	
	return (left * right) % md;
}

// Actual query tree
int t[500001 << 2];
long long Y[500001];

void build(int v, int tl, int tr){
	if(tl == tr){
		t[v] = tl;
		return;
	}
	int tm = (tl + tr) >> 1;
	build(v * 2, tl, tm);
	build(v * 2 + 1, tm + 1, tr);
	
	long long Yl = Y[t[v * 2]], Yr = Y[t[v * 2 + 1]];
	long long prod = query_product(1, 0, N - 1, t[v * 2] + 1, t[v * 2 + 1]);
	
	if(prod == -1 || prod * Yr > Yl) t[v] = t[v * 2 + 1];
	else t[v] = t[v * 2];
}

void ins(int v, int tl, int tr, int pos){
	if(tl == pos && tl == tr) return;
	int tm = (tl + tr) >> 1;
	
	if(pos <= tm) ins(v * 2, tl, tm, pos);
	else ins(v * 2 + 1, tm + 1, tr, pos);
	
	long long Yl = Y[t[v * 2]], Yr = Y[t[v * 2 + 1]];
	long long prod = query_product(1, 0, N - 1, t[v * 2] + 1, t[v * 2 + 1]);
	
	if(prod == -1 || prod * Yr > Yl) t[v] = t[v * 2 + 1];
	else t[v] = t[v * 2];
}

int init(int N_tmp, int X[], int Y_tmp[]){
	N = N_tmp;
	for(int i = 0; i < N; i++) upd_product(1, 0, N - 1, i, X[i]);
	for(int i = 0; i < N; i++) Y[i] = Y_tmp[i] * 1ll;
	build(1, 0, N - 1);
	
	return (query_md(1, 0, N - 1, 0, t[1]) * Y[t[1]]) % md;
}

int updateX(int pos, int val){
	upd_product(1, 0, N - 1, pos, val);
	ins(1, 0, N - 1, pos);
	return (query_md(1, 0, N - 1, 0, t[1]) * Y[t[1]]) % md;
}

int updateY(int pos, int val){
	Y[pos] = val;
	ins(1, 0, N - 1, pos);
	return (query_md(1, 0, N - 1, 0, t[1]) * Y[t[1]]) % md;
}

Compilation message (stderr)

horses.cpp: In function 'void upd_product(int, int, int, int, int)':
horses.cpp:18:9: warning: conversion from 'long long int' to 'double' may change value [-Wconversion]
   18 |  if(tp[v] > 1e9 || tp[v * 2] == -1 || tp[v * 2 + 1] == -1) tp[v] = -1;
      |     ~~~~^
horses.cpp: In function 'long long int query_product(int, int, int, int, int)':
horses.cpp:29:39: warning: conversion from 'long long int' to 'double' may change value [-Wconversion]
   29 |  if(left == -1 || right == -1 || left * right > 1e9) return -1;
      |                                  ~~~~~^~~~~~~
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:85:52: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   85 |  return (query_md(1, 0, N - 1, 0, t[1]) * Y[t[1]]) % md;
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:91:52: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   91 |  return (query_md(1, 0, N - 1, 0, t[1]) * Y[t[1]]) % md;
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:97:52: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   97 |  return (query_md(1, 0, N - 1, 0, t[1]) * Y[t[1]]) % md;
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~

• Subtask 1: 17 / 17
• Subtask 2: 0 / 17
• Subtask 3: 0 / 20
• Subtask 4: 0 / 23
• Subtask 5: 0 / 23