fork download
  1. a =[1,2,3,4,5,6]
  2. n = len(a)
  3. c0 = 0
  4. c1 = 0
  5. c2 = 0
  6. for i in range(n):
  7. if a[i] % 3 == 0 :
  8. c0 +=1
  9. elif a[i] % 3 == 1:
  10. c1 +=1
  11. else:
  12. c2 +=1
  13. print(c0*(c0 -1 )// 2 + c1 * c2)
Success #stdin #stdout 0.11s 14160KB
stdin
5
3 4 2 3 4 
stdout
5