ABC236 B - Who is missing?

atcoder.jp

計算量の問題で後者のコードだとTLEになってしまい、
ACの解答にたどり着くまで少し苦労しました

ACな解答

n = int(input())
a = list(map(int,input().split()))
gokei = 0
for i in range(1,n+1):
    gokei += (i*4)
print(gokei-sum(a))


TLEな解答

n = int(input())
a = list(map(int,input().split()))
for i in range(1,n+1):
    if a.count(i) != 4:
        print(i)

おそらくfor文の中でcountメソッドを使っているのが原因でしょう
ループの度に毎回数えさせてたら、そりゃ時間かかりますよね