[E二分] lc1337. 矩阵中战斗力最弱的 K 行(模拟+STL+二分优化)

Posted 2021-08-30 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：1337. 矩阵中战斗力最弱的 K 行

2. 题目解析

建军节纪念。军人守护在人民前面！致敬。

我是标题党…

• 暴力解法，统计每一行中军人总数和行号，放到 pair 中，直接排序即可。

优化：

• 统计每行的话，可以二分来统计，找到最后一个 1 的位置即可，就能知道 1 的个数了。 但在此就懒得写了，直接暴力做了。

---

• 时间复杂度：$O(nm)$
• 空间复杂度：$O(n)$

---

代码：

pair 存就行了。

class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        vector<pair<int, int>> q;
        for (int i = 0; i < mat.size(); i ++ ) {
            int s = 0;
            for (int j = 0; j < mat[0].size(); j ++ ) s += mat[i][j];
            q.push_back({s, i});
        }
        sort(q.begin(), q.end());
        vector<int> res;
        for (int i = 0; i < k; i ++ ) res.push_back(q[i].second);

        return res;
    }
};