// 人外有人,天外有天
// author: Ausp3x
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include "horses.h"
typedef long long lng;
typedef unsigned int uint;
typedef unsigned long long ulng;
using namespace std;
using namespace __gnu_pbds;
int const INF32 = 0x3f3f3f3f;
lng const INF64 = 0x3f3f3f3f3f3f3f3f;
int const MOD = 1000000007;
int N;
vector<int> X, Y;
lng modPow(lng x, lng y) {
lng res = 1;
while (y > 0) {
if (y & 1) {
res *= x;
res %= MOD;
}
y >>= 1;
x *= x;
x %= MOD;
}
return (res + MOD) % MOD;
}
lng almostAllHorses = 1;
int getMaxValue(int opt) {
if (opt == 1) {
lng horses = 1;
for (int i = 0; i < N; i++) {
(horses *= X[i]) %= MOD;
bool chk = true;
lng cur = 1;
for (int j = i + 1; j < N; j++) {
if (cur == INF32 || cur * X[j] > INF32) {
chk = false;
break;
}
cur *= X[j];
if (cur * Y[j] > Y[i]) {
chk = false;
break;
}
}
if (chk) {
return horses * Y[i] % MOD;
}
}
} else if (opt == 2) {
lng horses = almostAllHorses;
for (int i = N - 64; i < N; i++) {
(horses *= X[i]) %= MOD;
bool chk = true;
lng cur = 1;
for (int j = i + 1; j < N; j++) {
if (cur == INF32 || cur * X[j] > INF32) {
chk = false;
break;
}
cur *= X[j];
if (cur * Y[j] > Y[i]) {
chk = false;
break;
}
}
if (chk) {
return horses * Y[i] % MOD;
}
}
}
return -1;
}
bool allXG2 = true;
int init(int n, int x[], int y[]) {
N = n;
X.resize(n);
Y.resize(n);
for (int i = 0; i < N; i++) {
if (i < N - 64)
(allHorses *= x[i]) %= MOD;
if (x[i] < 2)
allXG2 = false;
X[i] = x[i];
Y[i] = y[i];
}
return (allXG2 ? getMaxValue(2) : getMaxValue(1));
}
bool allValG2 = true;
int updateX(int pos, int val) {
if (pos < N - 64) {
(almostAllHorses *= modPow(X[pos], MOD - 2)) %= MOD;
(almostAllHorses *= val) %= MOD;
}
if (val < 2)
allValG2 = false;
X[pos] = val;
return (allXG2 && allValG2 ? getMaxValue(2) : getMaxValue(1));
}
int updateY(int pos, int val) {
if (val < 2)
allValG2 = false;
Y[pos] = val;
return (allXG2 && allValG2 ? getMaxValue(2) : getMaxValue(1));
}
// int main() {
// ios_base::sync_with_stdio(false);
// cin.tie(NULL);
// int x[3] = {2, 1, 3};
// int y[3] = {3, 4, 1};
// cout << init(3, x, y) << endl;
// cout << updateY(1, 2) << endl;
// return 0;
// }
Compilation message
horses.cpp: In function 'int getMaxValue(int)':
horses.cpp:61:38: warning: conversion from 'lng' {aka 'long long int'} to 'int' may change value [-Wconversion]
61 | return horses * Y[i] % MOD;
| ~~~~~~~~~~~~~~^~~~~
horses.cpp:85:38: warning: conversion from 'lng' {aka 'long long int'} to 'int' may change value [-Wconversion]
85 | return horses * Y[i] % MOD;
| ~~~~~~~~~~~~~~^~~~~
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:100:14: error: 'allHorses' was not declared in this scope
100 | (allHorses *= x[i]) %= MOD;
| ^~~~~~~~~