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

最新推荐文章于 2025-10-27 21:31:52 发布

皮卡猴

最新推荐文章于 2025-10-27 21:31:52 发布

阅读量230

点赞数

分类专栏： leetcode刷题文章标签： leetcode 矩阵算法

leetcode刷题专栏收录该内容

252 篇文章

订阅专栏

该博客介绍了如何解决LeetCode上的第1337题，即找出矩阵中战斗力最弱的K行。通过计算每行士兵的战斗力并进行排序，找出战斗力最小的K行。主要涉及算法和数据结构的应用，具体实现使用了Go语言。

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

1. 问题描述

在这里插入图片描述

2. 思路

计算每一行士兵的战斗力，存入power
对power排序

3. 代码

type Power struct {
    Value int
    Row int
}

func kWeakestRows(mat [][]int, k int) []int {
    var res []int
    var powers []Power
    for i := 0; i < len(mat); i++ {
        power := Power {
            Value : countPower(mat[i]),
            Row : i,
        }
        // fmt.Printf("i == %v, power == %v, row == %v\n", i, power.Value, power.Row)
        powers = append(powers, power)
    }

    sort.SliceStable(powers, func(i, j int) bool {
        return powers[i].Value < powers[j].Value
    })
    for i := 0; i < k; i++ {
        res = append(res, powers[i].Row)
    }
    return res
}

func countPower(arr []int) int {
    i := 0
    for i < len(arr) && arr[i] == 1 {
        i++
    }
    return i
}