python 30 剩下全TLE求助
查看原帖
python 30 剩下全TLE求助
973945
ISAQ楼主2024/3/7 14:10
import math
n=int(input())
count=0
zidian1=[0]*5001
for i in range(n):
    zidian1[int(input())]+=1
for i in range(2500):
    i2=zidian1[i*2]
    i1=zidian1[i]
    if(zidian1[i]==0):
        continue
    count+=i2*(i2-1)/2*(i1*(i1-1))/2
    for j in range(i+1,5000-i):
        if(zidian1[i+j]==0 or zidian1[j]==0):
            continue
        ij=zidian1[i+j]
        count+=ij*(ij-1)/2*i1*zidian1[j]
print(int(count%1000000007))
2024/3/7 14:10
加载中...