oj.uz

Submission #256114

# Submission time Handle Problem Language Result Execution time Memory
256114 2020-08-02T09:57:39 Z mode149256 Islands (IOI08_islands) C++14
30 / 100
 705 ms 131076 KB 
/*input
3
2 11
3 10
1 11
*/
#include <bits/stdc++.h>
using namespace std;

namespace my_template {
typedef long long ll;
typedef long double ld;
typedef complex<ld> cd;

typedef pair<int, int> pi;
typedef pair<ll, ll> pl;
typedef pair<ld, ld> pd;

typedef vector<int> vi;
typedef vector<vi> vii;
typedef vector<ld> vd;
typedef vector<ll> vl;
typedef vector<vl> vll;
typedef vector<pi> vpi;
typedef vector<vpi> vpii;
typedef vector<pl> vpl;
typedef vector<cd> vcd;
typedef vector<pd> vpd;
typedef vector<bool> vb;
typedef vector<vb> vbb;
typedef std::string str;
typedef std::vector<str> vs;

#define x first
#define y second
#define debug(...) cout<<"["<<#__VA_ARGS__<<": "<<__VA_ARGS__<<"]\n"

const ld PI = 3.14159265358979323846264338327950288419716939937510582097494L;

template<typename T>
pair<T, T> operator+(const pair<T, T> &a, const pair<T, T> &b) { return pair<T, T>(a.x + b.x, a.y + b.y); }
template<typename T>
pair<T, T> operator-(const pair<T, T> &a, const pair<T, T> &b) { return pair<T, T>(a.x - b.x, a.y - b.y); }
template<typename T>
T operator*(const pair<T, T> &a, const pair<T, T> &b) { return (a.x * b.x + a.y * b.y); }
template<typename T>
T operator^(const pair<T, T> &a, const pair<T, T> &b) { return (a.x * b.y - a.y * b.x); }

template<typename T>
void print(vector<T> vec, string name = "") {
	cout << name;
	for (auto u : vec)
		cout << u << ' ';
	cout << '\n';
}
}
using namespace my_template;

const int MOD = 1000000007;
const ll INF = std::numeric_limits<ll>::max();
const int MX = 1000101;

int N;
vpii edges(MX);
vpii edgesVisos(MX);

vb been;

vl circle; // lens
vi circleV;
vb inCircle(MX, false);

ll ats = 0;
ll proc = 0;
ll maxDis = 0;
int kur;

void fill(int x) {
	been[x] = true;

	for (auto u : edgesVisos[x])
		if (!been[u.x])
			fill(u.x);
}

// -1 ner rato
int find_circle(int x) {
	been[x] = true;

	for (auto u : edges[x]) {
		if (been[u.x]) {
			// printf("x = %d, u = %d %d\n", x, u.x, u.y);
			circleV.emplace_back(x);
			circle.emplace_back(u.y);
			been[x] = false;
			return u.x;
		}

		int k = find_circle(u.x);

		if (k == x) {
			// printf("x = %d, u = %d %d\n", x, u.x, u.y);
			circleV.emplace_back(x);
			circle.emplace_back(u.y);
			been[x] = false;
			return -1;
		}
		else if (k != -1) {
			// printf("x = %d, u = %d %d\n", x, u.x, u.y);
			circleV.emplace_back(x);
			circle.emplace_back(u.y);
			been[x] = false;
			return k;
		}
	}

	been[x] = false;
	return -1;
}

void longest(int x, int p, ll dis) {
	if (dis > maxDis) {
		maxDis = dis;
		kur = x;
	}

	for (auto u : edgesVisos[x]) {
		if (u.x == p or inCircle[u.x]) continue;
		longest(u.x, x, dis + u.y);
	}
}

void solve() {
	proc = 0;
	for (auto u : circleV)
		inCircle[u] = true;

	int C = (int)circleV.size();

	vl D(C, 0);

	for (int i = 0; i < C; ++i)
	{
		int x = circleV[i];
		vl kokie;

		for (auto u : edgesVisos[x]) {
			if (inCircle[u.x]) continue;
			maxDis = 0;
			longest(u.x, x, u.y);
			proc = max(proc, maxDis);
			kokie.push_back(maxDis);

			D[i] = max(D[i], maxDis);

			maxDis = 0;
			longest(kur, -1, 0);
			proc = max(proc, maxDis);

			// printf("i = %d, d = %d\n", i, D[i]);
		}

		if (kokie.size() >= 2) {
			sort(kokie.rbegin(), kokie.rend());
			proc = max(proc, kokie[0] + kokie[1]);
		}
	}

	// for (int i = 0; i < C; ++i) printf("v = %d, dis = %lld\n", circleV[i]+1, D[i]);

	deque<pi> dq;
	ll viso = accumulate(circle.begin(), circle.end(), 0);

	{
		for (int i = 1; i < C; ++i)
		{
			circle[i] += circle[i - 1];
			ll newVal = circle[i - 1] + D[i];

			while (dq.size() and dq.back().x <= newVal)
				dq.pop_back();

			dq.push_back({newVal, i});
		}

		for (int i = 0; i < C; ++i)
		{
			while (dq.size() and dq.front().y <= i) {
				dq.pop_front();
			}

			assert(dq.size());
			proc = max(proc, dq.front().x + D[i] - (i ? circle[i - 1] : 0));

			ll newVal = D[i] + viso + (i ? circle[i - 1] : 0);
			while (dq.size() and dq.back().x <= newVal)
				dq.pop_back();

			dq.push_back({newVal, i + C});
		}
	}

	ats += proc;
}

int main() {
	ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
	cin >> N;
	for (int i = 0; i < N; ++i)
	{
		int a, w;
		cin >> a >> w; a--;

		// edges[a].emplace_back(i, w);
		edges[i].emplace_back(a, w);

		edgesVisos[i].emplace_back(a, w);
		edgesVisos[a].emplace_back(i, w);
	}

	been.resize(N, false);
	for (int i = 0; i < N; ++i)
	{
		if (been[i]) continue;

		circle.clear();
		circleV.clear();

		find_circle(i);
		fill(i);
		reverse(circle.begin(), circle.end());
		reverse(circleV.begin(), circleV.end());

		// for (auto u : circleV)
		// 	printf("%d ", u + 1);
		// printf("\n");
		// for (auto u : circle)
		// 	printf("%lld ", u);
		// printf("\n");

		solve();
	}

	printf("%lld\n", ats);
}

/* Look for:
* special cases (n=1?)
* overflow (ll vs int?)
* the exact constraints (multiple sets are too slow for n=10^6 :( )
* array bounds
*/

Subtask #1
30.0 / 33.0

# Verdict Execution time Memory Grader output
1 Correct 29 ms 47480 KB Output is correct
2 Correct 30 ms 47488 KB Output is correct
3 Incorrect 28 ms 47488 KB Output isn't correct
4 Correct 28 ms 47488 KB Output is correct
5 Correct 29 ms 47488 KB Output is correct
6 Correct 29 ms 47488 KB Output is correct
7 Correct 35 ms 47480 KB Output is correct
8 Correct 28 ms 47488 KB Output is correct
9 Correct 29 ms 47488 KB Output is correct
10 Correct 29 ms 47516 KB Output is correct
11 Correct 31 ms 47488 KB Output is correct

Subtask #2
0 / 3.0

# Verdict Execution time Memory Grader output
1 Incorrect 31 ms 47504 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 35 ms 47612 KB Output is correct
2 Incorrect 36 ms 47888 KB Output isn't correct

Subtask #4
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 45 ms 48896 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #5
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 69 ms 53364 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #6
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 140 ms 65400 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #7
0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 279 ms 80888 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #8
0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 590 ms 131072 KB Output is correct
2 Runtime error 705 ms 131076 KB Execution killed with signal 9 (could be triggered by violating memory limits)
3 Halted 0 ms 0 KB -

Subtask #9
0 / 10.0

# Verdict Execution time Memory Grader output
1 Runtime error 479 ms 131076 KB Execution killed with signal 9 (could be triggered by violating memory limits)
2 Halted 0 ms 0 KB -