leetcode 1337. 矩阵中战斗力最弱的 K 行(java)

最新推荐文章于 2024-06-15 09:23:13 发布

姐姐的半熟芝士

最新推荐文章于 2024-06-15 09:23:13 发布

阅读量81

点赞数

分类专栏：学习做题leetcode 文章标签： leetcode 矩阵算法

本文链接：https://blog.csdn.net/qq_45908682/article/details/125121502

版权

学习做题leetcode 专栏收录该内容

341 篇文章 1 订阅

订阅专栏

简单题重拳出击！！！

直接暴力怎么想怎么做

遍历 + 排序

class Solution {
    public int[] kWeakestRows(int[][] mat, int k) {
        int m = mat.length;
        int n = mat[0].length;
        int[][] tmp = new int[m][2];
        for(int i = 0 ; i < m; i ++){
            int cur = 0;
            for(int j = 0 ; j < n ; j ++){
                if(mat[i][j] == 1){
                    cur ++;
                }
            }
            tmp[i] = new int[]{cur, i};
        }
        Arrays.sort(tmp, (o1, o2) -> {
            if(o1[0] != o2[0]) return o1[0] - o2[0];
            return o1[1] - o2[1];
        });
        int[] ans = new int[k];
        for(int i = 0 ; i < k; i ++){
            ans[i] = tmp[i][1];
        }
        return ans;
    }
}