leetcode1424. 对角线遍历 II（排序）

给你一个列表 nums ，里面每一个元素都是一个整数列表。请你依照下面各图的规则，按顺序返回 nums 中对角线上的整数。
示例 1：
输入：nums = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,4,2,7,5,3,8,6,9]

代码

class Solution {
    public int[] findDiagonalOrder(List<List<Integer>> nums) {
        PriorityQueue<int[]> priorityQueue=new PriorityQueue<>((o1, o2) -> o1[0]+o1[1]==o2[0]+o2[1]?o2[0]-o1[0]:o1[0]+o1[1]-o2[0]-o2[1]);//排序规则，col+row相同的在对角线上
        for(int i=0;i<nums.size();i++)
            for(int j=0;j<nums.get(i).size();j++)//创建三元组入队
                priorityQueue.add(new int[]{i,j,nums.get(i).get(j)});
            int[] t=new int[priorityQueue.size()];
           for (int i=0;i<t.length;i++)//结果出队
                t[i]=priorityQueue.poll()[2];
           return t;
    }
}