LeetCode_1329. Sort the Matrix Diagonally

這個有點難..

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