Submission #246808

#TimeUsernameProblemLanguageResultExecution timeMemory
246808srvltMeetings (IOI18_meetings)C++14
19 / 100
941 ms380056 KiB
#include "meetings.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
#define ll long long
#define ld long double
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define SZ(x) (int)(x).size()
template <typename T> using ord_set = tree <T, null_type, less <T>, rb_tree_tag, tree_order_statistics_node_update>;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const int n = 5003, n2 = 1e5 + 123, n3 = 2e6 + 123;
int mx[n][n], h[n2], m, ind[22][n2];
ll p[n][n], s[n][n], dp[n3];

struct Node {
	int l = 0, r = 0, val = 0;
	vector <int> v;
};
Node t[n3];

void dfs(int v) {
	dp[v] = 0;
	for (int i : t[v].v) {
		dfs(i);
		dp[v] = max(dp[v], dp[i]);
	}
	if (t[v].val < 21)
		dp[v] += t[v].r - t[v].l + 1;
}

ll get(int v, int l, int r) {
	if (t[v].r < l || t[v].l > r) return 0;
	if (t[v].l >= l && t[v].r <= r) return dp[v];
	ll res = 0;
	for (int i : t[v].v)
		res = max(res, get(i, l, r));
	if (t[v].val < 21)
		res += max(0, min(r, t[v].r) - max(l, t[v].l) + 1);
	return res;
}

std::vector<long long> minimum_costs(std::vector<int> H, std::vector<int> L,
                                     std::vector<int> R) {
  int Q = L.size();
  std::vector<long long> C(Q);
  int N = SZ(H);
  for (int i = 0; i < N; i++) h[i] = H[i];
  
  if (N <= 5000 && Q <= 5000) {
	for (int i = 0; i < N; i++)
		for (int j = i; j < N; j++)
			mx[i][j] = (i == j) ? H[i] : max(mx[i][j - 1], H[j]);
	for (int j = 0; j < N; j++)
		for (int i = j; i >= 0; i--)
			p[i][j] = (i == j) ? H[i] : (p[i + 1][j] + mx[i][j]);
	for (int i = 0; i < N; i++)
		for (int j = i; j < N; j++)
			s[i][j] = (i == j) ? H[i] : (s[i][j - 1] + mx[i][j]);
	for (int i = 0; i < Q; i++) {
		int l = L[i], r = R[i];
		ll res = 1e18;
		for (int i = l; i <= r; i++)
			res = min(res, p[l][i] + s[i][r] - H[i]);
		C[i] = res;
	}
  }	else {
	  for (int i = 21; i >= 2; i--) {
		  int last = -1;
		  for (int j = 0; j <= N; j++) {
			  if (j == N || h[j] >= i) {
				  if (last + 1 <= j - 1) {
					  m++;
					  t[m].l = last + 1, t[m].r = j - 1;
					  for (int k = t[m].l; k <= t[m].r; k++)
						  ind[i][k] = m;
					  t[m].val = i;
					  if (i < 21)
						  t[ind[i + 1][j - 1]].v.pb(m);
				  }
				  last = j;
			  }
		  }
	  }
	  dfs(1);
	  for (int i = 0; i < Q; i++) {
		  int l = L[i], r = R[i];
		  l++, r++;
		  C[i] = (ll)20 * (r - l + 1) - get(1, l, r);
	  }
  }
  return C;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...