Submission #925292

#	Time	Username	Problem	Language	Result	Execution time	Memory
925292		Karoot	Horses (IOI15_horses)	C++17	17 / 100	168 ms	85332 KiB

horses

#include "horses.h"
#include <iostream>
#include <cmath>
#include <unordered_map>
#include <map>
#include <set>
#include <queue>
#include <vector>
#include <string>
#include <iomanip>
#include <algorithm>

#define all(x)  (x).begin(), (x).end()
#define rall(x)  (x).rbegin(), (x).rend()

using namespace std;

typedef long long ll;

ll linf = 1e15+1;

inline void scoobydoobydoo(){
    ios::sync_with_stdio(false);
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
}

typedef unsigned long long ull;

ull modmul(ull a, ull b, ull M) {
	ll ret = a * b - M * ull(1.L / M * a * b);
	return ret + M * (ret < 0) - M * (ret >= (ll)M);
}
ull modpow(ull b, ull e, ull mod) {
	ull ans = 1;
	for (; e; b = modmul(b, b, mod), e /= 2)
		if (e & 1) ans = modmul(ans, b, mod);
	return ans;
}

const int M = 1e9+7;
const int MAXN = 5e5+1;

long double X[MAXN], Y[MAXN];

struct Node {   
    long double sum;
    long double maxi;
    ll multi;
    ll index;
};

Node ST[4*MAXN];
int leftOf[4*MAXN], rightOf[4*MAXN];
int indexToNode[MAXN];

void buildTree(int l, int r, int node = 1){
    leftOf[node] = l; rightOf[node] = r;
    if (l == r){
        indexToNode[l] = node;
        ST[node].index = l;
        ST[node].maxi = log2(Y[l]);
        ST[node].sum = log2(X[l]);
        ST[node].maxi += ST[node].sum;
        ST[node].multi = X[l];
        return;
    }
    int mid = (l+r)>>1;
    buildTree(l, mid, 2*node);
    buildTree(mid+1, r, 2*node+1);
    ST[node].sum = ST[2*node].sum + ST[2*node+1].sum;
    if (ST[2*node].maxi > ST[2*node+1].maxi+ST[2*node].sum){
        ST[node].index = ST[2*node].index;
        ST[node].maxi = ST[2*node].maxi;
    } else {
        ST[node].index = ST[2*node+1].index;
        ST[node].maxi = ST[2*node+1].maxi+ST[2*node].sum;
    }
    ST[node].multi = modmul(ST[2*node].multi, ST[2*node+1].multi, M);

    return;
}

ll multQuery(int l, int r, int node = 1){
    if (leftOf[node] == l && rightOf[node] == r){
        return ST[node].multi;
    }
    int a = node*2, b = node*2+1;
    if (r <= rightOf[a])return multQuery(l, r, a);

    return modmul(multQuery(l, rightOf[a], a), multQuery(leftOf[b], r, b), M);
}

int query(){
    ll index = ST[1].index;
    ll ans = modmul(Y[index], multQuery(0, index), M);
    
    return ans;
}

void update1(int node){
    if (!node)return;
    ST[node].sum = ST[2*node].sum + ST[2*node+1].sum;
    if (ST[2*node].maxi > ST[2*node+1].maxi+ST[2*node].sum){
        ST[node].index = ST[2*node].index;
        ST[node].maxi = ST[2*node].maxi;
    } else {
        ST[node].index = ST[2*node+1].index;
        ST[node].maxi = ST[2*node+1].maxi+ST[2*node].sum;
    }
    update1(node>>1);
}

int updateX(int pos, int val){
    ST[indexToNode[pos]].sum = log2((long double)val);
    ST[indexToNode[pos]].maxi = log2((long double)val) + log2(Y[pos]);
    update1(indexToNode[pos]>>1);
    X[pos] = val;
    return query();
}

int updateY(int pos, int val){
    ST[indexToNode[pos]].maxi = log2((long double)val) + log2(X[pos]);
    update1(indexToNode[pos]>>1);
    Y[pos] = val;
    return query();
}

int init(int n, int x[], int y[]) {
	for(int i = 0; i < n; i++){
		X[i] = x[i];
        Y[i] = y[i];
	}
    buildTree(0, n-1);
	return query();
}

Compilation message (stderr)

horses.cpp: In function 'void buildTree(int, int, int)':
horses.cpp:65:29: warning: conversion from 'long double' to 'll' {aka 'long long int'} may change value [-Wfloat-conversion]
   65 |         ST[node].multi = X[l];
      |                          ~~~^
horses.cpp: In function 'int query()':
horses.cpp:96:44: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
   96 |     ll ans = modmul(Y[index], multQuery(0, index), M);
      |                                            ^~~~~
horses.cpp:96:28: warning: conversion from 'long double' to 'ull' {aka 'long long unsigned int'} may change value [-Wfloat-conversion]
   96 |     ll ans = modmul(Y[index], multQuery(0, index), M);
      |                     ~~~~~~~^
horses.cpp:98:12: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
   98 |     return ans;
      |            ^~~

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