oj.uz

Submission #778252

# Submission time Handle Problem Language Result Execution time Memory
778252 2023-07-10T07:50:56 Z danikoynov Horses (IOI15_horses) C++14
37 / 100
 526 ms 53004 KB 
#include "horses.h"
#include<bits/stdc++.h>

using namespace std;
typedef long long ll;

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

ll prod_tree[4 * maxn];

void build_prod_tree(int root, int left, int right)
{
    if (left == right)
    {
        prod_tree[root] = x[left];
        return;
    }

    int mid = (left + right) / 2;
    build_prod_tree(root * 2, left, mid);
    build_prod_tree(root * 2 + 1, mid + 1, right);

    prod_tree[root] = (prod_tree[root * 2] * prod_tree[root * 2 + 1]) % mod;
}

void update_prod(int root, int left, int right, int pos)
{
    if (left == right)
    {
        prod_tree[root] = x[left];
        return;
    }

    int mid = (left + right) / 2;
    if (pos <= mid)
        update_prod(root * 2, left, mid, pos);
    else
        update_prod(root * 2 + 1, mid + 1, right, pos);

    prod_tree[root] = (prod_tree[root * 2] * prod_tree[root * 2 + 1]) % mod;
}

ll prod_query(int root, int left, int right, int qleft, int qright)
{
    if (left > qright || right < qleft)
        return 1;
    if (left >= qleft && right <= qright)
        return prod_tree[root];

    int mid = (left + right) / 2;
    return (prod_query(root * 2, left, mid, qleft, qright) *
            prod_query(root * 2 + 1, mid + 1, right, qleft, qright)) % mod;
}

ll breed_tree[4 * maxn];

void build_breed_tree(int root, int left, int right)
{
    if (left == right)
    {
        breed_tree[root] = y[left];
        return;
    }

    int mid = (left + right) / 2;
    build_breed_tree(root * 2, left, mid);
    build_breed_tree(root * 2 + 1, mid + 1, right);

    breed_tree[root] = max(breed_tree[root * 2], breed_tree[root * 2 + 1]);
}

void update_breed(int root, int left, int right, int pos)
{
    if (left == right)
    {
        breed_tree[root] = y[left];
        return;
    }

    int mid = (left + right) / 2;
    if (pos <= mid)
        update_breed(root * 2, left, mid, pos);
    else
        update_breed(root * 2 + 1, mid + 1, right, pos);

    breed_tree[root] = max(breed_tree[root * 2], breed_tree[root * 2 + 1]);
}

ll breed_query(int root, int left, int right, int qleft, int qright)
{
    if (left > qright || right < qleft)
        return 0;

    if (left >= qleft && right <= qright)
        return breed_tree[root];

    int mid = (left + right) / 2;
    return max(breed_query(root * 2, left, mid, qleft, qright),
               breed_query(root * 2 + 1, mid + 1, right, qleft, qright));
}

set < int > active;

ll action()
{
    ll up = y[n - 1], dw = 1, pos = n - 1;
    ll div = 1;
    set < int > :: reverse_iterator it;
    if (active.empty())
    {
        pos = 0;
        up = breed_tree[1];
    }
    for (it = active.rbegin(); it != active.rend(); it ++)
    {
        ll cur = breed_query(1, 0, n - 1, *it, n - 1);
        if (div > 1e9)
            break;
            ///cout << *it << " " << cur << " " << div << endl;
        if (cur * dw >= up * div)
        {
            up = cur;
            dw = div;
            pos = *it;
        }
        div = div * x[*it];
    }

    /**for (int i = n - 2; i >= 0; i --)
    {
        div = div * x[i + 1];
        if (div > 1e9)
            break;
        ///cout << i << " : " << y[i] << " " << div << endl;
        /// y[i] / div >? up / dw
        if (y[i] * dw >= up * div)
        {

            ///cout << " " << up << " : " << dw << endl;
            up = y[i];
            dw = div;
            pos = i;
        }
    }*/

    ///cout << pos << endl;
    ll ans = prod_query(1, 0, n - 1, 0, pos);
    //for (int i = 0; i <= pos; i ++)
    //  ans = (ans * x[i]) % mod;
    ans = (ans * breed_query(1, 0, n - 1, pos, n - 1)) % mod;
    return ans;
}
int init(int N, int X[], int Y[])
{
    n = N;
    for (int i = 0; i < n; i ++)
    {
        if (X[i] > 1)
            active.insert(i);
        x[i] = X[i];
        y[i] = Y[i];
    }
    build_prod_tree(1, 0, n - 1);
    build_breed_tree(1, 0, n - 1);
    return action();
}

int updateX(int pos, int val)
{
    if (x[pos] > 1)
        active.erase(pos);
    x[pos] = val;
    if (x[pos] > 1)
        active.insert(pos);
    update_prod(1, 0, n - 1, pos);
    return action();
}

int updateY(int pos, int val)
{
    y[pos] = val;
    update_breed(1, 0, n - 1, pos);
    return action();
}

Compilation message

horses.cpp: In function 'll action()':
horses.cpp:118:38: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  118 |         ll cur = breed_query(1, 0, n - 1, *it, n - 1);
      |                                    ~~^~~
horses.cpp:118:50: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  118 |         ll cur = breed_query(1, 0, n - 1, *it, n - 1);
      |                                                ~~^~~
horses.cpp:119:13: warning: conversion from 'll' {aka 'long long int'} to 'double' may change value [-Wconversion]
  119 |         if (div > 1e9)
      |             ^~~
horses.cpp:149:33: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  149 |     ll ans = prod_query(1, 0, n - 1, 0, pos);
      |                               ~~^~~
horses.cpp:149:41: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  149 |     ll ans = prod_query(1, 0, n - 1, 0, pos);
      |                                         ^~~
horses.cpp:152:38: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  152 |     ans = (ans * breed_query(1, 0, n - 1, pos, n - 1)) % mod;
      |                                    ~~^~~
horses.cpp:152:43: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  152 |     ans = (ans * breed_query(1, 0, n - 1, pos, n - 1)) % mod;
      |                                           ^~~
horses.cpp:152:50: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  152 |     ans = (ans * breed_query(1, 0, n - 1, pos, n - 1)) % mod;
      |                                                ~~^~~
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:165:29: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  165 |     build_prod_tree(1, 0, n - 1);
      |                           ~~^~~
horses.cpp:166:30: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  166 |     build_breed_tree(1, 0, n - 1);
      |                            ~~^~~
horses.cpp:167:18: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  167 |     return action();
      |            ~~~~~~^~
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:177:25: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  177 |     update_prod(1, 0, n - 1, pos);
      |                       ~~^~~
horses.cpp:178:18: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  178 |     return action();
      |            ~~~~~~^~
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:184:26: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  184 |     update_breed(1, 0, n - 1, pos);
      |                        ~~^~~
horses.cpp:185:18: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  185 |     return action();
      |            ~~~~~~^~

Subtask #1
17.0 / 17.0

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

Subtask #2
0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 1 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 0 ms 212 KB Output is correct
10 Correct 0 ms 220 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 0 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 0 ms 212 KB Output is correct
17 Correct 1 ms 212 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 0 ms 212 KB Output is correct
20 Correct 0 ms 224 KB Output is correct
21 Correct 0 ms 212 KB Output is correct
22 Correct 0 ms 212 KB Output is correct
23 Correct 1 ms 340 KB Output is correct
24 Correct 1 ms 340 KB Output is correct
25 Correct 1 ms 340 KB Output is correct
26 Correct 1 ms 340 KB Output is correct
27 Incorrect 3 ms 340 KB Output isn't correct
28 Halted 0 ms 0 KB -

Subtask #3
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 526 ms 52912 KB Output is correct
2 Correct 288 ms 52912 KB Output is correct
3 Correct 284 ms 53004 KB Output is correct
4 Correct 335 ms 52920 KB Output is correct

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 0 ms 212 KB Output is correct
4 Correct 1 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 0 ms 212 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 1 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 1 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 0 ms 212 KB Output is correct
17 Correct 0 ms 212 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 1 ms 212 KB Output is correct
20 Correct 1 ms 212 KB Output is correct
21 Correct 0 ms 212 KB Output is correct
22 Correct 0 ms 212 KB Output is correct
23 Correct 1 ms 340 KB Output is correct
24 Correct 1 ms 340 KB Output is correct
25 Correct 1 ms 340 KB Output is correct
26 Correct 1 ms 340 KB Output is correct
27 Incorrect 2 ms 340 KB Output isn't correct
28 Halted 0 ms 0 KB -

Subtask #5
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 212 KB Output is correct
2 Correct 0 ms 212 KB Output is correct
3 Correct 1 ms 212 KB Output is correct
4 Correct 0 ms 212 KB Output is correct
5 Correct 0 ms 212 KB Output is correct
6 Correct 0 ms 212 KB Output is correct
7 Correct 0 ms 212 KB Output is correct
8 Correct 0 ms 212 KB Output is correct
9 Correct 0 ms 212 KB Output is correct
10 Correct 0 ms 212 KB Output is correct
11 Correct 0 ms 212 KB Output is correct
12 Correct 0 ms 212 KB Output is correct
13 Correct 0 ms 212 KB Output is correct
14 Correct 0 ms 212 KB Output is correct
15 Correct 0 ms 212 KB Output is correct
16 Correct 0 ms 212 KB Output is correct
17 Correct 1 ms 212 KB Output is correct
18 Correct 0 ms 212 KB Output is correct
19 Correct 0 ms 212 KB Output is correct
20 Correct 0 ms 212 KB Output is correct
21 Correct 0 ms 212 KB Output is correct
22 Correct 0 ms 212 KB Output is correct
23 Correct 1 ms 340 KB Output is correct
24 Correct 1 ms 340 KB Output is correct
25 Correct 1 ms 340 KB Output is correct
26 Correct 1 ms 340 KB Output is correct
27 Incorrect 3 ms 340 KB Output isn't correct
28 Halted 0 ms 0 KB -