Snapchat 面经

看到钟神，过来点个赞

写了下第一道题的python, 如有bug请指正。

1. def zigzag(matrix):
   
2.         # edge case
   
3.         if matrix == None or len(matrix) == 0 or len(matrix[0]) == 0:
   
4.                 return 0
   
5. 
   
6.         m, n = len(matrix), len(matrix[0])
   
7.         res = []
   
8. 
   
9.         for k in range(m + n - 1):
   
10.                 temp = []
    
11.                 if k <= n - 1:
    
12.                         i = 0
    
13.                         j = k
    
14.                 else:
    
15.                         j = n - 1
    
16.                         i = k - j
    
17. 
    
18.                 while i < m and j >= 0:
    
19.                         temp.append(matrix[i][j])
    
20.                         i += 1
    
21.                         j -= 1
    
22. 
    
23.                 res.append(temp)
    
24. 
    
25.         return res

复制代码

补充内容 (2015-12-15 00:45):
抱歉发错楼了。。。

好像一般的语言里的TREEMAP都是用红黑树实现的，自己实现必须要用红黑树还是BST就可以