結果

問題 No.1328 alligachi-problem
ユーザー tamato
提出日時 2020-12-25 17:48:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 581 ms / 2,000 ms
コード長 1,907 bytes
コンパイル時間 410 ms
コンパイル使用メモリ 82,228 KB
実行使用メモリ 135,940 KB
最終ジャッジ日時 2024-09-22 10:51:17
合計ジャッジ時間 15,577 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())
    reqR = []
    reqB = []
    for i in range(1, N+1):
        C, X, Y = input().split()
        Y = int(Y)
        if X == "R":
            reqR.append((C, Y, i))
        else:
            reqB.append((C, Y, i))
    reqR.sort(key=lambda x: x[0])
    reqR.sort(key=lambda x: x[1])
    reqB.sort(key=lambda x: x[0], reverse=True)
    reqB.sort(key=lambda x: x[1])
    reqR.reverse()
    reqB.reverse()

    ans = []
    r = b = 0
    for _ in range(N):
        ok_r = 0
        ok_b = 0
        if reqR:
            if reqR[-1][1] == r:
                ok_r = 1
        if reqB:
            if reqB[-1][1] == b:
                ok_b = 1
        if ok_r == ok_b == 0:
            print("No")
            exit()
        if ok_r == 1 and ok_b == 0:
            c, _, i = reqR.pop()
            ans.append(i)
            if c == "R":
                r += 1
            else:
                b += 1
        elif ok_r == 0 and ok_b == 1:
            c, _, i = reqB.pop()
            ans.append(i)
            if c == "R":
                r += 1
            else:
                b += 1
        else:
            flg_r = flg_b = 0
            if reqR[-1][0] == "B":
                flg_r = 1
            if reqB[-1][0] == "R":
                flg_b = 1
            if flg_r == flg_b == 1:
                print("No")
                exit()
            if flg_r == 1:
                c, _, i = reqB.pop()
                ans.append(i)
                if c == "R":
                    r += 1
                else:
                    b += 1
            else:
                c, _, i = reqR.pop()
                ans.append(i)
                if c == "R":
                    r += 1
                else:
                    b += 1
    print("Yes")
    print(*ans)


if __name__ == '__main__':
    main()
0