腾讯笔试 密室探险

题目描述：小Q在假期的时候去一个密室探险，这个密室可以看做一个n*3的格子，小Q最开始可以选择第一行的任意一个格子进入，此后的每一步，他可以选择下一行的当前位置或左侧一格或右侧一格。简单地说，假设小Q的当前位置为(x,y)，那么他下一步可以前进至(x+1,y-1)，(x+1,y)或(x+1,y+1)，但他不能走出格子。每个格子都有一个数字，代表走到该格子可以获得的分数。但密室中有一些魔法格子，它们的数字为0，当小Q经过魔法格子后。他之后获得的分数就会变成相反数，分数可多次翻转，请问小Q到达密室最后一行时，最多可以获得多少分？
1<=n<=100000
-100000<=m<=100000
示例：
输入：
6
1 2 3
8 9 10
5 0 5
-9 -8 -10
0 1 2
5 4 6
输出：
27
当时看错题了，以为遇到0，之前的分数变成相反数，这道题还是递归，遇到0后变成求最小。

import sys
def solution(arr, x, y, flag):
    num = 0
    if y < 0 or y > 2 or x >= len(arr):
        return 0
    if arr[x][y] == 0:
        flag = - flag
    else:
        num += flag * arr[x][y]
    if flag:
        num += max(solution(arr, x + 1, y - 1, flag), solution(arr, x + 1, y, flag), solution(arr, x + 1, y + 1, flag))
    else:
        num += min(solution(arr, x + 1, y - 1, flag), solution(arr, x + 1, y, flag), solution(arr, x + 1, y + 1, flag))
    return num
loop = int(sys.stdin.readline().strip())
arr = []
for i in range(loop):    
    n = [int(x) for x in sys.stdin.readline().strip().split()]
    arr.append(n)
flag = 1
num = max(solution(arr, 0, 0, flag), solution(arr, 0, 1, flag), solution(arr, 0, 2, flag))
print(num)