oj.uz

Submission #1039987

# Submission time Handle Problem Language Result Execution time Memory
1039987 2024-07-31T13:27:27 Z c2zi6 Horses (IOI15_horses) C++14
54 / 100
 1500 ms 65364 KB 
#define _USE_MATH_DEFINES
#include <bits/stdc++.h>
#define ff first
#define ss second
#define pb push_back
#define all(a) (a).begin(), (a).end()
#define replr(i, a, b) for (int i = int(a); i <= int(b); ++i)
#define reprl(i, a, b) for (int i = int(a); i >= int(b); --i)
#define rep(i, n) for (int i = 0; i < int(n); ++i)
#define mkp(a, b) make_pair(a, b)
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> PII;
typedef vector<int> VI;
typedef vector<PII> VPI;
typedef vector<VI> VVI;
typedef vector<VVI> VVVI;
typedef vector<VPI> VVPI;
typedef pair<ll, ll> PLL;
typedef vector<ll> VL;
typedef vector<PLL> VPL;
typedef vector<VL> VVL;
typedef vector<VVL> VVVL;
typedef vector<VPL> VVPL;
template<class T> T setmax(T& a, T b) {if (a < b) return a = b; return a;}
template<class T> T setmin(T& a, T b) {if (a < b) return a; return a = b;}
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
template<typename T>
using indset = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
#include "horses.h"

const ll MOD = 1e9 + 7;

struct SEGTREEPROD {
    int n;
    VL tree;
    SEGTREEPROD(){}
    SEGTREEPROD(int sz) {
        n = 1;
        while (n < sz) n *= 2;
        tree = VL(2*n);
    }
    void upd(int N, int L, int R, int i, int s) {
        if (i < L || i > R) return;
        if (L == R) {
            tree[N] = s;
            return;
        }
        int M = (L + R) / 2;
        upd(2*N+1, L, M, i, s);
        upd(2*N+2, M+1, R, i, s);
        tree[N] = tree[2*N+1] * tree[2*N+2] % MOD;
    }
    int get(int N, int L, int R, int l, int r) {
        if (l <= L && R <= r) return tree[N];
        if (R < l || L > r) return 1;
        int M = (L + R) / 2;
        return 1ll * get(2*N+1, L, M, l, r) * get(2*N+2, M+1, R, l, r) % MOD;
    }
    void upd(int i, int s) {
        upd(0, 0, n-1, i, s);
    }
    int get(int l, int r) {
        return get(0, 0, n-1, l, r);
    }
};
struct SEGTREEMAX {
    int n;
    VL tree;
    SEGTREEMAX(){}
    SEGTREEMAX(int sz) {
        n = 1;
        while (n < sz) n *= 2;
        tree = VL(2*n);
    }
    void upd(int N, int L, int R, int i, int s) {
        if (i < L || i > R) return;
        if (L == R) {
            tree[N] = s;
            return;
        }
        int M = (L + R) / 2;
        upd(2*N+1, L, M, i, s);
        upd(2*N+2, M+1, R, i, s);
        tree[N] = max(tree[2*N+1], tree[2*N+2]);
    }
    int get(int N, int L, int R, int l, int r) {
        if (l <= L && R <= r) return tree[N];
        if (R < l || L > r) return -2e9;
        int M = (L + R) / 2;
        return max(get(2*N+1, L, M, l, r), get(2*N+2, M+1, R, l, r));
    }
    void upd(int i, int s) {
        upd(0, 0, n-1, i, s);
    }
    int get(int l, int r) {
        return get(0, 0, n-1, l, r);
    }
};

int n;
VL a, b;
SEGTREEPROD sega;
SEGTREEMAX segb;
set<int> nonone;

ll recalc() {
    ll leave = n-1;
    ll prod = 1;
    reprl(i, n-1, 0) {
    /*for (auto it = prev(nonone.end()); it != nonone.begin(); it--) {*/
    /*    int i = *it;*/
        if (a[i] == 1) {
            if (i == 0) {
                int i = 0;
                ll bi = segb.get(i, n-1);
                if (bi > prod) {
                    prod = a[i] * bi;
                    leave = i;
                } else {
                    prod = a[i] * prod;
                }
            }
            continue;
        }
        ll bi = segb.get(i, n-1);
        if (bi > prod) {
            prod = a[i] * bi;
            leave = i;
        } else {
            prod = a[i] * prod;
        }
        if (prod > 1e9) break;
    }

    /*if (a[0] == 1) {*/
    /*    int i = 0;*/
    /*    ll bi = segb.get(i, n-1);*/
    /*    if (bi > prod) {*/
    /*        prod = a[i] * bi;*/
    /*        leave = i;*/
    /*    } else {*/
    /*        prod = a[i] * prod;*/
    /*    }*/
    /*}*/

	return 1ll * sega.get(0, leave) * segb.get(leave, n-1) % MOD;
	return sega.get(0, leave) * b[leave] % MOD;
}

int init(int N, int X[], int Y[]) {
    n = N;
    a = b = VL(n);
    rep(i, n) {
        a[i] = X[i];
        b[i] = Y[i];
    }
    sega = SEGTREEPROD(n);
    rep(i, n) sega.upd(i, a[i]);
    segb = SEGTREEMAX(n);
    rep(i, n) segb.upd(i, b[i]);
    rep(i, n) if (a[i] != 1) nonone.insert(i);
    return recalc();
}

int updateX(int pos, int val) {	
    if (a[pos] != 1) nonone.erase(pos);
    a[pos] = val;
    if (a[pos] != 1) nonone.insert(pos);
    sega.upd(pos, a[pos]);
	return recalc();
}

int updateY(int pos, int val) {
    b[pos] = val;
    segb.upd(pos, b[pos]);
	return recalc();
}

Compilation message

horses.cpp: In member function 'int SEGTREEPROD::get(int, int, int, int, int)':
horses.cpp:57:44: 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]
   57 |         if (l <= L && R <= r) return tree[N];
      |                                            ^
horses.cpp:60:72: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   60 |         return 1ll * get(2*N+1, L, M, l, r) * get(2*N+2, M+1, R, l, r) % MOD;
      |                ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~
horses.cpp: In member function 'int SEGTREEMAX::get(int, int, int, int, int)':
horses.cpp:90:44: 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]
   90 |         if (l <= L && R <= r) return tree[N];
      |                                            ^
horses.cpp: In function 'll recalc()':
horses.cpp:117:21: warning: declaration of 'i' shadows a previous local [-Wshadow]
  117 |                 int i = 0;
      |                     ^
horses.cpp:112:11: note: shadowed declaration is here
  112 |     reprl(i, n-1, 0) {
      |           ^
horses.cpp:8:33: note: in definition of macro 'reprl'
    8 | #define reprl(i, a, b) for (int i = int(a); i >= int(b); --i)
      |                                 ^
horses.cpp:135:13: warning: conversion from 'll' {aka 'long long int'} to 'double' may change value [-Wconversion]
  135 |         if (prod > 1e9) break;
      |             ^~~~
horses.cpp:149:27: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  149 |  return 1ll * sega.get(0, leave) * segb.get(leave, n-1) % MOD;
      |                           ^~~~~
horses.cpp:149:45: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  149 |  return 1ll * sega.get(0, leave) * segb.get(leave, n-1) % MOD;
      |                                             ^~~~~
horses.cpp:150:21: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  150 |  return sega.get(0, leave) * b[leave] % MOD;
      |                     ^~~~~
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:161:31: 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]
  161 |     rep(i, n) sega.upd(i, a[i]);
      |                               ^
horses.cpp:163:31: 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]
  163 |     rep(i, n) segb.upd(i, b[i]);
      |                               ^
horses.cpp:165:18: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  165 |     return recalc();
      |            ~~~~~~^~
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:172:25: 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]
  172 |     sega.upd(pos, a[pos]);
      |                         ^
horses.cpp:173:15: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  173 |  return recalc();
      |         ~~~~~~^~
horses.cpp: In function 'int updateY(int, int)':
horses.cpp:178:25: 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]
  178 |     segb.upd(pos, b[pos]);
      |                         ^
horses.cpp:179:15: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  179 |  return recalc();
      |         ~~~~~~^~

Subtask #1
17.0 / 17.0

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

Subtask #2
17.0 / 17.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 348 KB Output is correct
15 Correct 0 ms 348 KB Output is correct
16 Correct 1 ms 348 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 1 ms 352 KB Output is correct
23 Correct 1 ms 352 KB Output is correct
24 Correct 1 ms 352 KB Output is correct
25 Correct 1 ms 360 KB Output is correct
26 Correct 1 ms 476 KB Output is correct
27 Correct 2 ms 444 KB Output is correct
28 Correct 1 ms 348 KB Output is correct
29 Correct 2 ms 348 KB Output is correct
30 Correct 1 ms 348 KB Output is correct
31 Correct 2 ms 516 KB Output is correct
32 Correct 2 ms 348 KB Output is correct

Subtask #3
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 360 ms 56832 KB Output is correct
2 Correct 315 ms 65364 KB Output is correct
3 Correct 300 ms 56656 KB Output is correct
4 Correct 319 ms 60752 KB Output is correct

Subtask #4
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 440 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 344 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 0 ms 348 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 0 ms 436 KB Output is correct
15 Correct 0 ms 348 KB Output is correct
16 Correct 0 ms 344 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 1 ms 348 KB Output is correct
24 Correct 1 ms 348 KB Output is correct
25 Correct 1 ms 348 KB Output is correct
26 Correct 1 ms 348 KB Output is correct
27 Correct 2 ms 348 KB Output is correct
28 Correct 1 ms 348 KB Output is correct
29 Correct 2 ms 348 KB Output is correct
30 Correct 1 ms 344 KB Output is correct
31 Correct 2 ms 348 KB Output is correct
32 Correct 2 ms 348 KB Output is correct
33 Correct 132 ms 32660 KB Output is correct
34 Correct 126 ms 32412 KB Output is correct
35 Correct 223 ms 61996 KB Output is correct
36 Correct 217 ms 62036 KB Output is correct
37 Correct 260 ms 30804 KB Output is correct
38 Correct 161 ms 43792 KB Output is correct
39 Execution timed out 1534 ms 30652 KB Time limit exceeded
40 Halted 0 ms 0 KB -

Subtask #5
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 600 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 348 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 0 ms 348 KB Output is correct
9 Correct 0 ms 348 KB Output is correct
10 Correct 0 ms 348 KB Output is correct
11 Correct 0 ms 348 KB Output is correct
12 Correct 0 ms 432 KB Output is correct
13 Correct 0 ms 348 KB Output is correct
14 Correct 1 ms 348 KB Output is correct
15 Correct 1 ms 344 KB Output is correct
16 Correct 0 ms 348 KB Output is correct
17 Correct 0 ms 348 KB Output is correct
18 Correct 0 ms 348 KB Output is correct
19 Correct 0 ms 348 KB Output is correct
20 Correct 0 ms 348 KB Output is correct
21 Correct 0 ms 348 KB Output is correct
22 Correct 0 ms 348 KB Output is correct
23 Correct 1 ms 344 KB Output is correct
24 Correct 1 ms 348 KB Output is correct
25 Correct 1 ms 468 KB Output is correct
26 Correct 1 ms 348 KB Output is correct
27 Correct 2 ms 348 KB Output is correct
28 Correct 1 ms 348 KB Output is correct
29 Correct 2 ms 348 KB Output is correct
30 Correct 1 ms 348 KB Output is correct
31 Correct 2 ms 348 KB Output is correct
32 Correct 2 ms 348 KB Output is correct
33 Correct 378 ms 56852 KB Output is correct
34 Correct 328 ms 65360 KB Output is correct
35 Correct 308 ms 56752 KB Output is correct
36 Correct 331 ms 60788 KB Output is correct
37 Correct 141 ms 32596 KB Output is correct
38 Correct 121 ms 32720 KB Output is correct
39 Correct 215 ms 62840 KB Output is correct
40 Correct 220 ms 62800 KB Output is correct
41 Correct 263 ms 30696 KB Output is correct
42 Correct 161 ms 43856 KB Output is correct
43 Execution timed out 1547 ms 30656 KB Time limit exceeded
44 Halted 0 ms 0 KB -