Submission #349794

#	Time	Username	Problem	Language	Result	Execution time	Memory
349794		NachoLibre	Race (IOI11_race)	C++14	0 / 100	4 ms	5100 KiB

race

#include <bits/stdc++.h>
using namespace std;
#ifndef wambule
#include "race.h"
#endif

const int N = 200005, K = 1000006;
int mn, globk, ak[K], fp = -1;
vector<pair<int, int>> v[N], ve;
bool bo[N];

void minn(int &a, int b) {
	if(a == -1 || b < a) a = b;
}

int ctr(int dg, int op) {
	int dr = mn - 1;
	bool mo = 1;
	for(int i = 0; i < v[dg].size(); ++i) {
		int x = v[dg][i].first;
		if(x != op && !bo[x]) {
			int y = ctr(x, dg);
			if(y < 0) return y;
			if(y > mn / 2) mo = 0;
			dr -= y;
		}
	}
	if(dr > mn / 2) mo = 0;
	if(mo) return -dg;
	return mn - dr;
}

int sra(int dg, int op) {
	int dr = 1;
	for(int i = 0; i < v[dg].size(); ++i) {
		int x = v[dg][i].first;
		if(x != op && !bo[x]) {
			dr += sra(x, dg);
		}
	}
	return dr;
}

int cntr(int x = 0) {
	mn = sra(x, -1);
	return -ctr(x, -1);
}

void ovo(int dg, int op, int wr, int md) {
	if(md > globk) return;
	ve.push_back({wr, md});
	for(int i = 0; i < v[dg].size(); ++i) {
		int x = v[dg][i].first;
		if(x != op && !bo[x]) {
			ovo(x, dg, wr + 1, md + v[dg][i].second);
		}
	}
}

void ama(int dg) {
	bo[dg] = 1;
	for(int i = 0; i < v[dg].size(); ++i) {
		int x = v[dg][i].first;
		if(!bo[x]) ama(cntr(x));
	}
	vector<int> vey;
	for(int i = 0; i < v[dg].size(); ++i) {
		int x = v[dg][i].first;
		if(!bo[x]) {
			ve.clear();
			ovo(x, -1, 1, v[dg][i].second);
			for(int i = 0; i < ve.size(); ++i) {
				if(ve[i].second == globk) minn(fp, ve[i].first);
				else {
					vey.push_back(ve[i].second);
					if(ak[globk - ve[i].second]) {
						minn(fp, ak[globk - ve[i].second] + ve[i].first);
					}
				}
			}
			for(int i = 0; i < ve.size(); ++i) {
				if(ak[ve[i].second] == 0 || ak[ve[i].second] > ve[i].first) {
					ak[ve[i].second] = ve[i].first;
				}
			}
		}
	}
	for(int i = 0; i < vey.size(); ++i) {
		ak[vey[i]] = 0;
	}
}

int best_path(int n, int k, int h[][2], int l[]) {
	globk = k;
	for(int i = 0; i < n; ++i) {
		v[i].clear();
		bo[i] = 0;
	}
	for(int i = 0; i < n - 1; ++i) {
		v[h[i][0]].push_back({h[i][1], l[i]});
		v[h[i][1]].push_back({h[i][0], l[i]});
	}
	ama(cntr());
	return fp;
}

#ifdef wambule
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	return 0;
}
#endif

Compilation message (stderr)

race.cpp: In function 'int ctr(int, int)':
race.cpp:19:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |  for(int i = 0; i < v[dg].size(); ++i) {
      |                 ~~^~~~~~~~~~~~~~
race.cpp: In function 'int sra(int, int)':
race.cpp:35:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   35 |  for(int i = 0; i < v[dg].size(); ++i) {
      |                 ~~^~~~~~~~~~~~~~
race.cpp: In function 'void ovo(int, int, int, int)':
race.cpp:52:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |  for(int i = 0; i < v[dg].size(); ++i) {
      |                 ~~^~~~~~~~~~~~~~
race.cpp: In function 'void ama(int)':
race.cpp:62:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |  for(int i = 0; i < v[dg].size(); ++i) {
      |                 ~~^~~~~~~~~~~~~~
race.cpp:67:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   67 |  for(int i = 0; i < v[dg].size(); ++i) {
      |                 ~~^~~~~~~~~~~~~~
race.cpp:72:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   72 |    for(int i = 0; i < ve.size(); ++i) {
      |                   ~~^~~~~~~~~~~
race.cpp:81:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   81 |    for(int i = 0; i < ve.size(); ++i) {
      |                   ~~^~~~~~~~~~~
race.cpp:88:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   88 |  for(int i = 0; i < vey.size(); ++i) {
      |                 ~~^~~~~~~~~~~~

• Subtask 1: 0 / 9
• Subtask 2: 0 / 12
• Subtask 3: 0 / 22
• Subtask 4: 0 / 57