fork download
  1. r = [3, 2, 2]
  2. rl = range(len(r))
  3. ni = [sum(r[j] + 1 for j in rl if j != i) for i in rl]
  4. ki = [n*(n - 1) for n in ni]
  5. k = sum(ki)
  6. print(k)
  7.  
Success #stdin #stdout 0.01s 7728KB
stdin
Standard input is empty
stdout
114