Submission #624589

#	Time	Username	Problem	Language	Result	Execution time	Memory
624589		NicolaAbusaad2014	Horses (IOI15_horses)	C++14	17 / 100	424 ms	75816 KiB

horses

#include "horses.h"
#include <bits/stdc++.h>

using namespace std;
#define REP(i,l,r) for(long long i=(l);i<(r);i++)
#define PER(i,l,r) for(long long i=(r)-1;i>=(l);i--)
const long long mod=1e9+7;
long long p(long long x){while(x&(x-1)){x=x&(x-1);}return x;}
long long squared(long long x){return (x*x)%mod;}
long long power(long long x,long long p){if(p==0){return 1;}if(p%2==1){return (power(x,p-1)*x)%mod;}return squared(power(x,p/2));}
long long inv(long long x){return power(x,mod-2);}
struct segment_tree
{
    struct node
    {
        double value;
        long pos,left,right;
    };
    node dum;
    vector<node>tree;
    vector<double>upd;
    node operation(node x,node z)
    {
        node ret;
        if(x.value>=z.value){
            ret.value=x.value;
            ret.pos=x.pos;
        }
        else{
            ret.value=z.value;
            ret.pos=z.pos;
        }
        ret.left=min(x.left,z.left);
        ret.right=max(x.right,z.right);
        return ret;
    }
    void push(long node)
    {
        tree[node].value+=upd[node];
        if(node<tree[0].value){
            upd[node*2]+=upd[node];
            upd[(node*2)+1]+=upd[node];
        }
        upd[node]=0;
    }
    void build(vector<double>v)
    {
        tree.clear();
        upd.clear();
        int x=((v.size())*2)-1;
        while(x&(x-1)){
            x=x&(x-1);
        }
        tree.resize(x*2);
        upd.resize(x*2);
        tree[0].value=x;
        dum.value=-1e9;
        dum.left=-1e9;
        dum.right=1e9;
        for(long i=0;i<v.size();i++){
            tree[i+x].value=v[i];
            tree[i+x].pos=i;
            tree[i+x].left=i;
            tree[i+x].right=i;
        }
        for(long i=v.size();i<x;i++){
            tree[i+x].value=-1e9;
            tree[i+x].pos=i;
            tree[i+x].left=i;
            tree[i+x].right=i;
        }
        for(long i=x-1;i>0;i--){
            tree[i]=operation(tree[i*2],tree[(i*2)+1]);
        }
    }
    node get(int l,int r,int node=1)
    {
        push(node);
        if(tree[node].left>r||tree[node].right<l){
            return dum;
        }
        if(tree[node].left>=l&&tree[node].right<=r){
            return tree[node];
        }
        return operation(get(l,r,node*2),get(l,r,(node*2)+1));
    }
    void update(int l,int r,double z,int node=1)
    {
        push(node);
        if(tree[node].left>r||tree[node].right<l){
            return;
        }
        if(tree[node].left>=l&&tree[node].right<=r){
            upd[node]+=z;
            push(node);
            return;
        }
        update(l,r,z,node*2);
        update(l,r,z,(node*2)+1);
        tree[node]=operation(tree[node*2],tree[(node*2)+1]);
    }
};
segment_tree st;
long n;
vector<long long>x,y,tot;
vector<double>v;
int init(int N, int X[], int Y[]) {
    n=N;
    x.resize(n);
    y.resize(n);
    tot.resize(n);
    v.resize(n);
    REP(i,0,n){
        x[i]=X[i];
        y[i]=Y[i];
    }
    v[0]=log(x[0])+log(y[0]);
    tot[0]=(x[0]*y[0])%mod;
    REP(i,1,n){
        v[i]=v[i-1]+log(x[i])+log(y[i])-log(y[i-1]);
        tot[i]=(((tot[i-1]*x[i])%mod)*((y[i]*inv(y[i-1]))%mod))%mod;
    }
    st.build(v);
    N=st.get(0,n-1).pos;
    return tot[N];
}

int updateX(int pos, int val) {
	st.update(pos,n-1,log(val)-log(x[pos]));
    tot[pos]*=val;
    tot[pos]%=mod;
    tot[pos]*=inv(x[pos]);
    tot[pos]%=mod;
    x[pos]=val;
    pos=st.get(0,n-1).pos;
    return tot[pos];
}

int updateY(int pos, int val) {
	st.update(pos,pos,log(val)-log(y[pos]));
    tot[pos]*=val;
    tot[pos]%=mod;
    tot[pos]*=inv(y[pos]);
    tot[pos]%=mod;
    y[pos]=val;
    pos=st.get(0,n-1).pos;
    return tot[pos];
}

Compilation message (stderr)

horses.cpp: In member function 'void segment_tree::push(long int)':
horses.cpp:37:20: warning: declaration of 'node' shadows a member of 'segment_tree' [-Wshadow]
   37 |     void push(long node)
      |               ~~~~~^~~~
horses.cpp:14:12: note: shadowed declaration is here
   14 |     struct node
      |            ^~~~
horses.cpp:40:12: warning: conversion from 'long int' to 'double' may change value [-Wconversion]
   40 |         if(node<tree[0].value){
      |            ^~~~
horses.cpp: In member function 'void segment_tree::build(std::vector<double>)':
horses.cpp:50:29: warning: conversion from 'std::vector<double>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   50 |         int x=((v.size())*2)-1;
      |               ~~~~~~~~~~~~~~^~
horses.cpp:60:23: warning: comparison of integer expressions of different signedness: 'long int' and 'std::vector<double>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |         for(long i=0;i<v.size();i++){
      |                      ~^~~~~~~~~
horses.cpp: In member function 'segment_tree::node segment_tree::get(int, int, int)':
horses.cpp:76:30: warning: declaration of 'node' shadows a member of 'segment_tree' [-Wshadow]
   76 |     node get(int l,int r,int node=1)
      |                          ~~~~^~~~~~
horses.cpp:14:12: note: shadowed declaration is here
   14 |     struct node
      |            ^~~~
horses.cpp: In member function 'void segment_tree::update(int, int, double, int)':
horses.cpp:87:42: warning: declaration of 'node' shadows a member of 'segment_tree' [-Wshadow]
   87 |     void update(int l,int r,double z,int node=1)
      |                                      ~~~~^~~~~~
horses.cpp:14:12: note: shadowed declaration is here
   14 |     struct node
      |            ^~~~
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:124:17: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  124 |     N=st.get(0,n-1).pos;
      |                ~^~
horses.cpp:124:21: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  124 |     N=st.get(0,n-1).pos;
      |       ~~~~~~~~~~~~~~^~~
horses.cpp:125:17: warning: conversion from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' may change value [-Wconversion]
  125 |     return tot[N];
      |                 ^
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:129:17: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  129 |  st.update(pos,n-1,log(val)-log(x[pos]));
      |                ~^~
horses.cpp:135:19: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  135 |     pos=st.get(0,n-1).pos;
      |                  ~^~
horses.cpp:135:23: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  135 |     pos=st.get(0,n-1).pos;
      |         ~~~~~~~~~~~~~~^~~
horses.cpp:136:19: warning: conversion from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' may change value [-Wconversion]
  136 |     return tot[pos];
      |                   ^
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:146:19: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  146 |     pos=st.get(0,n-1).pos;
      |                  ~^~
horses.cpp:146:23: warning: conversion from 'long int' to 'int' may change value [-Wconversion]
  146 |     pos=st.get(0,n-1).pos;
      |         ~~~~~~~~~~~~~~^~~
horses.cpp:147:19: warning: conversion from '__gnu_cxx::__alloc_traits<std::allocator<long long int>, long long int>::value_type' {aka 'long long int'} to 'int' may change value [-Wconversion]
  147 |     return tot[pos];
      |                   ^

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