之字形打印矩阵

题目描述

对于一个矩阵,请设计一个算法,将元素按“之”字形打印。具体见样例。

给定一个整数矩阵mat,以及他的维数nxm,请返回一个数组,其中元素依次为打印的数字。

测试样例:
[[1,2,3],[4,5,6],[7,8,9],[10,11,12]],4,3
返回:[1,2,3,6,5,4,7,8,9,12,11,10]

简单题
# -*- coding:utf-8 -*-

class Printer:
    def printMatrix(self, mat, n, m):
        # write code here
        res = []
        flag = False
        for x in mat:
        	if flag == False:
        		res.extend(x)
        		flag = True
        	else:
        		res.extend(x[::-1])
        		flag = False

        return res

if __name__ == "__main__":
	a = Printer()
	print a.printMatrix([[1,2,3],[4,5,6],[7,8,9],[10,11,12]],4,3)


你可能感兴趣的:(ACM-2016校招真题)