Submission #929212

#	Time	Username	Problem	Language	Result	Execution time	Memory
929212		OAleksa	Horses (IOI15_horses)	C++14	17 / 100	456 ms	69616 KiB

horses

#include "horses.h"
#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;
const long long inf = 1e18 + 69;
const int N = 5e5 + 69;
long long n, a[N], b[N], c[N], p[4 * N], g[4 * N];
set<int> st;
int mul(int x, int y) {
   long long r = 1ll * x * y;
   if (r >= mod)
      r %= mod;
   return r;
}
void modify(int v, int tl, int tr, int pos, int x) {
   if (tl == tr)
      p[v] = g[v] = x;
   else {
      int mid = (tl + tr) / 2;
      if (pos <= mid)
         modify(v * 2, tl, mid, pos, x);
      else
         modify(v * 2 + 1, mid + 1, tr, pos, x);
      if (p[v * 2] == -1 || p[v * 2 + 1] == -1 || p[v * 2] > inf / p[v * 2 + 1])
         p[v] = -1;
      else {
         assert(p[v * 2] * p[v * 2 + 1] < inf);
         p[v] = p[v * 2] * p[v * 2 + 1];
      }
      g[v] = mul(g[v * 2], g[v * 2 + 1]);
   }
}
int get(int v, int tl, int tr, int l, int r) {
   if (tl > r || tr < l)
      return 1;
   else if (tl >= l && tr <= r)
      return p[v];
   else {
      int mid = (tl + tr) / 2;
      auto l1 = get(v * 2, tl, mid, l, r);
      auto r1 = get(v * 2 + 1, mid + 1, tr, l, r);
      if (l1 == -1 || r1 == -1 || l1 > inf / r1)
         return -1;
      assert(l1 * r1 < inf);
      return l1 * r1;
   }
}
int Get(int v, int tl, int tr, int l, int r) {
   if (tl > r || tr < l)
      return 1;
   else if (tl >= l && tr <= r)
      return g[v];
   else {
      int mid = (tl + tr) / 2;
      return mul(Get(v * 2, tl, mid, l, r), Get(v * 2 + 1, mid + 1, tr, l, r));
   }
}
int Better(int i, int j) {
   if (i > j)
      swap(i, j);
   int x = get(1, 1, n, i + 1, j);
   if (x < 0 || x > inf / b[j])
      return j;
   return (b[i] > b[j] * x ? i : j);
}
int Result(int i) {
   return mul(Get(1, 1, n, 1, i), b[i]);
}
int Solve() {
   int bst = 1;
   st.insert(1);
   for (int i = 2;i <= n;i++) {
      if (Better(bst, i) == i) {
         bst = i;
         st.insert(i);
      }
   }
   return Result(bst);
}

int init(int N, int X[], int Y[]) {
   n = N;
   fill(p, p + 4 * n + 10, 1);
   fill(g, g + 4 * n + 10, 1);
   for (int i = 1;i <= n;i++) {
      a[i] = X[i - 1], b[i] = Y[i - 1];
      modify(1, 1, n, i, a[i]);
   }
   return Solve();
}

int updateX(int pos, int val) {
   pos++;
   a[pos] = val;
   modify(1, 1, n, pos, a[pos]);
   auto u = st.lower_bound(pos);
   if (u != st.begin()) {
      if (Better(*--u, pos) == pos)
         st.insert(pos);
      else if (st.count(pos))
         st.erase(pos);
   }
   u = st.upper_bound(pos);
   while (u != st.end() && Better(*u, pos) == pos) {
      st.erase(u);
      u = st.upper_bound(pos);
   }
   assert(!st.empty());
	return Result(*st.rbegin());
}

int updateY(int pos, int val) {
   pos++;
	b[pos] = val;
	auto u = st.lower_bound(pos);
   if (u != st.begin()) {
      if (Better(*--u, pos) == pos)
         st.insert(pos);
      else if (st.count(pos))
         st.erase(pos);
   }
   u = st.upper_bound(pos);
   while (u != st.end() && Better(*u, pos) == pos) {
      st.erase(u);
      u = st.upper_bound(pos);
   }
   assert(!st.empty());
	return Result(*st.rbegin());
}
/*
3
2 1 3
3 4 1
1
2 1 2

*/

Compilation message (stderr)

horses.cpp: In function 'int mul(int, int)':
horses.cpp:13:11: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   13 |    return r;
      |           ^
horses.cpp: In function 'void modify(int, int, int, int, int)':
horses.cpp:30:25: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   30 |       g[v] = mul(g[v * 2], g[v * 2 + 1]);
      |                  ~~~~~~~^
horses.cpp:30:39: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   30 |       g[v] = mul(g[v * 2], g[v * 2 + 1]);
      |                            ~~~~~~~~~~~^
horses.cpp: In function 'int get(int, int, int, int, int)':
horses.cpp:37:17: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   37 |       return p[v];
      |              ~~~^
horses.cpp: In function 'int Get(int, int, int, int, int)':
horses.cpp:52:17: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   52 |       return g[v];
      |              ~~~^
horses.cpp: In function 'int Better(int, int)':
horses.cpp:61:22: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   61 |    int x = get(1, 1, n, i + 1, j);
      |                      ^
horses.cpp: In function 'int Result(int)':
horses.cpp:67:25: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   67 |    return mul(Get(1, 1, n, 1, i), b[i]);
      |                         ^
horses.cpp:67:38: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   67 |    return mul(Get(1, 1, n, 1, i), b[i]);
      |                                   ~~~^
horses.cpp: In function 'int init(int, int*, int*)':
horses.cpp:81:14: warning: declaration of 'N' shadows a global declaration [-Wshadow]
   81 | int init(int N, int X[], int Y[]) {
      |          ~~~~^
horses.cpp:6:11: note: shadowed declaration is here
    6 | const int N = 5e5 + 69;
      |           ^
horses.cpp:87:20: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   87 |       modify(1, 1, n, i, a[i]);
      |                    ^
horses.cpp:87:29: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   87 |       modify(1, 1, n, i, a[i]);
      |                          ~~~^
horses.cpp: In function 'int updateX(int, int)':
horses.cpp:95:17: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   95 |    modify(1, 1, n, pos, a[pos]);
      |                 ^
horses.cpp:95:30: warning: conversion from 'long long int' to 'int' may change value [-Wconversion]
   95 |    modify(1, 1, n, pos, a[pos]);
      |                         ~~~~~^

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