美团点评2016研发工程师编程题(二) python

[编程题]二维数组打印

有一个二维数组(n*n),写程序实现从右上角到左下角沿主对角线方向打印。
给定一个二位数组arr及题目中的参数n，请返回结果数组。
测试样例：
[[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]],4
返回：[4,3,8,2,7,12,1,6,11,16,5,10,15,9,14,13]
思路先打印对角线上部分和对角线.每当打印到最后一列,令行为0,列为上次循环开始行的值减一,并且开始行减一.行小于0时停止迭代
在打印对角线下部分,思路一致

def fun(arr,n):
    if n==0:
        return []
    j=j1=n-1;que=[arr[0][n-1]]
    while j>0:
        i=0
        if j1==n-1:
            j1=j-1;j-=1
        while j1 <n:
            que+=[arr[i][j1]]
            i+=1;j1+=1
        j1-=1
    i=i1=1
    while i<n:
        j=0
        while i1<n:
            que+=[arr[i1][j]]
            i1+=1;j+=1
        i1-=1
        if i1==n-1:
            i1=i+1;i+=1
    return que

[编程题]股票交易日

在股市的交易日中，假设最多可进行两次买卖(即买和卖的次数均小于等于2)，规则是必须一笔成交后进行另一笔(即买-卖-买-卖的顺序进行)。给出一天中的股票变化序列，请写一个程序计算一天可以获得的最大收益。请采用实践复杂度低的方法实现。
给定价格序列prices及它的长度n，请返回最大收益。保证长度小于等于500。
测试样例：
[10,22,5,75,65,80],6
返回：87
**思路:**以第i天为分界线，计算第i天之前进行一次交易的最大收益，和第i天之后进行一次交易的最大收益，最后遍历i取值

class Stock:
    def fun(self,arr): # 一次交易最大利润函数
        if arr==[]:
            return 0
        maxs=0
        mins=arr[0]
        for i in range(1,len(arr)):
            mins=min(mins,arr[i])
            maxs=max(maxs,arr[i]-mins)
        return maxs
    def maxProfit(self, prices, n):
        maxs = 0
        for i in range(n-1):
            temp=self.fun(prices[:i])+self.fun(prices[i:])
            maxs=max(maxs,temp)
        return maxs