Submission #978927

---

Source Code Expand

n = 3
p = [2, 1, 0]

n = input()
p = map(lambda x: x - 1, map(eval, raw_input().strip().split()))

a = []
b = []
for i in xrange(n):
  x = (i + 1) * 2
  a += [x]
  y = (n - i) * 2
  b += [y]

for i, x in enumerate(p):
  if a[x] + b[x] <= a[p[i-1]] + b[p[i-1]]:
    a[x] += a[p[i-1]] + b[p[i-1]] - a[x] + b[x] + 1
  print a[x] + b[x]

print a, b

Submission Info

Judge Result