結果

問題 No.1328 alligachi-problem
ユーザー titiatitia
提出日時 2020-12-25 03:07:31
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,244 ms / 2,000 ms
コード長 1,528 bytes
コンパイル時間 271 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 80,652 KB
最終ジャッジ日時 2024-09-21 17:35:42
合計ジャッジ時間 26,742 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

from operator import itemgetter

N=int(input())
S=[input().split() for i in range(N)]

for i in range(N):
    S[i][2]=int(S[i][2])

SB=[]
SR=[]

for i in range(N):
    if S[i][1]=="B":
        SB.append(S[i]+[i+1])
    else:
        SR.append(S[i]+[i+1])

SB.sort(key=itemgetter(0),reverse=True)
SB.sort(key=itemgetter(2))

SR.sort(key=itemgetter(0))
SR.sort(key=itemgetter(2))

#print(SB)
#print(SR)

SB.reverse()
SR.reverse()

B=0
R=0

ANS=[]

while SB or SR:
    if SB and SR:
        if SB[-1][2]==B and SB[-1][0]=="B":
            x=SB.pop()
            ANS.append(x[3])
            B+=1
            continue

        if SR[-1][2]==R and SR[-1][0]=="R":
            x=SR.pop()
            ANS.append(x[3])
            R+=1
            continue

        if SB[-1][2]==B and SB[-1][0]=="R" and SR[-1][2]!=R:
            x=SB.pop()
            ANS.append(x[3])
            R+=1
            continue

        if SR[-1][2]==R and SR[-1][0]=="B" and SB[-1][2]!=B:
            x=SR.pop()
            ANS.append(x[3])
            B+=1
            continue

    else:
        if SB and SB[-1][2]==B:
            x=SB.pop()
            ANS.append(x[3])
            if x[0]=="B":
                B+=1
            else:
                R+=1
            continue

        if SR and SR[-1][2]==R:
            x=SR.pop()
            ANS.append(x[3])
            if x[0]=="B":
                B+=1
            else:
                R+=1
            continue

    print("No")
    break

else:
    print("Yes")
    print(*ANS)
0