年终奖:[走二维数组求最大情况]
题目描述:小东所在公司要发年终奖,而小东恰好获得了最高福利,他要在公司年会上参与一个抽奖游戏,游戏在一个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拥有如此精巧的用法,让人不得不感叹:人生苦短