Submission #67050782
Source Code Expand
Copy
#include <stdio.h>#include <stdlib.h>#include <inttypes.h>int N;int x[114514];struct edge_s {int to, cost;};int ec[114514];struct edge_s* es[114514];void ae(int f, int t, int c) {es[f] = realloc(es[f], sizeof(*es[f]) * (ec[f] + 1));if (es[f] == NULL) exit(2);es[f][ec[f]++] = (struct edge_s){ t, c };}uint64_t ans = 0;
#include <stdio.h> #include <stdlib.h> #include <inttypes.h> int N; int x[114514]; struct edge_s { int to, cost; }; int ec[114514]; struct edge_s* es[114514]; void ae(int f, int t, int c) { es[f] = realloc(es[f], sizeof(*es[f]) * (ec[f] + 1)); if (es[f] == NULL) exit(2); es[f][ec[f]++] = (struct edge_s){ t, c }; } uint64_t ans = 0; int dfs(int node, int parent) { int sum = x[node]; int i; for (i = 0; i < ec[node]; i++) { if (es[node][i].to != parent) { int cnt = dfs(es[node][i].to, node); sum += cnt; ans += (uint64_t)es[node][i].cost * abs(cnt); } } return sum; } int main(void) { int i; if (scanf("%d", &N) != 1) return 1; for (i = 1; i <= N; i++) { if (scanf("%d", &x[i]) != 1) return 1; } for (i = 1; i < N; i++) { int u, v, w; if (scanf("%d%d%d", &u, &v, &w) != 3) return 1; ae(u, v, w); ae(v, u, w); } dfs(1, 0); printf("%" PRIu64 "\n", ans); return 0; } /* 「迎えに行く」と「持ってくる」でコストは変わらない (予想) よって、適当な根を決めてそこに集めればいい (予想) */
Submission Info
Submission Time | |
---|---|
Task | E - Pair Annihilation |
User | mikecat |
Language | C (gcc 12.2.0) |
Score | 425 |
Code Size | 1139 Byte |
Status | AC |
Exec Time | 40 ms |
Memory | 11076 KiB |
Judge Result
Set Name | Sample | All | ||||
---|---|---|---|---|---|---|
Score / Max Score | 0 / 0 | 425 / 425 | ||||
Status |
|
|
Set Name | Test Cases |
---|---|
Sample | 00_sample_01.txt, 00_sample_02.txt, 00_sample_03.txt |
All | 00_sample_01.txt, 00_sample_02.txt, 00_sample_03.txt, 01_test_01.txt, 01_test_02.txt, 01_test_03.txt, 01_test_04.txt, 01_test_05.txt, 01_test_06.txt, 01_test_07.txt, 01_test_08.txt, 01_test_09.txt, 01_test_10.txt, 01_test_11.txt, 01_test_12.txt, 01_test_13.txt, 01_test_14.txt, 01_test_15.txt, 01_test_16.txt, 01_test_17.txt, 01_test_18.txt, 01_test_19.txt, 01_test_20.txt, 01_test_21.txt, 01_test_22.txt, 01_test_23.txt, 01_test_24.txt |
Case Name | Status | Exec Time | Memory |
---|---|---|---|
00_sample_01.txt | AC | 0 ms | 1628 KiB |
00_sample_02.txt | AC | 0 ms | 1628 KiB |
00_sample_03.txt | AC | 0 ms | 1632 KiB |
01_test_01.txt | AC | 11 ms | 2964 KiB |
01_test_02.txt | AC | 39 ms | 6540 KiB |
01_test_03.txt | AC | 5 ms | 2332 KiB |
01_test_04.txt | AC | 38 ms | 6572 KiB |
01_test_05.txt | AC | 35 ms | 6208 KiB |
01_test_06.txt | AC | 39 ms | 6524 KiB |
01_test_07.txt | AC | 24 ms | 4724 KiB |
01_test_08.txt | AC | 38 ms | 6660 KiB |
01_test_09.txt | AC | 35 ms | 6096 KiB |
01_test_10.txt | AC | 39 ms | 6656 KiB |
01_test_11.txt | AC | 38 ms | 6548 KiB |
01_test_12.txt | AC | 39 ms | 6664 KiB |
01_test_13.txt | AC | 16 ms | 3640 KiB |
01_test_14.txt | AC | 39 ms | 6660 KiB |
01_test_15.txt | AC | 18 ms | 3976 KiB |
01_test_16.txt | AC | 40 ms | 6588 KiB |
01_test_17.txt | AC | 11 ms | 3200 KiB |
01_test_18.txt | AC | 38 ms | 6540 KiB |
01_test_19.txt | AC | 26 ms | 4992 KiB |
01_test_20.txt | AC | 39 ms | 6516 KiB |
01_test_21.txt | AC | 26 ms | 11076 KiB |
01_test_22.txt | AC | 26 ms | 10992 KiB |
01_test_23.txt | AC | 27 ms | 7116 KiB |
01_test_24.txt | AC | 24 ms | 7140 KiB |