結果

問題 No.1328 alligachi-problem
ユーザー persimmon_t01
提出日時 2021-01-08 21:21:01
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 2,253 bytes
コンパイル時間 262 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 521,472 KB
最終ジャッジ日時 2024-11-16 09:47:59
合計ジャッジ時間 49,363 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 6 TLE * 18 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main(n,cxy):
  # xで分類
  bary=[[set(),set()] for _ in range(n+1)]
  rary=[[set(),set()] for _ in range(n+1)]
  dc={'B':0,'R':1}
  B=0
  R=1
  for i,(c,x,y) in enumerate(cxy):
    if x=='B':
      bary[int(y)][dc[c]].add(i+1)
    else:
      rary[int(y)][dc[c]].add(i+1)
  # bart[i][B]:青がi個の状態で列に加えないといけない青ボール。
  # bart[i][R]:青がi個の状態で列に加えないといけない赤ボール。
  # rart[i][B]:赤がi個の状態で列に加えないといけない青ボール。
  # rart[i][R]:赤がi個の状態で列に加えないといけない赤ボール。
  numb=0
  numr=0
  ret=[]
  while len(ret)<n:
    ary=[len(bary[numb][B])>0,len(rary[numr][B])>0,len(bary[numb][R])>0,len(rary[numr][R])>0]
    num=0
    for i in range(4):
      if ary[i]:num+=1<<i
    if ary[1] and ary[2]:
      return [-1]
    elif num==0:
      return [-1]
    elif num==1:
      ret.append(bary[numb][B].pop())
      numb+=1
    elif num==2:
      ret.append(rary[numr][B].pop())
      numb+=1
    elif num==4:
      ret.append(bary[numb][R].pop())
      numr+=1
    elif num==8:
      ret.append(rary[numr][R].pop())
      numr+=1
    elif num==3:
      ret.append(bary[numb][B].pop())
      numb+=1
    elif num==5:
      ret.append(bary[numb][R].pop())
      numr+=1
    elif num==6:pass
    elif num==7:pass
    elif num==9:
      ret.append(bary[numb][B].pop())
      numb+=1
      ret.append(rary[numr][R].pop())
      numr+=1
    elif num==10:
      ret.append(rary[numr][B].pop())
      numb+=1
    elif num==11:
      ret.append(bary[numb][B].pop())
      numb+=1
    elif num==12:
      ret.append(rary[numr][R].pop())
      numr+=1
    elif num==13:
      ret.append(rary[numr][R].pop())
      numr+=1
    elif num==14:pass
    elif num==15:pass
  return ret

def chk(n,cxy,ret):
  numb=0
  numr=0
  for i in ret:
    i-=1
    c,x,y=cxy[i]
    y=int(y)
    if x=='R':
      if numr!=y:return False
    elif x=='B':
      if numb!=y:return False
    if c=='R':
      numr+=1
    else:
      numb+=1
  return True

if __name__ == "__main__":
  n=int(input())
  cxy=[list(input().split()) for _ in range(n)]
  ret=main(n,cxy)
  if ret[0]==-1:
    print('No')
  else:
    print('Yes')
    print(*ret)
0