結果

問題 No.2726 Rooted Tree Nim
コンテスト
ユーザー kakel-san
提出日時 2024-04-23 23:11:33
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 468 ms / 2,000 ms
コード長 1,769 bytes
コンパイル時間 11,353 ms
コンパイル使用メモリ 166,660 KB
実行使用メモリ 232,224 KB
最終ジャッジ日時 2024-11-06 05:31:50
合計ジャッジ時間 18,560 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 13
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /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/

ソースコード

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 t = NN;
        var ans = new bool[t];
        for (var u = 0; u < t; ++u)
        {
            var c = NList;
            var (n, k) = (c[0], c[1]);
            var map = NMap(n - 1);
            var a = NList;
            ans[u] = Rooted(n, k, map, a);
        }
        WriteLine(string.Join("\n", ans.Select(f => f ? "K" : "P")));
    }
    static bool Rooted(int n, int k, int[][] map, int[] a)
    {
        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 len = Enumerable.Repeat(int.MaxValue / 2, n).ToArray();
        len[0] = 0;
        var q = new Queue<int>();
        q.Enqueue(0);
        while (q.Count > 0)
        {
            var cur = q.Dequeue();
            foreach (var next in tree[cur])
            {
                if (len[next] <= len[cur] + 1) continue;
                len[next] = len[cur] + 1;
                q.Enqueue(next);
            }
        }
        var g = 0;
        for (var i = 0; i < n; ++i) if (len[i] % 2 == 1) g ^= a[i] % (k + 1);
        return g != 0;
    }
}
0