oj.uz

답안 #677104

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
677104 2023-01-02T09:47:07 Z SanguineChameleon Hard route (IZhO17_road) C++17
0 / 100
 21 ms 35560 KB 
// BEGIN BOILERPLATE CODE

#include <bits/stdc++.h>
using namespace std;

#ifdef KAMIRULEZ
	const bool kami_loc = true;
	const long long kami_seed = 69420;
#else
	const bool kami_loc = false;
	const long long kami_seed = chrono::steady_clock::now().time_since_epoch().count();
#endif

const string kami_fi = "kamirulez.inp";
const string kami_fo = "kamirulez.out";
mt19937_64 kami_gen(kami_seed);

long long rand_range(long long rmin, long long rmax) {
	uniform_int_distribution<long long> rdist(rmin, rmax);
	return rdist(kami_gen);
}

long double rand_real(long double rmin, long double rmax) {
	uniform_real_distribution<long double> rdist(rmin, rmax);
	return rdist(kami_gen);
}

void file_io(string fi, string fo) {
	if (fi != "" && (!kami_loc || fi == kami_fi)) {
		freopen(fi.c_str(), "r", stdin);
	}
	if (fo != "" && (!kami_loc || fo == kami_fo)) {
		freopen(fo.c_str(), "w", stdout);
	}
}

void set_up() {
	if (kami_loc) {
		file_io(kami_fi, kami_fo);
	}
	ios_base::sync_with_stdio(0);
	cin.tie(0);
}

void just_do_it();

void just_exec_it() {
	if (kami_loc) {
		auto pstart = chrono::steady_clock::now();
		just_do_it();
		auto pend = chrono::steady_clock::now();
		long long ptime = chrono::duration_cast<chrono::milliseconds>(pend - pstart).count();
		string bar(50, '=');
		cout << '\n' << bar << '\n';
		cout << "Time: " << ptime << " ms" << '\n';
	}
	else {
		just_do_it();
	}
}

int main() {
	set_up();
	just_exec_it();
	return 0;
}

// END BOILERPLATE CODE

// BEGIN MAIN CODE

#define int long long

const int ms = 5e5 + 20;
const int inf = 1e9 + 20;
vector<int> adj[ms];
pair<int, int> down[ms];
pair<int, int> up[ms];
pair<long long, long long> res;

struct group {
	int sz = 0;
	long long s1 = 0;
	long long s2 = 0;

	void add(int x) {
		sz++;
		s1 += x;
		s2 += 1LL * x * x;
	}

	void rem(int x) {
		sz--;
		s1 -= x;
		s2 -= 1LL * x * x;
	}
};

struct merge_ {
	map<int, group, greater<int>> f;

	void add(pair<int, int> x) {
		f[x.first].add(x.second);
	}

	void rem(pair<int, int> x) {
		auto it = f.find(x.first);
		it->second.rem(x.second);
		if (it->second.sz == 0) {
			f.erase(it);
		}
	}

	pair<int, long long> best1() {
		if (f.empty()) {
			return {-inf, 0};
		}
		auto it = f.begin();
		return {it->first, it->second.s1};
	}

	pair<int, long long> best2() {
		if (f.empty()) {
			return {-inf, 0};
		}
		auto it = f.begin();
		int x = it->first;
		long long s1 = it->second.s1;
		long long s2 = it->second.s2;
		if (it->second.sz > 1) {
			return {x + x, (s1 * s1 - s2) / 2};
		}
		it++;
		if (it == f.end()) {
			return {-inf, 0};
		}
		int y = it->first;
		s2 = it->second.s1;
		return {x + y, s1 * s2};
	}
};

merge_ g[ms];

pair<int, int> operator+(pair<int, int> x, int d) {
	return {x.first + d, x.second};
}

pair<int, long long> operator+(pair<int, long long> x, pair<int, long long> y) {
	if (x.first == -1 || y.first == 1) {
		return {-1, 0};
	}
	return {x.first + y.first, x.second * y.second};
}

pair<long long, long long> operator*(pair<int, long long> x, int d) {
	if (x.first == -1) {
		return {-1, 0};
	}
	return {1LL * x.first * d, x.second};
}

pair<long long, long long> max(pair<long long, long long> x, pair<long long, long long> y) {
	if (x.first > y.first) {
		return x;
	}
	else if (x.first < y.first) {
		return y;
	}
	else {
		return {x.first, x.second + y.second};
	}
}

// pair<int, int> max(pair<int, int> x, pair<int, int> y) {
// 	if (x.first > y.first) {
// 		return x;
// 	}
// 	else if (x.first < y.first) {
// 		return y;
// 	}
// 	else {
// 		return {x.first, x.second + y.second};
// 	}
// }

void dfs1(int u, int pr) {
	down[u] = {0, 1};
	for (auto v: adj[u]) {
		if (v != pr) {
			dfs1(v, u);
			down[u] = max(down[u], down[v] + 1);
		}
	}
}

void dfs2(int u, int pr) {
	pair<int, int> mx;
	for (auto v: adj[u]) {
		if (v != pr) {
			up[v] = max(up[u] + 1, mx + 2);
			mx = max(mx, down[v]);
		}
	}
	reverse(adj[u].begin(), adj[u].end());
	mx = {0, 0};
	for (auto v: adj[u]) {
		if (v != pr) {
			up[v] = max(up[v], mx + 2);
			dfs2(v, u);
			mx = max(mx, down[v]);
		}
	}
}

void dfs3(int u, int pr) {
	for (auto v: adj[u]) {
		if (v != pr) {
			g[u].add(down[v] + 1);
		}
	}
	if (pr != -1) {
		res = max(res, g[u].best2() * up[u].first);
	}
	for (auto v: adj[u]) {
		if (v != pr) {
			g[u].rem(down[v] + 1);
			res = max(res, g[u].best2() * (down[v].first + 1));
			if (pr != -1) {
				res = max(res, (g[u].best1() + up[u]) * (down[v].first + 1));
			}
			g[u].add(down[v] + 1);
		}
	}
	for (auto v: adj[u]) {
		if (v != pr) {
			dfs3(v, u);
		}
	}
}

void just_do_it() {
	int n;
	cin >> n;
	if (n == 2) {
		cout << "0 1";
		return;
	}
	for (int i = 0; i < n - 1; i++) {
		int u, v;
		cin >> u >> v;
		adj[u].push_back(v);
		adj[v].push_back(u);
	}
	int root = -1;
	int cnt = 0;
	for (int i = 1; i <= n; i++) {
		if ((int)adj[i].size() == 1) {
			cnt++;
		}
		else {
			if (root == -1) {
				root = i;
			}
		}
	}
	res = {0, 1LL * cnt * (cnt - 1) / 2};
	dfs1(root, -1);
	up[root] = {0, 1};
	dfs2(root, -1);
	dfs3(root, -1);
	cout << res.first << " " << res.second;
}

// END MAIN CODE

Compilation message

road.cpp: In function 'void file_io(std::string, std::string)':
road.cpp:30:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |   freopen(fi.c_str(), "r", stdin);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
road.cpp:33:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   33 |   freopen(fo.c_str(), "w", stdout);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 20 ms 35508 KB Output is correct
2 Correct 17 ms 35540 KB Output is correct
3 Correct 17 ms 35528 KB Output is correct
4 Correct 17 ms 35560 KB Output is correct
5 Correct 17 ms 35560 KB Output is correct
6 Correct 20 ms 35544 KB Output is correct
7 Correct 18 ms 35468 KB Output is correct
8 Correct 17 ms 35540 KB Output is correct
9 Correct 20 ms 35540 KB Output is correct
10 Correct 17 ms 35540 KB Output is correct
11 Correct 20 ms 35540 KB Output is correct
12 Correct 18 ms 35540 KB Output is correct
13 Correct 21 ms 35540 KB Output is correct
14 Correct 17 ms 35540 KB Output is correct
15 Correct 16 ms 35540 KB Output is correct
16 Correct 17 ms 35540 KB Output is correct
17 Correct 17 ms 35540 KB Output is correct
18 Correct 17 ms 35552 KB Output is correct
19 Correct 17 ms 35540 KB Output is correct
20 Correct 18 ms 35540 KB Output is correct
21 Incorrect 20 ms 35488 KB Output isn't correct
22 Halted 0 ms 0 KB -

Subtask #2
0 / 33.0

# 결과 실행 시간 메모리 Grader output
1 Correct 20 ms 35508 KB Output is correct
2 Correct 17 ms 35540 KB Output is correct
3 Correct 17 ms 35528 KB Output is correct
4 Correct 17 ms 35560 KB Output is correct
5 Correct 17 ms 35560 KB Output is correct
6 Correct 20 ms 35544 KB Output is correct
7 Correct 18 ms 35468 KB Output is correct
8 Correct 17 ms 35540 KB Output is correct
9 Correct 20 ms 35540 KB Output is correct
10 Correct 17 ms 35540 KB Output is correct
11 Correct 20 ms 35540 KB Output is correct
12 Correct 18 ms 35540 KB Output is correct
13 Correct 21 ms 35540 KB Output is correct
14 Correct 17 ms 35540 KB Output is correct
15 Correct 16 ms 35540 KB Output is correct
16 Correct 17 ms 35540 KB Output is correct
17 Correct 17 ms 35540 KB Output is correct
18 Correct 17 ms 35552 KB Output is correct
19 Correct 17 ms 35540 KB Output is correct
20 Correct 18 ms 35540 KB Output is correct
21 Incorrect 20 ms 35488 KB Output isn't correct
22 Halted 0 ms 0 KB -

Subtask #3
0 / 48.0

# 결과 실행 시간 메모리 Grader output
1 Correct 20 ms 35508 KB Output is correct
2 Correct 17 ms 35540 KB Output is correct
3 Correct 17 ms 35528 KB Output is correct
4 Correct 17 ms 35560 KB Output is correct
5 Correct 17 ms 35560 KB Output is correct
6 Correct 20 ms 35544 KB Output is correct
7 Correct 18 ms 35468 KB Output is correct
8 Correct 17 ms 35540 KB Output is correct
9 Correct 20 ms 35540 KB Output is correct
10 Correct 17 ms 35540 KB Output is correct
11 Correct 20 ms 35540 KB Output is correct
12 Correct 18 ms 35540 KB Output is correct
13 Correct 21 ms 35540 KB Output is correct
14 Correct 17 ms 35540 KB Output is correct
15 Correct 16 ms 35540 KB Output is correct
16 Correct 17 ms 35540 KB Output is correct
17 Correct 17 ms 35540 KB Output is correct
18 Correct 17 ms 35552 KB Output is correct
19 Correct 17 ms 35540 KB Output is correct
20 Correct 18 ms 35540 KB Output is correct
21 Incorrect 20 ms 35488 KB Output isn't correct
22 Halted 0 ms 0 KB -