給你一個列表 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;}
}