結果

問題 No.2618 除霊
ユーザー kakel-sankakel-san
提出日時 2024-01-26 22:23:20
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 417 ms / 2,000 ms
コード長 1,955 bytes
コンパイル時間 7,829 ms
コンパイル使用メモリ 167,464 KB
実行使用メモリ 237,660 KB
最終ジャッジ日時 2024-09-28 08:25:28
合計ジャッジ時間 25,967 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 43
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (84 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/

ソースコード

diff # 

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[] NMi => ReadLine().Split().Select(c => int.Parse(c) - 1).ToArray();
    static int[][] NMap(int n) => Enumerable.Repeat(0, n).Select(_ => NMi).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var n = NN;
        var map = NMap(n - 1);
        var m = NN;
        var v = NMi;

        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]].Add(edge[1]);
            tree[edge[1]].Add(edge[0]);
        }

        var g = new bool[n];
        foreach (var vi in v) g[vi] = true;
        var nx = new int[n];
        for (var i = 0; i < n; ++i)
        {
            foreach (var next in tree[i]) if (g[next]) ++nx[i];
        }

        var olist = (bool[]) g.Clone();
        for (var i = 0; i < n; ++i) if (g[i]) foreach (var next in tree[i]) olist[next] = true;
        var org = olist.Count(f => f);

        var rem = new int[n];
        for (var i = 0; i < n; ++i)
        {
            foreach (var n2 in tree[i])
            {
                if (!g[n2] && g[i] && nx[n2] == 1) ++rem[i];
            }
        }
        var ans = Enumerable.Repeat(org, n).ToArray();
        for (var i = 0; i < n; ++i)
        {
            if (g[i] || nx[i] > 0) --ans[i];
            foreach (var n1 in tree[i])
            {
                if ((g[n1] || nx[n1] > 0) && nx[n1] == (g[i] ? 1 : 0)) --ans[i];
                ans[i] -= rem[n1];
                if (!g[i] && g[n1] && nx[i] == 1) ++ans[i];
            }
        }
        WriteLine(string.Join("\n", ans));
    }
}
0