Submission #886269

#	Time	Username	Problem	Language	Result	Execution time	Memory
886269		javotaz	Factories (JOI14_factories)	C++17	100 / 100	2631 ms	176028 KiB

factories

// In the Name of Allah
 
#include<bits/stdc++.h>
using namespace std;

#pragma  GCC optimize ("O2,unroll-loops")
#pragma  GCC optimize ("Ofast")
#pragma  gcc optimize("Ofast")
#pragma  optimize(Ofast)

typedef long long ll;

const ll N = 5e5 + 12, inf = 1e18 + 12, M = 22;
vector<pair<int, int>> g[N];
int sz[N], cpar[N], q, n, h[N];
ll val[N], cdis[N][M];
bool mrk[N];

void dfs(int u, int par = -1) {
	sz[u] = 1;
	for (auto i: g[u])
		if (i.first != par && !mrk[i.first]) {
			dfs(i.first, u);
			sz[u] += sz[i.first];
		}
}

int find_centroid(int u, int c, int par = -1) {
	for (auto i: g[u])
		if (!mrk[i.first] && i.first != par && sz[i.first] > (c - 1) / 2)
			return find_centroid(i.first, c, u);
	return u;
}

void fill_cdis(int u, int lay, ll c = 0, int par = -1) {
	cdis[u][lay] = c;
	for (auto i: g[u])
		if (!mrk[i.first] && i.first != par)
			fill_cdis(i.first, lay, c + i.second, u);
}

void decompose(int u, int lay = 0, int par = -1) {
	dfs(u);
	u = find_centroid(u, sz[u]);
	fill_cdis(u, lay);
	cpar[u] = par;
	h[u] = lay;
	mrk[u] = true;
	for (auto i: g[u])
		if (!mrk[i.first])
			decompose(i.first, lay + 1, u);
}

void add(int u, int v, int lay) {
	if (u == -1)
		return;
	val[u] = min(val[u], cdis[v][lay]);
	add(cpar[u], v, lay - 1);
}

ll ask(int u, int v, int lay) {
	if (u == -1)
		return inf;
	return min(ask(cpar[u], v, lay - 1), cdis[v][lay] + val[u]);
}

void rem(int u) {
	val[u] = inf;
	if (cpar[u] != -1)
		rem(cpar[u]);
}

void Init(int c, int a[], int b[], int d[]) {
	n = c;
	for (int i = 0; i < n - 1; i++) {
		g[a[i]].push_back({b[i], d[i]});
		g[b[i]].push_back({a[i], d[i]});
	}
	decompose(0);
	fill(val, val + n, inf);
}

ll Query(int s, int x[], int t, int y[]) {
	for (int i = 0; i < s; i++)
		add(x[i], x[i], h[x[i]]);
	ll res = inf;
	for (int i = 0; i < t; i++)
		res = min(res, ask(y[i], y[i], h[y[i]]));
	for (int i = 0; i < s; i++)
		rem(x[i]);
	return res;
}

Compilation message (stderr)

factories.cpp:8: warning: ignoring '#pragma gcc optimize' [-Wunknown-pragmas]
    8 | #pragma  gcc optimize("Ofast")
      | 
factories.cpp:9: warning: ignoring '#pragma optimize ' [-Wunknown-pragmas]
    9 | #pragma  optimize(Ofast)
      |

• Subtask 1: 15 / 15
• Subtask 2: 18 / 18
• Subtask 3: 67 / 67