제출 #501179

#제출 시각아이디문제언어결과실행 시간메모리
501179rainboyWells (CEOI21_wells)C11
50 / 100
388 ms503908 KiB
#include <stdio.h>
#include <string.h>

#define N	10000
#define INF	0x3f3f3f3f3f3f3f3fLL
#define MD	1000000007

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

int mem[(N - 1) * 2], *p = mem, *ej[N], eo[N], n, k;

int ll[N], jj[N], ta[N], tb[N], kk[N + 1], qu[N * 2];

void dfs1(int p, int i, int d) {
	int o;

	kk[d + 1]++;
	ll[i] = 0, jj[i] = -1;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p) {
			dfs1(i, j, d + 1);
			if (ll[i] < ll[j])
				ll[i] = ll[j], jj[i] = j;
		}
	}
	ll[i]++;
}

void dfs2(int p, int i, int d) {
	static int time;
	int o;

	qu[ta[i] = time++] = i, tb[i] = kk[d]++;
	if (jj[i] != -1)
		dfs2(i, jj[i], d + 1);
	else
		time++;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p && j != jj[i])
			dfs2(i, j, d + 1);
	}
}

int cnt[N * 2]; long long xx[N + 1];

void get_range(int a, int *l, int *r) {
	*l = tb[qu[a]], *r = *l + cnt[a];
}

void dfs3(int p, int i) {
	int o, d, l1, r1, l2, r2;
	long long c;

	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p)
			dfs3(i, j);
	}
	if (k <= ll[i]) {
		get_range(ta[i] + k - 1, &l1, &r1);
		xx[l1]++, xx[r1]--;
		if (p != -1)
			xx[tb[p]] -= r1 - l1, xx[tb[p] + 1] += r1 - l1;
	}
	cnt[ta[i]]++;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p && j != jj[i]) {
			for (d = 1; d <= ll[j] && d < k; d++)
				if (k - d <= ll[i]) {
					get_range(ta[i] + k - d - 1, &l1, &r1), get_range(ta[j] + d - 1, &l2, &r2);
					c = (long long) (r1 - l1) * (r2 - l2);
					xx[l1] += r2 - l2, xx[r1] -= r2 - l2;
					xx[l2] += r1 - l1, xx[r2] -= r1 - l1;
					xx[tb[i]] -= c, xx[tb[i] + 1] += c;
					if (p != -1)
						xx[tb[p]] -= c, xx[tb[p] + 1] += c;
				}
			for (d = 1; d <= ll[j]; d++)
				cnt[ta[i] + d] += cnt[ta[j] + d - 1];
		}
	}
}

long long ww[N];

void dfs4(int p, int i) {
	int o;

	ww[i] = xx[tb[i]];
	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p) {
			dfs4(i, j);
			ww[i] += ww[j];
		}
	}
}

long long dp[N][N + 1]; int dq[N][N + 1], sz[N + 1];

void dfs5(int p, int i) {
	int o, di, dj;

	dp[i][0] = ww[i], dq[i][0] = 1, dp[i][1] = 0, dq[i][1] = 1, sz[i] = 1;
	for (o = eo[i]; o--; ) {
		int j = ej[i][o];

		if (j != p) {
			dfs5(i, j);
			memset(dp[i] + sz[i] + 1, 0x3f, sz[j] * sizeof *dp[i]);
			memset(dq[i] + sz[i] + 1, 0, sz[j] * sizeof *dq[i]);
			for (di = sz[i]; di >= 0; di--) {
				long long pi = dp[i][di];
				int qi = dq[i][di];

				dp[i][di] = INF, dq[i][di] = 0;
				for (dj = sz[j]; dj >= 0; dj--) {
					long long pj = dp[j][dj];
					int qj = dq[j][dj];

					if (di + dj < k) {
						int d = di == 0 ? 0 : max(di, dj + 1);
						long long p = pi + pj;
						int q = (long long) qi * qj % MD;

						if (dp[i][d] > p)
							dp[i][d] = p, dq[i][d] = 0;
						if (dp[i][d] == p)
							dq[i][d] = (dq[i][d] + q) % MD;
					}
				}
			}
			sz[i] += sz[j];
		}
	}
}

int main() {
	static int uu[N - 1], vv[N - 1];
	int h, i, j, d, yes, q;
	long long w;

	scanf("%d%d", &n, &k);
	for (h = 0; h < n - 1; h++) {
		scanf("%d%d", &i, &j), i--, j--;
		uu[h] = i, vv[h] = j;
		eo[i]++, eo[j]++;
	}
	for (i = 0; i < n; i++) {
		ej[i] = p, p += eo[i];
		eo[i] = 0;
	}
	for (h = 0; h < n - 1; h++) {
		i = uu[h], j = vv[h];
		ej[i][eo[i]++] = j, ej[j][eo[j]++] = i;
	}
	dfs1(-1, 0, 0);
	for (d = 1; d <= n; d++)
		kk[d] += kk[d - 1];
	memset(qu, -1, n * 2 * sizeof *qu);
	dfs2(-1, 0, 0);
	dfs3(-1, 0);
	for (i = 1; i <= n; i++)
		xx[i] += xx[i - 1];
	dfs4(-1, 0);
	w = 0;
	for (i = 0; i < n; i++)
		w += ww[i];
	w /= k;
	dfs5(-1, 0);
	yes = 0, q = 0;
	for (d = 0; d < n; d++)
		if (dp[0][d] == w)
			yes = 1, q = (q + dq[0][d]) % MD;
	printf(yes ? "YES\n" : "NO\n");
	printf("%d\n", q);
	return 0;
}

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

wells.c: In function 'main':
wells.c:152:2: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
  152 |  scanf("%d%d", &n, &k);
      |  ^~~~~~~~~~~~~~~~~~~~~
wells.c:154:3: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
  154 |   scanf("%d%d", &i, &j), i--, j--;
      |   ^~~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...