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)