Submission #1025116

#	Time	Username	Problem	Language	Result	Execution time	Memory
1025116		YassirSalama	Horses (IOI15_horses)	C++17	17 / 100	301 ms	112760 KiB

horses

#include<bits/stdc++.h>
using namespace std;
#define F first
#define S second
#define all(v) v.begin(),v.end()
#define ll long long
template<typename T> 
void dbg_out(const T& t){
	cout<<t<<endl;
}
template<typename T,typename... Args>
void dbg_out(const T& t,const Args&... args){
	cout<<t<<" , ";
	dbg_out(args...);
}
#define dbg(...) cout<<"("<<#__VA_ARGS__<<"): ";dbg_out(__VA_ARGS__);
const int mod=1e9+7;
const int maxn=5e5;
int n;
vector<ll> x;
vector<ll> y;
ll dp[maxn];
ll solve(int i){
	if(i==n) return 0LL;
	if(dp[i]!=-1) return dp[i];
	ll h=1;
	ll &ans=dp[i];
	ans=0;
	int t=0;
	for(int k=i;k<n&&t<=40;k++){
		h=(h%mod*x[k]%mod)%mod;
		if(x[k]!=1) t++;
		ans=max(ans,((ll)solve(k+1)%mod+(ll)(h-1)%mod*(ll)y[k]%mod)%mod);
		ans=max(ans,((ll)h%mod*(ll)y[k]%mod)%mod);
	}
	ans%=mod;
	return ans;
}
void clear(){
	for(int j=0;j<=n;j++){
		dp[j]=-1;
	}
}
int init(int N, int X[], int Y[]) {
	n=N;
	x.resize(n,0);
	y.resize(n,0);
	for(int i=0;i<n;i++){
		x[i]=X[i];
	}
	for(int i=0;i<n;i++){
		y[i]=Y[i];
	}
	// clear();
	memset(dp,-1,sizeof(dp));
	return (int)((ll)solve(0)%mod);
}

int updateX(int pos, int val) {	
	x[pos]=val;
	// clear();
	memset(dp,-1,sizeof(dp));
	return solve(0)%mod;
}

int updateY(int pos, int val) {
	y[pos]=val;
	clear();
	return solve(0)%mod;
}


#ifdef IOI


static char _buffer[1024];
static int _currentChar = 0;
static int _charsNumber = 0;
static FILE *_inputFile, *_outputFile;

static inline int _read() {
    if (_charsNumber < 0) {
        exit(1);
    }
    if (!_charsNumber || _currentChar == _charsNumber) {
        _charsNumber = (int)fread(_buffer, sizeof(_buffer[0]), sizeof(_buffer), _inputFile);
        _currentChar = 0;
    }
    if (_charsNumber <= 0) {
        return -1;
    }
    return _buffer[_currentChar++];
}

static inline int _readInt() {
    int c, x, s;
    c = _read();
    while (c <= 32) c = _read();
    x = 0;
    s = 1;
    if (c == '-') {
        s = -1;
        c = _read();
    }
    while (c > 32) {
        x *= 10;
        x += c - '0';
        c = _read();
    }
    if (s < 0) x = -x;
    return x;
}


int main() {
	// _inputFile = fopen("horses.in", "rb");
	// _outputFile = fopen("horses.out", "w");
	
	int N; 
	cin>>N;

	int *X = (int*)malloc(sizeof(int)*(unsigned int)N);
	int *Y = (int*)malloc(sizeof(int)*(unsigned int)N);

	for (int i = 0; i < N; i++) {
		cin>>X[i];
	}

	for (int i = 0; i < N; i++) {
		cin>>Y[i];
	}	
	cout<<init(N,X,Y)<<endl;


	int M; 
	// M = _readInt();
	cin>>M;

	for (int i = 0; i < M; i++) {
		int type; 
		int pos; 
		int val; 
		cin>>type>>pos>>val;
		if (type == 1) {
			cout<<updateX(pos,val)<<endl;
			// fprintf(_outputFile,"%d\n",updateX(pos,val));
		} else if (type == 2) {
			cout<<updateY(pos,val)<<endl;
			// fprintf(_outputFile,"%d\n",updateY(pos,val));
		}
	}

	return 0;
}

#endif 

Compilation message (stderr)

horses.cpp: In function 'int updateX(int, int)':
horses.cpp:63:17: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   63 |  return solve(0)%mod;
      |         ~~~~~~~~^~~~
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:69:17: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   69 |  return solve(0)%mod;
      |         ~~~~~~~~^~~~

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