結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
|
提出日時 | 2024-07-14 23:24:53 |
言語 | C# (.NET 8.0.404) |
結果 |
AC
|
実行時間 | 229 ms / 2,000 ms |
コード長 | 1,440 bytes |
コンパイル時間 | 8,429 ms |
コンパイル使用メモリ | 167,976 KB |
実行使用メモリ | 184,628 KB |
最終ジャッジ日時 | 2024-07-14 23:25:10 |
合計ジャッジ時間 | 16,465 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
コンパイルメッセージ
復元対象のプロジェクトを決定しています... /home/judge/data/code/main.csproj を復元しました (97 ms)。 MSBuild のバージョン 17.9.6+a4ecab324 (.NET) main -> /home/judge/data/code/bin/Release/net8.0/main.dll main -> /home/judge/data/code/bin/Release/net8.0/publish/
ソースコード
using System;using static System.Console;using System.Linq;using System.Collections.Generic;class Program{static int NN => int.Parse(ReadLine());static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();static int[][] NArr(long n) => Enumerable.Repeat(0, (int)n).Select(_ => NList).ToArray();static int[] LList(long n) => Enumerable.Repeat(0, (int)n).Select(_ => int.Parse(ReadLine())).ToArray();public static void Main(){Solve();}static void Solve(){var n = NN;var map = NArr(n - 1);var tree = new List<int>[n];for (var i = 0; i < n; ++i) tree[i] = new List<int>();foreach (var edge in map){tree[edge[0] - 1].Add(edge[1] - 1);tree[edge[1] - 1].Add(edge[0] - 1);}DFS(0, -1, tree);WriteLine(ans);}static long ans = 0;static int DFS(int cur, int prev, List<int>[] tree){var sum = 1;var list = new List<int>();foreach (var next in tree[cur]){if (prev == next) continue;list.Add(DFS(next, cur, tree));sum += list[^1];}list.Add(tree.Length - sum);foreach (var li in list){ans += (long)li * (tree.Length - li);}ans += tree.Length;return sum;}}