fork(1) download
  1. #!/usr/bin/env python
  2.  
  3. # read dominoes
  4. N = 7
  5. dominoes = [[0] * N for _ in range(N)]
  6. for _ in range(int(input())):
  7. i, j = map(int, input().split())
  8. dominoes[j][i] = dominoes[i][j] = dominoes[i][j] + 1
  9.  
  10. # count "friendly" pairs
  11. npairs = 0
  12. for i in range(N): # for each domino i:j where i <= j
  13. for j in range(i, N):
  14. n = dominoes[i][j]
  15. if n:
  16. # {n \choose 2} pairs with itself
  17. npairs += n * (n - 1) // 2
  18. for k in range(i+1, N): # top-down
  19. npairs += dominoes[k][j] * n
  20. if i != j: # don't count twice
  21. for k in range(j+1, N): # left-right
  22. npairs += dominoes[i][k] * n
  23. print(npairs)
Success #stdin #stdout 0.04s 9528KB
stdin
5
1 2
2 3
3 1
4 3
4 3
stdout
8