結果

問題 No.2148 ひとりUNO
ユーザー H3PO4H3PO4
提出日時 2022-12-20 21:14:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 113 ms / 2,000 ms
コード長 1,338 bytes
コンパイル時間 372 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 77,320 KB
最終ジャッジ日時 2024-11-18 02:08:21
合計ジャッジ時間 5,454 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys

input = sys.stdin.readline


def color2int(c: str):
    if c == "R":
        return 1
    elif c == "G":
        return 2
    elif c == "B":
        return 4
    raise ValueError


def solve(N, cards):
    lst = [0] * N
    used_color = 0
    for c, d in cards:
        i = color2int(c)
        lst[d] += i
        used_color |= i
    used_color_num = bin(used_color).count("1")
    ct = [0] * 8
    for x in lst:
        ct[x] += 1

    if used_color_num == 1:
        return True
    elif used_color_num == 2:
        return ct[1 | 2] or ct[1 | 4] or ct[2 | 4]
    else:
        return ((ct[1 | 2] and ct[1 | 4])
                or (ct[1 | 2] and ct[2 | 4])
                or (ct[1 | 4] and ct[2 | 4])
                or (ct[1 | 2] and ct[1 | 2 | 4])
                or (ct[1 | 4] and ct[1 | 2 | 4])
                or (ct[2 | 4] and ct[1 | 2 | 4])
                or ct[1 | 2 | 4] >= 2
                or (ct[1] + ct[1 | 2] + ct[1 | 4] == 0 and ct[1 | 2 | 4])
                or (ct[2] + ct[2 | 4] + ct[2 | 1] == 0 and ct[1 | 2 | 4])
                or (ct[4] + ct[4 | 1] + ct[4 | 2] == 0 and ct[1 | 2 | 4]))


T = int(input())
for _ in range(T):
    N = int(input())
    cards = []
    for __ in range(N):
        c, d = input().split()
        cards.append((c, int(d) - 1))
    print("YES" if solve(N, cards) else "NO")
0