Summat

#!/usr/bin/python

from itertools import product, combinations
import sys

juttu1 = [[4, 4, 6, 4, 6, 6], [4, 5, 7, 5, 7, 8], [4, 6, 4, 6, 4, 6], [4, 7, 5, 7, 5, 8], [5, 4, 7, 5, 8, 7], [5, 7, 4, 8, 5, 7], [6, 4, 4, 6, 6, 4], [7, 4, 5, 7, 8, 5], [7, 5, 4, 8, 7, 5], [5, 5, 8, 4, 7, 7], [5, 8, 5, 7, 4, 7], [6, 6, 8, 6, 8, 8], [6, 8, 6, 8, 6, 8], [8, 5, 5, 7, 7, 4], [8, 6, 6, 8, 8, 6], [6, 6, 6, 4, 4, 4], [7, 7, 8, 4, 5, 5], [7, 8, 7, 5, 4, 5], [8, 7, 7, 5, 5, 4], [8, 8, 8, 6, 6, 6]]
juttu2 = [[2, 2, 2, 4], [2, 2, 3, 5], [2, 2, 4, 2], [2, 2, 5, 3], [2, 3, 2, 5], [2, 3, 5, 2], [2, 4, 2, 2], [2, 5, 2, 3], [2, 5, 3, 2], [3, 2, 2, 5], [3, 2, 5, 2], [3, 3, 3, 5], [3, 3, 5, 3], [3, 5, 2, 2], [3, 5, 3, 3], [4, 2, 2, 2], [5, 2, 2, 3], [5, 2, 3, 2], [5, 3, 2, 2], [5, 3, 3, 3]]

n = int(raw_input())
paska = [int(x) for x in raw_input().split(" ")]
lista = sorted(paska)

if n < 3 or n > 5:
	raise Exception('asdf')

for x in product([1,2,3,4,5], repeat=n):
	candidate = [a+b for a,b in combinations(x, 2)]
	if sorted(candidate) == lista:
		for asd, sad in enumerate(juttu1):
			if sad == paska:
				x = juttu2[asd]
		print ' '.join([str(y) for y in x])
		sys.exit()
print 'asdfadsfasdf'
raise Exception('asdfasdf')

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

Test 5

Group: 1, 2, 3

Verdict: ACCEPTED

Test 6

Group: 2, 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 7

Group: 2, 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 8

Group: 2, 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 9

Group: 2, 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 10

Group: 2, 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 11

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 12

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 13

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 14

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf

Test 15

Group: 3

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 14, in <module>
    raise Exception('asdf')
Exception: asdf