年终奖:[走二维数组求最大情况]

题目描述:小东所在公司要发年终奖,而小东恰好获得了最高福利,他要在公司年会上参与一个抽奖游戏,游戏在一个6*6的棋盘上进行,上面放着36个价值不等的礼物,每个小的棋盘上面放置着一个礼物,他需要从左上角开始游戏,每次只能向下或者向右移动一步,到达右下角停止,一路上的格子里的礼物小东都能拿到,请设计一个算法使小东拿到价值最高的礼物。

给定一个6*6的矩阵board,其中每个元素为对应格子的礼物价值,左上角为[0,0],请返回能获得的最大价值,保证每个礼物价值大于100小于1000

 

花里胡哨,其实就是遍历二维数组(笑)


class Bonus:
    def getMost(self, board):
        # write code here
        global maxScore
        maxScore = board[0][0]
        Bonus.turnWay(self, board, board[0][0])
        return maxScore

    def turnWay(self, board, sumScore):
        global maxScore
        if len(board) + len(board[0]) <= 2:
            maxScore = max(maxScore, sumScore)
        if len(board[0]) > 1:
            arr = map(lambda x: x[1:], board)
            Bonus.turnWay(self, arr, sumScore + arr[0][0])
        if len(board) > 1:
            arr = board[1:]
            Bonus.turnWay(self, arr, sumScore + arr[0][0])
bonus = Bonus()
test_arr = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
print  bonus.getMost(test_arr)

代码的思路非常简单,主要收获是在Python处理二维数组时的便捷性。

arr = map(lambda x: x[1:], board)

之前已经了解到map函数的作用是对arg2每个元素执行arg1表示的方法,这里的lambda(其实就是λ)是一个匿名函数,执行的功能是对参数x执行后面的方法,这里先把board二维数组的每一行作为变量x传到arg1中,然后对x也就是board每一行执行board[i][1:]操作。

这里使用for…in…是同理的:

arr = [boardrow[1:] for boardrow in board]

Python拥有如此精巧的用法,让人不得不感叹:人生苦短


posted @ 2017-04-05 22:47  kuqs(奇小东)  阅读(246)  评论(0)    收藏  举报