想讨论下这题lc 1424

我用java写了个排序版本的代码，但是不通过，不知道bug为何。。想请求下大佬帮助

题目链接 https://leetcode.com/problems/diagonal-traverse-ii/题目链接

1. class Solution {
   
2.     public int[] findDiagonalOrder(List<List<Integer>> nums) {
   
3.         class comp implements Comparator<List<Integer>>{
   
4.             public int compare(List<Integer> a, List<Integer> b){
   
5.                 return a.get(1) == b.get(1) ? b.get(2).compareTo(a.get(2)):a.get(1).compareTo(b.get(1));
   
6.             }
   
7.         }
   
8.         
   
9.         List<List<Integer>> list = new ArrayList<>();
   
10.         for (int i = 0; i < nums.size(); i++){
    
11.             for (int j = 0; j < nums.get(i).size(); j++){
    
12.                 List<Integer> element = new ArrayList<>();
    
13.                 element.add(nums.get(i).get(j));
    
14.                 element.add(i+j);
    
15.                 element.add(i);
    
16.                 list.add(element);
    
17.             }
    
18.         }
    
19.         Collections.sort(list, new comp());
    
20.         int[] res = new int[list.size()];
    
21.         for (int i = 0; i < res.length; i++){
    
22.             res[i] = list.get(i).get(0);
    
23.         }
    
24.         return res;
    
25.     }
    
26. }

复制代码

[/i]

我来帮下楼主吧，你代码第5行应该改成return a.get(1).equals(b.get(1)) ? b.get(2).compareTo(a.get(2)):a.get(1).compareTo(b.get(1));

之所以你会通过大部分的test case是因为Java对于-128到127的Integer有缓存机制，所以在数据范围比较小的情况下你用a.get(1) == b.get(1)是可以比较两个Integer值是否相等的。

bitzyq 发表于 2020-7-29 13:55
我来帮下楼主吧，你代码第5行应该改成return a.get(1).equals(b.get(1)) ? b.get(2).compareTo(a.get(2)):a ...

Oh 差点忘了这茬了。。。谢谢帮助！！