Submission #388033

#	Time	Username	Problem	Language	Result	Execution time	Memory
388033		rainboy	Dreaming (IOI13_dreaming)	C11	100 / 100	120 ms	21152 KiB

dreaming

#include "dreaming.h"
#include <stdio.h>
#include <stdlib.h>

#define N	100000
#define INF	0x3f3f3f3f

int min(int a, int b) { return a < b ? a : b; }
int max(int a, int b) { return a > b ? a : b; }

int *ej[N], *ew[N], eo[N];

void append(int i, int j, int w) {
	int o = eo[i]++;

	if (o >= 2 && (o & o - 1) == 0) {
		ej[i] = (int *) realloc(ej[i], o * 2 * sizeof *ej[i]);
		ew[i] = (int *) realloc(ew[i], o * 2 * sizeof *ew[i]);
	}
	ej[i][o] = j, ew[i][o] = w;
}

char visited[N]; int dp[N], ans;

void dfs1(int p, int i) {
	int o, d1, d2;

	if (visited[i])
		return;
	visited[i] = 1, d1 = 0, d2 = 0;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o], w = ew[i][o];

		if (j != p) {
			dfs1(i, j);
			if (d1 < dp[j] + w)
				d2 = d1, d1 = dp[j] + w;
			else if (d2 < dp[j] + w)
				d2 = dp[j] + w;
		}
	}
	ans = max(ans, d1 + d2);
	dp[i] = d1;
}

int r;

void dfs2(int p, int i, int d) {
	int o, d1, d2, j1;

	r = min(r, max(dp[i], d));
	d1 = 0, d2 = 0, j1 = -1;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o], w = ew[i][o];

		if (j != p) {
			if (d1 < dp[j] + w)
				d2 = d1, d1 = dp[j] + w, j1 = j;
			else if (d2 < dp[j] + w)
				d2 = dp[j] + w;
		}
	}
	for (o = eo[i]; o--; ) {
		int j = ej[i][o], w = ew[i][o];

		if (j != p)
			dfs2(i, j, max(d, j == j1 ? d2 : d1) + w);
	}
}

int travelTime(int n, int m, int l, int ii[], int jj[], int ww[]) {
	int h, i, r1, r2, r3;

	for (i = 0; i < n; i++) {
		ej[i] = (int *) malloc(2 * sizeof *ej[i]);
		ew[i] = (int *) malloc(2 * sizeof *ew[i]);
	}
	for (h = 0; h < m; h++)
		append(ii[h], jj[h], ww[h]), append(jj[h], ii[h], ww[h]);
	r1 = r2 = r3 = -INF;
	for (i = 0; i < n; i++)
		if (!visited[i]) {
			r = INF, dfs1(-1, i), dfs2(-1, i, 0);
			if (r1 < r)
				r3 = r2, r2 = r1, r1 = r;
			else if (r2 < r)
				r3 = r2, r2 = r;
			else if (r3 < r)
				r3 = r;
		}
	ans = max(ans, max(r1 + l + r2, r2 + l + l + r3));
	return ans;
}

Compilation message (stderr)

dreaming.c: In function 'append':
dreaming.c:16:23: warning: suggest parentheses around '-' in operand of '&' [-Wparentheses]
   16 |  if (o >= 2 && (o & o - 1) == 0) {
      |                     ~~^~~

• Subtask 1: 14 / 14
• Subtask 2: 10 / 10
• Subtask 3: 23 / 23
• Subtask 4: 18 / 18
• Subtask 5: 12 / 12
• Subtask 6: 23 / 23