Löse den AtCoder-Anfängerwettbewerb159 D - Verbotenes K (ABC159D) mit Python (Anzahl ist zu langsam!)

import sys
from collections import Counter
N=int(sys.stdin.readline())
L=list(map(int,sys.stdin.readline().split()))
C=Counter(L)           #In L enthalten{Brief:Anzahl der Auftritte}Ich werde mit dem Wörterbuchtyp zurückkommen
x=0
for a in C.values():   #Extrahieren Sie nur den Wert (Anzahl der Vorkommen)
  x += int(a*(a-1)/2)  
for K in range(N):
  print(x-C[L[K]]+1)   #Subtrahieren Sie die Kombination, die K enthält, von der gleichen Anzahl von Kombinationen

import sys
N=int(sys.stdin.readline())
L=list(map(int,sys.stdin.readline().split()))
C=list(set(L))
x=0
for i in range(len(C)):
  a = L.count(C[i])
  x += int(a*(a-1)/2)  
for K in range(N):
  print(x-L.count(L[K])+1)

Recommended Posts