Submission #310653

#	Time	Username	Problem	Language	Result	Execution time	Memory
310653		talant117408	Horses (IOI15_horses)	C++17	0 / 100	84 ms	15576 KiB

horses

#include "horses.h"
#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
typedef pair <ll, ll> pii;
 
#define precision(n) fixed << setprecision(n)
#define pb push_back
#define ub upper_bound
#define lb lower_bound
#define mp make_pair
#define eps (double)1e-9
#define PI 2*acos(0.0)
#define endl "\n"
#define sz(v) (int)(v).size()
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define do_not_disturb ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);

const ll mod = 1e9+7;
const int mxn = 5e5+7;
ll x[mxn], y[mxn];
int n;

struct Node{
    ll x, y;
    double _x, _y;
};

ll mult(ll a, ll b){
    return (a*b)%mod;
}

Node combine(Node a, Node b){
    Node res;
    
    if(a._y > a._x+b._y){
        res.y = a.y;
        res._y = a._y;
    }
    else{
        res.y = mult(a.x, b.y);
        res._y = a._x+b._y;
    }
    res.x = mult(a.x, b.x);
    res._x = a._x+b._x;
    
    return res;
}

struct Tree{
    Node tree[mxn*2];
    
    void build(int v, int l, int r){
        if(l == r){
            tree[v].x = x[l];
            tree[v].y = mult(y[l], x[l]);
            tree[v]._x = log2(double(x[l]));
            tree[v]._y = log2(double(y[l]))+log2(double(x[l]));
            return;
        }
        int mid = (l+r) >> 1;
        build(v>>1, l, mid);
        build((v>>1)+1, mid+1, r);
        tree[v] = combine(tree[v>>1], tree[(v>>1)+1]);
    }
    
    void update(int v, int l, int r, int pos){
        if(pos == l && r == l){
            tree[v].x = x[l];
            tree[v].y = mult(y[l], x[l]);
            tree[v]._x = log2(double(x[l]));
            tree[v]._y = log2(double(y[l]))+log2(double(x[l]));
            return;
        }
        int mid = (l+r) >> 1;
        if(pos <= mid){
            update(v>>1, l, mid, pos);
        }
        else{
            update((v>>1)+1, mid+1, r, pos);
        }
        tree[v] = combine(tree[v>>1], tree[(v>>1)+1]);
    }
    
    ll query(){
        return tree[1].y;
    }
} tree;

int init(int N, int X[], int Y[]) {
    n = N; 
    copy(X, X+n, x);
    copy(Y, Y+n, y);
    tree.build(1, 0, n-1);
    
    return tree.query();
}

int updateX(int pos, int val){
    x[pos] = ll(val);
    tree.update(1, 0, n-1, pos);
    
	return tree.query();
}

int updateY(int pos, int val){
    y[pos] = ll(val);
    tree.update(1, 0, n-1, pos);
    
    //for(int i = 0; i < n; i++){
      //  cout << x[i] << ' ' << y[i] << endl;
   // }
    
	return tree.query();
}

/*
3
2 1 3
3 4 1
1
2 1 2

8
6
*/

Compilation message (stderr)

horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:99:22: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
   99 |     return tree.query();
      |            ~~~~~~~~~~^~
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:106:19: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  106 |  return tree.query();
      |         ~~~~~~~~~~^~
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:117:19: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  117 |  return tree.query();
      |         ~~~~~~~~~~^~

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