Submission #1017769

#	Time	Username	Problem	Language	Result	Execution time	Memory
1017769		Muhammet	Horses (IOI15_horses)	C++17	0 / 100	28 ms	36388 KiB

horses

#include <bits/stdc++.h>
#include "horses.h"

using namespace std;

#define ll long long

const ll M = 1e9 + 7;

int n;

vector <int> a1, b1;

int f(vector <int> a1, vector <int> b1){
	vector <int> a(1,0), b(1,0), p(1,1), mx(n+2,0);
	for(auto i : a1) a.push_back(i);
	for(auto i : b1) b.push_back(i);
	for(int i = 1; i <= n; i++){
		p[i] = p[i-1] * a[i];
	}
	for(int i = n; i >= 1; i--){
		mx[i] = max(mx[i+1], p[i] * b[i]);
	}
	int ind = 1;
	ll ans = 0, k = 0;
	for(int i = 1; i <= n; i++){
		if(p[i] * b[i] == mx[ind]){
			k = max(k,ans+((p[i] * (b[i])) / p[ind-1]));
			if(i == n) b[i]++;
			ans += ((p[i] * (b[i]-1)) / p[ind-1]);
			ind = i+1;
			ans %= M;
			k %= M;
		}
	}
	ans = max(ans,k);
	ans %= M;
	return ans;
}

int init(int N, int X[], int Y[]) {
	n = N;
	a1.resize(n);
	b1.resize(n);
	for(int i = 0; i < n; i++){
		a1[i] = X[i];
		b1[i] = Y[i];
	}
	return f(a1,b1);
}

int updateX(int pos, int val) {	
	a1[pos] = val;
	return f(a1,b1);
}

int updateY(int pos, int val) {
	b1[pos] = val;
	return f(a1,b1);
}

Compilation message (stderr)

horses.cpp: In function 'int f(std::vector<int>, std::vector<int>)':
horses.cpp:14:37: warning: declaration of 'b1' shadows a global declaration [-Wshadow]
   14 | int f(vector <int> a1, vector <int> b1){
      |                        ~~~~~~~~~~~~~^~
horses.cpp:12:18: note: shadowed declaration is here
   12 | vector <int> a1, b1;
      |                  ^~
horses.cpp:14:20: warning: declaration of 'a1' shadows a global declaration [-Wshadow]
   14 | int f(vector <int> a1, vector <int> b1){
      |       ~~~~~~~~~~~~~^~
horses.cpp:12:14: note: shadowed declaration is here
   12 | vector <int> a1, b1;
      |              ^~
horses.cpp:38:9: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   38 |  return ans;
      |         ^~~

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