結果

問題 No.1637 Easy Tree Query
ユーザー kakel-sankakel-san
提出日時 2021-08-06 21:29:26
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 358 ms / 2,000 ms
コード長 1,353 bytes
コンパイル時間 922 ms
コンパイル使用メモリ 115,580 KB
実行使用メモリ 51,196 KB
最終ジャッジ日時 2024-09-17 03:40:04
合計ジャッジ時間 10,145 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 33
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using System.Collections.Generic;
using static System.Console;
using System.Linq;

class yuki308
{
    static int NN => int.Parse(ReadLine());
    static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();
    static int[][] NMap(int n) => Enumerable.Repeat<int>(0, n).Select(_ => NList).ToArray();
  
    static void Main()
    {
        var c = NList;
        var (n, q) = (c[0], c[1]);
        var map = NMap(n - 1);
        var queries = NMap(q);

        var tree = new List<int>[n];
        for (var i = 0; i < tree.Length; ++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);
        }

        var cnum = new int[n];
        DFS(-1, 0, tree, cnum);

        var res = 0L;
        var rlist = new long[q];
        for (var i = 0; i < q; ++i)
        {
            res += (long)cnum[queries[i][0] - 1] * queries[i][1];
            rlist[i] = res;
        }
        WriteLine(string.Join("\n", rlist));
    }
    static int DFS(int prev, int cur, List<int>[] tree, int[] cnum)
    {
        var num = 1;
        foreach (var next in tree[cur])
        {
            if (prev == next) continue;
            num += DFS(cur, next, tree, cnum);
        }
        cnum[cur] = num;
        return num;
    }
}
0