给你一个列表 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;}
}