# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
60086 |
2018-07-23T16:00:40 Z |
Flugan42 |
Horses (IOI15_horses) |
C++14 |
|
0 ms |
0 KB |
#include "horses.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef vector<ll> vi;
typedef pair<ll,ll> ii;
typedef vector<ii> vii;
#define rep(i,a,b) for(ll i = a; i < b; i++)
#define per(i,a,b) for(ll i = a; i >= b; i--)
#define all(x) x.begin(),x.end()
#define inf 1000000000000000000
#define sz(x) (ll)(x).size()
const ll mod = 1000000007;
ll pw(ll a, ll x){
if (x == 0) return 1;
if (x == 1) return a%mod;
ll temp = pw(a,x/2);
if (x%2) return (((temp*temp)%mod)*a)%mod;
else return ((temp*temp)%mod);
}
class segT {
private:
vector<pair<ld,ll> > tree;
vector<ld> a;
ll n;
ll left(ll v) { return 2*v; }
ll right(ll v) { return 2*v+1; }
void build(ll v, ll l, ll r){
if (l == r) tree[v] = make_pair(a[l],l);
else {
build(left(v),l,(l+r)/2);
build(right(v),(l+r)/2+1,r);
if (tree[left(v)].first > tree[right(v)].first) tree[v] = tree[left(v)];
else tree[v] = tree[right(v)];
}
}
pair<ld,ll> rmq(ll v, ll l, ll r, ll a, ll b){
if (l > b || r < a) return make_pair(-1,-1);
if (a <= l && r <= b) return tree[v];
else {
pair<ld,ll> p1 = rmq(left(v),l,(l+r)/2,a,b);
pair<ld,ll> p2 = rmq(right(v),(l+r)/2+1,r,a,b);
if (p1.first == -1 && p2.first == -1) return -1;
if (p1.first == -1) return p2;
if (p2.first == -1) return p1;
if (p1.first > p2.first) return p1;
else return p2;
}
}
void rangeadd(ll v, ll l, ll r, ll i, ll j, ld val){
if (l > j || r < i) return;
if (i <= l && r <= j) tree[v] = make_pair(tree[v].first+val,tree[v].second);
else {
rangeadd(left(v), l, (l+r)/2, i, j, val);
rangeadd(right(v), (l+r)/2+1, r, i, j, val);
if (tree[left(v)].first > tree[right(v)].first) tree[v] = tree[left(v)];
else tree[v] = tree[right(v)];
}
}
public:
void build(vector<ld> &_a){
a = _a;
n = sz(_a);
tree.assign(4*n,make_pair(0.0,0));
build(1,0,n-1);
}
pair<ld,ll> rmq(ll i, ll j) { return rmq(1,0,n-1,i,j); }
void rangeadd(ll i, ll j, ld val) { rangeadd(1,0,n-1,i,j,val); }
} ;
class segTp {
private:
vi tree,a;
ll n;
ll left(ll v) { return 2*v; }
ll right(ll v) { return 2*v+1; }
void build(ll v, ll l, ll r){
if (l == r) tree[v] = a[l];
else {
build(left(v),l,(l+r)/2);
build(right(v),(l+r)/2+1,r);
tree[v] = (tree[left(v)]*tree[right(v)])%mod;
}
}
ll rmq(ll v, ll l, ll r, ll a, ll b){
if (l > b || r < a) return -1;
if (a <= l && r <= b) return tree[v];
else {
ll p1 = rmq(left(v),l,(l+r)/2,a,b);
ll p2 = rmq(right(v),(l+r)/2+1,r,a,b);
if (p1 == -1 && p2 == -1) return -1;
if (p1 == -1) return p2;
if (p2 == -1) return p1;
return (p1*p2)%mod;
}
}
void rangeprod(ll v, ll l, ll r, ll i, ll j, ll val){
if (l > j || r < i) return;
if (i <= l && r <= j) tree[v] = (tree[v]*pw(val,(r-l+1)))%mod;
else {
rangeprod(left(v), l, (l+r)/2, i, j, val);
rangeprod(right(v), (l+r)/2+1, r, i, j, val);
tree[v] = (tree[left(v)]*tree[right(v)])%mod;
}
}
public:
void build(vi &_a){
a = _a;
n = sz(_a);
tree.assign(4*n,0);
build(1,0,n-1);
}
ll rmq(ll i, ll j) { return rmq(1,0,n-1,i,j); }
void rangeprod(ll i, ll j, ll val) { rangeprod(1,0,n-1,i,j,val); }
} ;
vi inp;
vector<ld> lnsum;
pair<ld,ll> res;
segT myT;
segTp pT;
int init(int N, int X[], int Y[]) {
inp[0] = X[0]%mod;
lnsum[0] = log(ld(X[0]));
rep(i,1,N){
inp[i] = (inp[i-1]*X[0])%mod;
lnsum[i] = lnsum[i-1]+log(X[i]);
}
rep(i,0,N){
lnsum[i] += log(Y[i]);
}
myT.build(lnsum);
pT.build(inp);
res = myT.rmq(0,N-1);
return (pT.rmq(0,res.second)*Y[res.second])%mod;
}
int updateX(int pos, int val) {
myT.rangeadd(0,pos,log(val));
pT.rangeprod(pos,pos,val);
res = myT.rmq(0,N-1);
return (pT.rmq(0,res.second)*Y[res.second])%mod;
}
int updateY(int pos, int val) {
myT.rangeadd(pos,pos,log(val));
res = myT.rmq(0,N-1);
return (pT.rmq(0,res.second)*Y[res.second])%mod;
}
Compilation message
horses.cpp: In member function 'std::pair<long double, long long int> segT::rmq(ll, ll, ll, ll, ll)':
horses.cpp:40:48: warning: declaration of 'a' shadows a member of 'segT' [-Wshadow]
pair<ld,ll> rmq(ll v, ll l, ll r, ll a, ll b){
^
horses.cpp:27:14: note: shadowed declaration is here
vector<ld> a;
^
horses.cpp:46:53: error: could not convert '-1' from 'int' to 'std::pair<long double, long long int>'
if (p1.first == -1 && p2.first == -1) return -1;
^
horses.cpp: In member function 'll segTp::rmq(ll, ll, ll, ll, ll)':
horses.cpp:89:39: warning: declaration of 'a' shadows a member of 'segTp' [-Wshadow]
ll rmq(ll v, ll l, ll r, ll a, ll b){
^
horses.cpp:77:11: note: shadowed declaration is here
vi tree,a;
^
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:140:45: warning: conversion to 'int' from 'll {aka long long int}' may alter its value [-Wconversion]
return (pT.rmq(0,res.second)*Y[res.second])%mod;
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:146:18: error: 'N' was not declared in this scope
res = myT.rmq(0,N-1);
^
horses.cpp:147:31: error: 'Y' was not declared in this scope
return (pT.rmq(0,res.second)*Y[res.second])%mod;
^
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:152:18: error: 'N' was not declared in this scope
res = myT.rmq(0,N-1);
^
horses.cpp:153:31: error: 'Y' was not declared in this scope
return (pT.rmq(0,res.second)*Y[res.second])%mod;
^