LeeCode_1337. 矩阵中战斗力最弱的 K 行(纵向扫描)

最新推荐文章于 2021-12-01 14:29:59 发布

Netceor

最新推荐文章于 2021-12-01 14:29:59 发布

阅读量162

点赞数

分类专栏： LeeCode/PTA/C++ 文章标签： leecode c++

本文链接：https://blog.csdn.net/Netceor/article/details/119495052

版权

LeeCode/PTA/C++ 专栏收录该内容

85 篇文章 2 订阅

订阅专栏

该博客介绍了如何解决LeetCode上的一个问题——寻找矩阵中最弱的k行。博主提出了一种双指针的解决方案，通过遍历矩阵，优先选择军人数量较少的行，以确定战斗力最弱的行。在遇到0时，认为该行的战斗力减弱。如果k行不足，则将全为1的行按顺序补充。

摘要由CSDN通过智能技术生成

一、介绍

1.题目描述

题目链接：https://leetcode-cn.com/problems/the-k-weakest-rows-in-a-matrix/

给你一个大小为 m * n 的矩阵 mat，矩阵由若干军人和平民组成，分别用 1 和 0 表示。

请你返回矩阵中战斗力最弱的 k 行的索引，按从最弱到最强排序。

如果第 i 行的军人数量少于第 j 行，或者两行军人数量相同但 i 小于 j，那么我们认为第 i 行的战斗力比第 j 行弱。

军人总是排在一行中的靠前位置，也就是说 1 总是出现在 0 之前。

2.测试样例

[[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]]
3
# [2,0,3]

[[1,0,0,0],
 [1,1,1,1],
 [1,0,0,0],
 [1,0,0,0]]
2  # [0,2]

二、题解🟢（双指针）

由于1（军人）必定在0（平民）之前，所以一旦出现0，说明之后已经没有士兵了。

所谓每行战斗力即每行1的个数
战斗力弱即1少
答案要求用 vector 返回

思路：最先出现0的行战斗力最弱

class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        int m=mat.size();  // 有多少行
        int n=mat[0].size();   // 每行多少个人
        vector<int> ans;
        // flag 标记改行是否记录过，0表示未记录，1则跳过
        // t 记录当前选中多少行
        int flag[m],t=0;
        for(int i=0;i<m;i++) flag[i]=0;
        // 纵向遍历，每列第 j 个
        for(int j=0;j<n;j++){
            // 横向遍历，每行第 i 个
            for(int i=0;i<m;i++){
                // 如果未被标记过且遍历到数字0
                if(flag[i]==0&&mat[i][j]==0){
                    // 标记为1并加入答案
                    flag[i]=1;
                    ans.push_back(i);
                    t++;
                    // 当个数足够，结束循环
                    if(t==k) break;
                }
            }
            if(t==k) break;
        }
        int i=0;
        // 个数不够，说明必有些未标记的行全为1，按行数先后加入答案
        while(t<k){
            if(flag[i]==0){
                ans.push_back(i);
                t++;
            }
            i++;
        }
        return ans;
    }
};