oj.uz

Submission #984285

# Submission time Handle Problem Language Result Execution time Memory
984285 2024-05-16T12:41:15 Z LucaIlie Horses (IOI15_horses) C++17
34 / 100
 1500 ms 58052 KB 
#include "horses.h"
#include <bits/stdc++.h>

using namespace std;

const int MAX_N = 5e5;
const int MOD = 1e9 + 7;

struct AIB1 {
    int aib[MAX_N + 1];

    void init() {
        for ( int i = 0; i <= MAX_N; i++ )
            aib[i] = 1;
    }

    void update( int i, int x ) {
        while ( i <= MAX_N ) {
            aib[i] = (long long)aib[i] * x % MOD;
            i += (i & -i);
        }
    }

    int query( int i ) {
        int p = 1;
        while ( i > 0 ) {
            p = (long long)p * aib[i] % MOD;
            i -= (i & -i);
        }
        return p;
    }
} horses;

struct AIB2 {
    long double aib[MAX_N + 1];

    void init() {
        for ( int i = 0; i <= MAX_N; i++ )
            aib[i] = 0;
    }

    void update( int i, long double x ) {
        while ( i <= MAX_N ) {
            aib[i] += x;
            i += (i & -i);
        }
    }

    long double query( int i ) {
        long double p = 0;
        while ( i > 0 ) {
            p += aib[i];
            i -= (i & -i);
        }
        return p;
    }
} cost;

int lgPut( int a, int n ) {
    if ( n == 0 )
        return 1;

    int p = lgPut( a, n / 2 );
    p = (long long)p * p % MOD;
    if ( n % 2 == 1 )
        p = (long long)p * a % MOD;

    return p;
}

int inv( int x ) {
    return lgPut( x, MOD - 2 );
}

int n;
int x[MAX_N + 1], y[MAX_N + 1];
set<pair<long double, int>> s;


void calc() {
    s.clear();
    horses.init();
    cost.init();
    for ( int i = 1; i <= n; i++ ) {
        horses.update( i, x[i] );
        cost.update( i, log( x[i] ) );
        s.insert( { cost.query( i ) + log( y[i] ), i } );
    }
}
void calc2() {
    s.clear();
    for ( int i = 1; i <= n; i++ ) {
        s.insert( { cost.query( i ) + log( y[i] ), i } );
    }
}

int answer() {
    int i = s.rbegin()->second;
    return (long long)horses.query( i ) * y[i] % MOD;
}

int init( int N, int X[], int Y[] ) {
    n = N;

    for ( int i = n - 1; i >= 0; i-- )
        x[i + 1] = X[i], y[i + 1] = Y[i];


    calc();
    return answer();
}

int updateX( int pos, int val ) {
    int i = pos + 1;

    auto ptr = s.lower_bound( { cost.query( i ) + log( y[i] ), 0 } );
    while ( ptr != s.end() && ptr->second != i )
        ptr++;
    if ( ptr == s.end() ) {
        ptr--;
        while ( ptr->second != i )
            ptr--;
    }
    s.erase( ptr );
    
    cost.update( i, -log( x[i] ) );
    horses.update( i, inv( x[i] ) );

    x[i] = val;

    cost.update( i, log( x[i] ) );
    horses.update( i, x[i] );
    s.insert( { cost.query( i ) + log( y[i] ), i } );
  calc2();
    
    return answer();
}

int updateY( int pos, int val ) {
    int i = pos + 1;

    s.erase( { cost.query( i ) + log( y[i] ), i } );
    y[i] = val;
    s.insert( { cost.query( i ) + log( y[i] ), i } );


    return answer();
}

Compilation message

horses.cpp: In member function 'void AIB1::update(int, int)':
horses.cpp:19:44: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   19 |             aib[i] = (long long)aib[i] * x % MOD;
      |                      ~~~~~~~~~~~~~~~~~~~~~~^~~~~
horses.cpp: In member function 'int AIB1::query(int)':
horses.cpp:27:39: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   27 |             p = (long long)p * aib[i] % MOD;
      |                 ~~~~~~~~~~~~~~~~~~~~~~^~~~~
horses.cpp: In function 'int lgPut(int, int)':
horses.cpp:64:26: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   64 |     p = (long long)p * p % MOD;
      |         ~~~~~~~~~~~~~~~~~^~~~~
horses.cpp:66:30: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   66 |         p = (long long)p * a % MOD;
      |             ~~~~~~~~~~~~~~~~~^~~~~
horses.cpp: In function 'int answer()':
horses.cpp:99:48: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   99 |     return (long long)horses.query( i ) * y[i] % MOD;
      |            ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~

Subtask #1
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 13916 KB Output is correct
2 Correct 2 ms 13916 KB Output is correct
3 Correct 3 ms 13972 KB Output is correct
4 Correct 3 ms 13912 KB Output is correct
5 Correct 2 ms 13916 KB Output is correct
6 Correct 2 ms 13916 KB Output is correct
7 Correct 2 ms 13912 KB Output is correct
8 Correct 2 ms 13916 KB Output is correct
9 Correct 2 ms 13916 KB Output is correct
10 Correct 2 ms 13916 KB Output is correct
11 Correct 2 ms 13916 KB Output is correct
12 Correct 3 ms 13916 KB Output is correct
13 Correct 2 ms 13916 KB Output is correct
14 Correct 3 ms 13916 KB Output is correct
15 Correct 2 ms 13916 KB Output is correct
16 Correct 3 ms 13916 KB Output is correct
17 Correct 2 ms 13916 KB Output is correct
18 Correct 3 ms 13916 KB Output is correct
19 Correct 2 ms 13916 KB Output is correct
20 Correct 2 ms 13916 KB Output is correct

Subtask #2
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 13912 KB Output is correct
2 Correct 2 ms 13916 KB Output is correct
3 Correct 3 ms 13916 KB Output is correct
4 Correct 2 ms 13916 KB Output is correct
5 Correct 2 ms 13916 KB Output is correct
6 Correct 2 ms 13960 KB Output is correct
7 Correct 2 ms 13912 KB Output is correct
8 Correct 2 ms 13912 KB Output is correct
9 Correct 2 ms 13916 KB Output is correct
10 Correct 2 ms 13916 KB Output is correct
11 Correct 3 ms 13912 KB Output is correct
12 Correct 3 ms 13912 KB Output is correct
13 Correct 2 ms 13916 KB Output is correct
14 Correct 2 ms 13916 KB Output is correct
15 Correct 2 ms 13916 KB Output is correct
16 Correct 3 ms 13916 KB Output is correct
17 Correct 2 ms 13916 KB Output is correct
18 Correct 2 ms 13916 KB Output is correct
19 Correct 3 ms 13916 KB Output is correct
20 Correct 2 ms 13916 KB Output is correct
21 Correct 2 ms 13916 KB Output is correct
22 Correct 2 ms 13720 KB Output is correct
23 Correct 62 ms 13916 KB Output is correct
24 Correct 64 ms 13916 KB Output is correct
25 Correct 48 ms 14060 KB Output is correct
26 Correct 50 ms 13912 KB Output is correct
27 Correct 61 ms 13916 KB Output is correct
28 Correct 106 ms 13916 KB Output is correct
29 Correct 91 ms 14168 KB Output is correct
30 Correct 87 ms 13916 KB Output is correct
31 Correct 5 ms 13912 KB Output is correct
32 Correct 50 ms 14064 KB Output is correct

Subtask #3
0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 212 ms 58052 KB Output is correct
2 Execution timed out 1555 ms 57068 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 13916 KB Output is correct
2 Correct 2 ms 13916 KB Output is correct
3 Correct 2 ms 13916 KB Output is correct
4 Correct 3 ms 13916 KB Output is correct
5 Correct 2 ms 13916 KB Output is correct
6 Correct 2 ms 13916 KB Output is correct
7 Correct 2 ms 13912 KB Output is correct
8 Correct 2 ms 13916 KB Output is correct
9 Correct 3 ms 13912 KB Output is correct
10 Correct 2 ms 13916 KB Output is correct
11 Correct 3 ms 13916 KB Output is correct
12 Correct 3 ms 13912 KB Output is correct
13 Correct 2 ms 13916 KB Output is correct
14 Correct 3 ms 13912 KB Output is correct
15 Correct 2 ms 13916 KB Output is correct
16 Correct 2 ms 13912 KB Output is correct
17 Correct 4 ms 13916 KB Output is correct
18 Correct 2 ms 13916 KB Output is correct
19 Correct 3 ms 13912 KB Output is correct
20 Correct 2 ms 13912 KB Output is correct
21 Correct 3 ms 13916 KB Output is correct
22 Correct 8 ms 13916 KB Output is correct
23 Correct 64 ms 13912 KB Output is correct
24 Correct 62 ms 14060 KB Output is correct
25 Correct 51 ms 14040 KB Output is correct
26 Correct 68 ms 13916 KB Output is correct
27 Correct 66 ms 14164 KB Output is correct
28 Correct 109 ms 14048 KB Output is correct
29 Correct 109 ms 10076 KB Output is correct
30 Correct 90 ms 10316 KB Output is correct
31 Correct 8 ms 10332 KB Output is correct
32 Correct 54 ms 10076 KB Output is correct
33 Execution timed out 1531 ms 57004 KB Time limit exceeded
34 Halted 0 ms 0 KB -

Subtask #5
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 13916 KB Output is correct
2 Correct 4 ms 13752 KB Output is correct
3 Correct 2 ms 13916 KB Output is correct
4 Correct 3 ms 13912 KB Output is correct
5 Correct 2 ms 13912 KB Output is correct
6 Correct 2 ms 13916 KB Output is correct
7 Correct 3 ms 13916 KB Output is correct
8 Correct 3 ms 13916 KB Output is correct
9 Correct 3 ms 13912 KB Output is correct
10 Correct 3 ms 13916 KB Output is correct
11 Correct 2 ms 13916 KB Output is correct
12 Correct 3 ms 13916 KB Output is correct
13 Correct 3 ms 13916 KB Output is correct
14 Correct 2 ms 13916 KB Output is correct
15 Correct 3 ms 13916 KB Output is correct
16 Correct 3 ms 13916 KB Output is correct
17 Correct 3 ms 13916 KB Output is correct
18 Correct 3 ms 14084 KB Output is correct
19 Correct 2 ms 13916 KB Output is correct
20 Correct 2 ms 13916 KB Output is correct
21 Correct 3 ms 13752 KB Output is correct
22 Correct 3 ms 13916 KB Output is correct
23 Correct 67 ms 14048 KB Output is correct
24 Correct 62 ms 14048 KB Output is correct
25 Correct 60 ms 14168 KB Output is correct
26 Correct 52 ms 14236 KB Output is correct
27 Correct 69 ms 10312 KB Output is correct
28 Correct 113 ms 10308 KB Output is correct
29 Correct 95 ms 10304 KB Output is correct
30 Correct 91 ms 10328 KB Output is correct
31 Correct 9 ms 10332 KB Output is correct
32 Correct 52 ms 10120 KB Output is correct
33 Correct 214 ms 58032 KB Output is correct
34 Execution timed out 1568 ms 56948 KB Time limit exceeded
35 Halted 0 ms 0 KB -