Submission #1055260

#	Time	Username	Problem	Language	Result	Execution time	Memory
1055260		lovrot	Islands (IOI08_islands)	C++17	44 / 100	786 ms	131072 KiB

islands

#include <cstdio> 
#include <vector>
#include <algorithm> 
#include <set>

#define debug(...) fprintf(stderr, __VA_ARGS__)
#define X first
#define Y second
#define PB push_back

using namespace std;

typedef long long ll;
typedef pair<int, int> pii;

const int N = 1e6 + 10;

int n;
ll weg[N]; 
vector<pii> g[N];

int bio[N];
vector<pair<int, ll>> cyc;

int dfs(int u, int p) { 
	bio[u] = 1;
	for(pii e : g[u]) { 
		int v = e.X, w = e.Y;
		if(w != p) { 
			if(bio[v]) { 
				cyc.PB({e.X, weg[e.Y]});
				return v;
			} else {
				int res = dfs(v, w);
				if(res) { 
					cyc.PB({e.X, weg[e.Y]});
					return res != u ? res : 0;
				}
			}
		}
	}
	bio[u] = 0;
	return 0;
}

void mark(int u) { 
	bio[u] = 2;
	for(pii e : g[u]) { 
		if(bio[e.X] != 2) { 
			mark(e.X);
		}
	}
}

ll dep[N];

ll depth(int u, int p) { 
	for(pii e : g[u]) { 
		int v = e.X, w = e.Y;
		if(w != p && !bio[v]) { 
			dep[u] = max(dep[u], (ll) weg[w] + depth(v, w));
		}
	}
	return dep[u];
}

ll in[N];
multiset<ll> s;

int main() { 
	scanf("%d", &n);
	for(int i = 1; i <= n; ++i) { 
		int u;
		scanf("%d%lld", &u, weg + i);
		g[i].PB({u, i});
		g[u].PB({i, i});
	}

	ll ans = 0;
	for(int i = 1; i <= n; ++i) { 
		if(bio[i]) { continue; }

		cyc.clear();
		s.clear();
		dfs(i, 0);

		for(pii e : cyc) { 
			depth(e.X, 0);
			debug("%d(%d, %lld)\n", e.X, e.Y, dep[e.X]);
		}

		mark(i);

		ll prf = 0;
		for(pii e : cyc) { 
			int v = e.X, w = e.Y;	
			s.insert(dep[v] + prf);
			prf += w;
		}

		ll prf_ = 0, res = 0, len = prf;
		for(pii e : cyc) {
			int v = e.X, w = e.Y;
			s.erase(dep[v] + prf - len);

			res = max(res, *prev(s.end()) + dep[v] - prf + len);

			s.insert(dep[v] + prf);
			prf += w;
		}

		ans += res;
	}

	printf("%lld\n", ans);
	return 0;
}

Compilation message (stderr)

islands.cpp: In function 'int main()':
islands.cpp:101:6: warning: unused variable 'prf_' [-Wunused-variable]
  101 |   ll prf_ = 0, res = 0, len = prf;
      |      ^~~~
islands.cpp:71:7: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   71 |  scanf("%d", &n);
      |  ~~~~~^~~~~~~~~~
islands.cpp:74:8: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   74 |   scanf("%d%lld", &u, weg + i);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 27 / 33
• Subtask 2: 3 / 3
• Subtask 3: 4 / 4
• Subtask 4: 0 / 10
• Subtask 5: 10 / 10
• Subtask 6: 0 / 10
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 10