oj.uz

Submission #53873

# Submission time Handle Problem Language Result Execution time Memory
53873 2018-07-01T13:13:22 Z baactree Islands (IOI08_islands) C++17
100 / 100
 1602 ms 171656 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
struct SGT {
	vector<ll> tree;
	int sz;
	SGT(int n) {
		sz = 1;
		while (sz < n)sz *= 2;
		tree.resize(sz * 2);
	}
	ll query(int le, int ri) {
		le += sz;
		ri += sz;
		ll ret = 0;
		while (le <= ri) {
			if (le & 1)ret = max(ret, tree[le++]);
			if (!(ri & 1))ret = max(ret, tree[ri--]);
			le /= 2;
			ri /= 2;
		}
		return ret;
	}
};
int n;
vector<tuple<int,int,int> > adj[1000005];
int trip[1000005], w[1000005];
bool chk[1000005], chk2[1000005];
vector<int> cycle,len;
ll ds[1000005], dd[1000005];
vector<ll> d;
int r[1000005];
int vn;
void dfs(int now, int pi) {
	r[now] = trip[now] = ++vn;
	for (auto &e : adj[now]) {
		if (get<2>(e)==pi)continue;
		if (!trip[get<0>(e)]) {
			dfs(get<0>(e), get<2>(e));
			r[now] = min(r[now], r[get<0>(e)]);
		}
		else if(!chk[get<0>(e)]){
			r[now] = min(r[now], trip[get<0>(e)]);
			cycle.push_back(get<0>(e));
			chk[get<0>(e)] = true;
			len.push_back(get<1>(e));
		}
	}
	if (r[now] < trip[now]) {
		chk[now] = true;
		cycle.push_back(now);
		len.push_back(w[pi]);
	}
}
ll val;
void dfs2(int now) {
	chk[now] = true;
	for (auto &e : adj[now]) {
		if (chk[get<0>(e)])continue;
		dfs2(get<0>(e));
		val = max(val, dd[now] + dd[get<0>(e)] + get<1>(e));
		dd[now] = max(dd[now], dd[get<0>(e)] + get<1>(e));
	}
}
ll calc(vector<ll> &d, vector<int> &len, int m) {
	for (int i = 1; i < m; i++)
		ds[i] = ds[i - 1] + len[i - 1];
	SGT sgt(m * 2);
	ll sum = len[0];
	for (int i = 1; i < m * 2; i++) {
		sgt.tree[i + sgt.sz] = sum + d[i%m];
		sum += len[i%m];
	}
	for (int i = sgt.sz - 1; i; i--)
		sgt.tree[i] = max(sgt.tree[i * 2], sgt.tree[i * 2 + 1]);
	ll ret = 0;
	for (int i = 0; i < m; i++) {
		ll temp = d[i] + sgt.query(i + 1, i + m - 1) - ds[i];
		ret = max(ret, temp);
	}
	return ret;
}
int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; i++) {
		int a, b;
		scanf("%d%d", &a, &b);
		adj[i].emplace_back(a, b, i);
		adj[a].emplace_back(i, b, i);
		w[i] = b;
	}
	ll ans = 0;
	for (int i = 1; i <= n; i++) {
		if (!trip[i]) {
			cycle.clear();
			len.clear();
			dfs(i, 0);
			int m = cycle.size();
			d = vector<ll>(m);
			val = 0;
			for (int i = 0; i < cycle.size(); i++) {
				dfs2(cycle[i]);
				d[i] = dd[cycle[i]];
			}
			val = max(val,calc(d, len, m));
			reverse(len.begin(), len.end() - 1);
			reverse(d.begin(), d.end());
			val = max(val, calc(d, len, m));
			ans += val;
		}
	}
	printf("%lld\n", ans);
	return 0;
}

Compilation message

islands.cpp: In function 'int main()':
islands.cpp:101:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for (int i = 0; i < cycle.size(); i++) {
                    ~~^~~~~~~~~~~~~~
islands.cpp:84:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d", &n);
  ~~~~~^~~~~~~~~~
islands.cpp:87:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d", &a, &b);
   ~~~~~^~~~~~~~~~~~~~~~

Subtask #1
33.0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 25 ms 23800 KB Output is correct
2 Correct 21 ms 23908 KB Output is correct
3 Correct 23 ms 23984 KB Output is correct
4 Correct 22 ms 24000 KB Output is correct
5 Correct 21 ms 24056 KB Output is correct
6 Correct 23 ms 24056 KB Output is correct
7 Correct 21 ms 24108 KB Output is correct
8 Correct 22 ms 24108 KB Output is correct
9 Correct 21 ms 24108 KB Output is correct
10 Correct 22 ms 24108 KB Output is correct
11 Correct 23 ms 24108 KB Output is correct

Subtask #2
3.0 / 3.0

# Verdict Execution time Memory Grader output
1 Correct 22 ms 24212 KB Output is correct
2 Correct 23 ms 24212 KB Output is correct

Subtask #3
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 22 ms 24212 KB Output is correct
2 Correct 24 ms 24552 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 32 ms 25336 KB Output is correct
2 Correct 49 ms 28152 KB Output is correct
3 Correct 37 ms 28152 KB Output is correct
4 Correct 28 ms 28152 KB Output is correct

Subtask #5
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 59 ms 29932 KB Output is correct
2 Correct 81 ms 32648 KB Output is correct

Subtask #6
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 138 ms 39020 KB Output is correct
2 Correct 171 ms 44596 KB Output is correct
3 Correct 243 ms 59208 KB Output is correct

Subtask #7
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 306 ms 59208 KB Output is correct
2 Correct 412 ms 79068 KB Output is correct
3 Correct 404 ms 79068 KB Output is correct
4 Correct 581 ms 108844 KB Output is correct

Subtask #8
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 473 ms 108844 KB Output is correct
2 Correct 1317 ms 135736 KB Output is correct
3 Correct 529 ms 135736 KB Output is correct
4 Correct 756 ms 135736 KB Output is correct
5 Correct 734 ms 135736 KB Output is correct
6 Correct 1602 ms 135736 KB Output is correct
7 Correct 829 ms 163640 KB Output is correct

Subtask #9
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 587 ms 163640 KB Output is correct
2 Correct 562 ms 163640 KB Output is correct
3 Correct 833 ms 171656 KB Output is correct
4 Correct 683 ms 171656 KB Output is correct
5 Correct 682 ms 171656 KB Output is correct
6 Correct 569 ms 171656 KB Output is correct
7 Correct 1565 ms 171656 KB Output is correct