제출 #741221

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
741221		rainboy	Museum (CEOI17_museum)	C11	100 / 100	205 ms	2248 KiB

museum

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define N	10000
#define INF	0x3f3f3f3f

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

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

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

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

int *dp[N], *dq[N], sz[N];

void dfs(int p, int c_, int i) {
	int o, s, s1, s2;

	sz[i] = 1;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o], c = ec[i][o];

		if (j != p) {
			dfs(i, c, j);
			sz[i] += sz[j];
		}
	}
	dp[i] = (int *) malloc((sz[i] + 1) * sizeof *dp[i]);
	dq[i] = (int *) malloc((sz[i] + 1) * sizeof *dq[i]);
	memset(dp[i], 0x3f, (sz[i] + 1) * sizeof *dp[i]);
	memset(dq[i], 0x3f, (sz[i] + 1) * sizeof *dq[i]);
	sz[i] = 1;
	dp[i][0] = 0, dq[i][0] = 0, dp[i][1] = 0, dq[i][1] = 0;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p) {
			for (s1 = sz[i]; s1 > 0; s1--)
				for (s2 = sz[j]; s2 > 0; s2--) {
					s = s1 + s2;
					dp[i][s] = min(dp[i][s], dp[i][s1] + dp[j][s2]);
					dq[i][s] = min(dq[i][s], min(dq[i][s1] + dp[j][s2], dp[i][s1] + dq[j][s2]));
				}
			free(dp[j]), free(dq[j]);
			sz[i] += sz[j];
		}
	}
	for (s = 1; s <= sz[i]; s++)
		dp[i][s] += c_ * 2, dq[i][s] += c_;
}

int main() {
	int n, k, r, h, i, j, c;

	scanf("%d%d%d", &n, &k, &r), r--;
	for (i = 0; i < n; i++) {
		ej[i] = (int *) malloc(2 * sizeof *ej[i]);
		ec[i] = (int *) malloc(2 * sizeof *ec[i]);
	}
	for (h = 0; h < n - 1; h++) {
		scanf("%d%d%d", &i, &j, &c), i--, j--;
		append(i, j, c), append(j, i, c);
	}
	dfs(-1, 0, r);
	printf("%d\n", dq[r][k]);
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

museum.c: In function 'append':
museum.c:15:23: warning: suggest parentheses around '-' in operand of '&' [-Wparentheses]
   15 |  if (o >= 2 && (o & o - 1) == 0) {
      |                     ~~^~~
museum.c: In function 'main':
museum.c:63:2: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   63 |  scanf("%d%d%d", &n, &k, &r), r--;
      |  ^~~~~~~~~~~~~~~~~~~~~~~~~~~
museum.c:69:3: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
   69 |   scanf("%d%d%d", &i, &j, &c), i--, j--;
      |   ^~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 20 / 20
• Subtask 2: 25 / 25
• Subtask 3: 35 / 35
• Subtask 4: 20 / 20