这个有点难..
from collections import defaultdict
m=len(mat[0])
n=len(mat)
d=collections.defaultdict(list)
for i in range(n):
for j in range(m):
d[i-j].append(mat[i][j])
for k in d:
d[k].sort(reverse=1)
for i in range(n):
for j in range(m):
mat[i][j]=d[i-j].pop()
return mat
网友评论