#26432: python分享(1.6s)求更快解


andy.lin61013@gmail.com (我是你爸)

學校 : 國立清華大學
編號 : 158445
來源 : [118.166.40.151]
最後登入時間 :
2022-08-02 21:19:43
b557. 直角三角形 | From: [36.231.61.64] | 發表日期 : 2021-08-06 19:03

from itertools import combinations
from collections import Counter
from sys import stdin
s=stdin.readline()
for i in range(int(s)):
    input()
    list1=[int(x) for x in input().split()]
    a=dict(Counter(list1))
    ans=sum(a[smal]*a[mid]*a[larg] for smal,mid,larg in combinations(sorted(a),3)
    if smal**2+mid**2==larg**2)
    print(ans)
 
ZeroJudge Forum