python参考

ins=[]

while True:

try:

ins.append(list(input().split()))

except:

break

n=int(ins[0][0])

a=[int(val) for val in ins[1]]

col=ins[2][0]

#print (col)

val=dict()

for i in range(n):

if a[i] not in val:

val[a[i]] = []

val[a[i]].append((i,col[i]))

#print (val)

#print (len(val))

#for i in range(len(val)):

sumv=0

for key in val:

#print (key)

nbr=[0,0]

for vali in val[key]:

#print (vali)

if vali[1] == 'B':

nbr[0] += 1

else:

nbr[1] += 1

sumv += nbr[0]*nbr[1]

print(sumv)