結果

問題 No.1523 +/- Tree
ユーザー lam6er
提出日時 2025-03-20 20:30:39
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,030 bytes
コンパイル時間 199 ms
コンパイル使用メモリ 82,404 KB
実行使用メモリ 70,040 KB
最終ジャッジ日時 2025-03-20 20:31:27
合計ジャッジ時間 5,981 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 26 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

n, k = map(int, input().split())

if k == 1 or k >= n - 1:
    print("No")
else:
    if n - k != 2:
        print("No")
    else:
        print("Yes")
        # Construct a chain of edges
        edges = []
        edges.append((1, 2, 2))
        if k > 2:
            for i in range(3, k + 1):
                edges.append((i-1, i, 0))
        edges.append((k + 1 - 1, k + 1, -3))  # The second edge is -3
        # Add the remaining edges as zeros
        for i in range(k + 2, n):
            edges.append((i-1, i, 0))
        # The last edge should be 2 if there are remaining edges beyond the first K
        if k + 1 < n:
            edges.append((n-1 - (n - (k + 1)), n, 2))
        # Correction for the actual chain
        edges = []
        edges.append((1, 2, 2))
        edges.append((2, 3, -3))
        if k > 2:
            for i in range(3, k + 1):
                edges.append((i, i + 1, 0))
        if n > k + 1:
            edges.append((k + 1, k + 2, 2))
        for u, v, w in edges:
            print(u, v, w)
0