ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LeetCode 1337. 矩阵中战斗力最弱的 K 行

2021-08-01 21:34:09  阅读:260  来源: 互联网

标签:map mat int 1337 list 最弱 排序 LeetCode


一、题目

1. 描述

给你一个大小为 m × n m \times n m×n 的矩阵 mat,矩阵由若干军人和平民组成,分别用 1 1 1 和 0 0 0 表示。请你返回矩阵中战斗力最弱的 k k k 行的索引,按从最弱到最强排序。如果第 i i i 行的军人数量少于第 j j j 行,或者两行军人数量相同但 i i i 小于 j j j,那么我们认为第 i 行的战斗力比第 j 行弱。军人总是排在一行中的靠前位置,也就是说 1 1 1 总是出现在 0 0 0 之前。

2. 示例

  • 示例一
输入:mat = 
[[1,1,0,0,0],
 [1,1,1,1,0],
 [1,0,0,0,0],
 [1,1,0,0,0],
 [1,1,1,1,1]], 
k = 3
输出:[2,0,3]
解释:
每行中的军人数目:
 0 -> 2 
 1 -> 4 
 2 -> 1 
 3 -> 2 
 4 -> 5 
从最弱到最强对这些行排序后得到 [2,0,3,1,4]
  • 示例二
输入:mat = 
[[1,0,0,0],
 [1,1,1,1],
 [1,0,0,0],
 [1,0,0,0]], 
k = 2
输出:[0,2]
解释: 
每行中的军人数目:
 0 -> 1 
 1 -> 4 
 2 -> 1 
 3 -> 1 
从最弱到最强对这些行排序后得到 [0,2,3,1]

二、解题

1. 思路

  • 需要每行的能量的和:求和函数
  • 将每行的能量和进行比较(排序),同时需要记录这些能量和的下标:用哈希表实现,key(行数),value(行的能量和)。
  • 难点:哈希表按照值排序。

2.代码

class Solution {
    public int[] kWeakestRows(int[][] mat, int k) {
        Map<Integer,Integer> map = new HashMap<Integer, Integer>();
        int n = mat.length; 
        int i =0;
        
        for(int[] ma:mat){
            map.put(i++,getSum(ma));
        }
       
       //HashMap按照值排序
       //创建HashMap的键list,根据值将list中的键排序,用排好序的键去搜索HashMap。
        List<Integer> list = new ArrayList<Integer>(map.keySet());
        Collections.sort(list, (a,b) -> map.get(a)-map.get(b));
        
        System.out.print(list.size());

        int[] ans = new int[k];
        for(int j=0;j<k;j++){
            ans[j] = list.get(j);
        }
     
        return ans;
    }

    public  int getSum(int[] nums){
        int sum=0;
        for(int num: nums){
            sum+=num;
        }
        return sum;
    }
}

结果

在这里插入图片描述

标签:map,mat,int,1337,list,最弱,排序,LeetCode
来源: https://blog.csdn.net/wodexinhaoleng1/article/details/119303729

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有