Submission #649502

#	Time	Username	Problem	Language	Result	Execution time	Memory
649502		ToroTN	Horses (IOI15_horses)	C++14	17 / 100	74 ms	28728 KiB

horses

#include<bits/stdc++.h>
using namespace std;
#include "horses.h"
#define ll long long
#define X first
#define Y second
ll n,x[500005],y[500005],md=1e9+7,nw,fight;
ll seg[2000005],val,str,ans,last,mul[2000005],big;
void build(ll tree,ll st,ll ed)
{
	ll md=(st+ed)/2;
	if(st==ed)
	{
		seg[tree]=y[ed];
		mul[tree]=x[ed];
		return;
	}
	build(2*tree,st,md);
	build(2*tree+1,md+1,ed);
	seg[tree]=max(seg[2*tree],seg[2*tree+1]);
	mul[tree]=1;
	if(mul[2*tree]!=0)
	{
		mul[tree]=(mul[tree]*mul[2*tree]);
		mul[tree]%=1000000007;
	}
	if(mul[2*tree+1]!=0)
	{
		mul[tree]=(mul[tree]*mul[2*tree+1]);
		mul[tree]%=1000000007;
	}
}
void update(ll tree,ll st,ll ed,ll idx,ll val)
{
	ll md=(st+ed)/2;
	if(st==ed)
	{
		seg[tree]=y[idx];
		mul[tree]=x[idx];
		return;
	}
	if(idx<=md)
	{
		update(2*tree,st,md,idx,val);
	}else
	{
		update(2*tree+1,md+1,ed,idx,val);
	}
	seg[tree]=max(seg[2*tree],seg[2*tree+1]);
	mul[tree]=(mul[2*tree]*mul[2*tree+1])%1000000007;
}
ll query(ll tree,ll st,ll ed,ll l,ll r)
{
	ll md=(st+ed)/2;
	if(st>r||ed<l)return -1e9;
	if(st>=l&&ed<=r)return seg[tree];
	return max(query(2*tree,st,md,l,r),query(2*tree+1,md+1,ed,l,r));
}
ll query2(ll tree,ll st,ll ed,ll l,ll r)
{
	ll md=(st+ed)/2;
	if(st>r||ed<l)return 1;
	if(st>=l&&ed<=r)return mul[tree];
	return (query2(2*tree,st,md,l,r)*query2(2*tree+1,md+1,ed,l,r))%1000000007;
}
set<ll> s;
stack<pair<ll,ll> > stk;
int init(int N, int X[], int Y[]) {
	n=N;
	for(int i=1;i<=n;i++)
	{
		x[i]=X[i-1];
		y[i]=Y[i-1];
	}
	build(1,1,n);
	/*printf("%lld\n",mul[1]);
	for(int i=1;i<=7;i++)
	{
		printf("%d %lld\n",i,mul[i]);
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			printf("%d %d %lld\n",i,j,query2(1,1,n,(ll)i,(ll)j));
		}
	}*/
	for(int i=1;i<n;i++)
	{
		if(x[i]>1)s.insert(i);
		if(s.size()>20)s.erase(s.find(*s.begin()));
	}
	for(auto it=s.begin();it!=s.end();it++)
	{
		stk.push({*it,query(1,1,n,(*it)+1,n)});
	}
	nw=n;
	str=y[n];
	ans=str;
	while(!stk.empty())
	{
		fight=stk.top().X;
		val=stk.top().Y;
		//printf("%lld %lld\n",fight,val);
		stk.pop();
		if(str>1e9)
		{
			str=1e18;
			ans*=x[nw];
			ans%=md;
		}else
		{
			if(val>str*x[nw])
			{
				str=val;
				ans=val;
			}else
			{
				str*=x[nw];
				ans*=x[nw];
				ans%=md;
			}
		}
		if(y[fight]>str)
		{
			str=y[fight];
			ans=y[fight];
		}
		nw=fight;
	}
	big=query2(1,1,n,1,nw);
	//printf("%lld\n",big);
	if(str>1e9)
	{
		str=1e18;
		ans*=big;
		ans%=md;
		return ans;
	}else
	{
		last=seg[1];
		if(last>str*big)
		{
			return last;
		}else
		{
			ans*=big;
			ans%=md;
			return ans;
		}
	}
}

int updateX(int pos, int val) {
	++pos;
	return 0;
}

int updateY(int pos, int val) {
	++pos;
	return 0;
}

Compilation message (stderr)

horses.cpp: In function 'void build(long long int, long long int, long long int)':
horses.cpp:11:5: warning: declaration of 'md' shadows a global declaration [-Wshadow]
   11 |  ll md=(st+ed)/2;
      |     ^~
horses.cpp:7:26: note: shadowed declaration is here
    7 | ll n,x[500005],y[500005],md=1e9+7,nw,fight;
      |                          ^~
horses.cpp: In function 'void update(long long int, long long int, long long int, long long int, long long int)':
horses.cpp:33:43: warning: declaration of 'val' shadows a global declaration [-Wshadow]
   33 | void update(ll tree,ll st,ll ed,ll idx,ll val)
      |                                           ^
horses.cpp:8:17: note: shadowed declaration is here
    8 | ll seg[2000005],val,str,ans,last,mul[2000005],big;
      |                 ^~~
horses.cpp:35:5: warning: declaration of 'md' shadows a global declaration [-Wshadow]
   35 |  ll md=(st+ed)/2;
      |     ^~
horses.cpp:7:26: note: shadowed declaration is here
    7 | ll n,x[500005],y[500005],md=1e9+7,nw,fight;
      |                          ^~
horses.cpp: In function 'long long int query(long long int, long long int, long long int, long long int, long long int)':
horses.cpp:54:5: warning: declaration of 'md' shadows a global declaration [-Wshadow]
   54 |  ll md=(st+ed)/2;
      |     ^~
horses.cpp:7:26: note: shadowed declaration is here
    7 | ll n,x[500005],y[500005],md=1e9+7,nw,fight;
      |                          ^~
horses.cpp: In function 'long long int query2(long long int, long long int, long long int, long long int, long long int)':
horses.cpp:61:5: warning: declaration of 'md' shadows a global declaration [-Wshadow]
   61 |  ll md=(st+ed)/2;
      |     ^~
horses.cpp:7:26: note: shadowed declaration is here
    7 | ll n,x[500005],y[500005],md=1e9+7,nw,fight;
      |                          ^~
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:106:6: warning: conversion from 'long long int' to 'double' may change value [-Wconversion]
  106 |   if(str>1e9)
      |      ^~~
horses.cpp:133:5: warning: conversion from 'long long int' to 'double' may change value [-Wconversion]
  133 |  if(str>1e9)
      |     ^~~
horses.cpp:138:10: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  138 |   return ans;
      |          ^~~
horses.cpp:144:11: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  144 |    return last;
      |           ^~~~
horses.cpp:149:11: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
  149 |    return ans;
      |           ^~~
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:154:26: warning: declaration of 'val' shadows a global declaration [-Wshadow]
  154 | int updateX(int pos, int val) {
      |                      ~~~~^~~
horses.cpp:8:17: note: shadowed declaration is here
    8 | ll seg[2000005],val,str,ans,last,mul[2000005],big;
      |                 ^~~
horses.cpp:154:26: warning: unused parameter 'val' [-Wunused-parameter]
  154 | int updateX(int pos, int val) {
      |                      ~~~~^~~
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:159:26: warning: declaration of 'val' shadows a global declaration [-Wshadow]
  159 | int updateY(int pos, int val) {
      |                      ~~~~^~~
horses.cpp:8:17: note: shadowed declaration is here
    8 | ll seg[2000005],val,str,ans,last,mul[2000005],big;
      |                 ^~~
horses.cpp:159:26: warning: unused parameter 'val' [-Wunused-parameter]
  159 | int updateY(int pos, int val) {
      |                      ~~~~^~~

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