Editing 1357 - Plus Minus (section)

==Rezolvare==
<syntaxhighlight lang="python3" line="1">
def solve_plus_minus(n, signs, current_sum, current_expression, result):

   if len(current_expression) == n:
       if current_sum == 0:
           result.append(" ".join(current_expression))
       return
   for sign in signs:
       new_sum = current_sum + sign * (current_expression[-1] + 1) ** 2
       if new_sum <= n:
           solve_plus_minus(n, signs, new_sum, current_expression + [sign], result)

def plus_minus(n):

   signs = [-1, 1]  # -1 pentru minus, 1 pentru plus
   result = []
   # Inițial, începem cu primul termen, care este întotdeauna un plus
   solve_plus_minus(n, signs, 1, [1], result)
   if not result:
       result.append("IMPOSIBIL")
   return result
Citim input-ul

with open("plusminus.in", "r") as f:

   n = int(f.readline().strip())
Calculăm rezultatul

result = plus_minus(n)

Scriem rezultatul în fișierul de ieșire

with open("plusminus.out", "w") as f:

   for line in result:
       f.write(line + "\n")
</syntaxhighlight>