方阵中战斗力最弱的 K 行 - 熟悉 Multimap 的使用

方阵中战斗力最弱的 K 行 - 熟悉 Multimap 的使用

题目链接：CLICK ME

给你一个大小为 m * n 的方阵 mat，方阵由若干军人和平民组成，分别用 0 和 1 表示。
请你返回方阵中战斗力最弱的 k 行的索引，按从最弱到最强排序。

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

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

Input:
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

Output:
[2,0,3]

解题思路：
首先遍历 mat，确定每一行中的军人数量，形成 军人数量 - 索引号 结构，利用 BST 中 key 有序的特性考虑使用 STL 中的 map，考虑到存在战斗力相同但索引号不同的情况，应使用 multimap，最后通过迭代器输出 multimap 中的前 k 个数据的索引号即可。

class Solution {
public:
    vector<int> kWeakestRows(vector<vector<int>>& mat, int k) {
        std::vector<int> res;
        int rows = mat.size(), cols = mat[0].size();
        if (rows < 1 || cols < 1 || k < 1){
            return res;
        }
        std::multimap<int, int> powerMap;
        for (int i = 0; i < rows; ++i) {
            int linePower = 0;
            for (auto& num : mat[i]) {
            	// 提前结束军人数量的计算
                if (num == 0) {
                    break;
                }
                linePower += num;
            }
            powerMap.insert(std::pair(linePower, i));
        }
        auto it = powerMap.begin();
        for(int j = 0; j < k; ++j) {
            res.push_back(it->second);
            ++it;
        }
        return res;
    }
};

关于 multimap 用法的一些要点：

• multimap 中保存的是有序的键值对，即 std::pair<K, V> 类型的元素，它能够保存重复的 Key，默认的比较键的函数为 less<K>()，因此本题直接用默认的比较器即可。

• multimap 的 insert 操作可以插入一个或多个元素并且总能成功，如果 multimap 中的 Key 已经存在，则插入到相同 Key 区域的上界，而 map 的 insert 操作在 Key 存在的情况下会插入失败。

• multimap 的 erase 操作既可以按照迭代器来擦除，也可以按照 Key 来擦除，如果使用迭代器擦除，指向被擦出元素的引用和迭代器会失效，其他元素的引用和迭代器不受影响；如果使用 Key 擦除，会移除键值为 Key 的所有元素。

• multimap 的 find 操作在所查 Key 存在的情况下可能返回任意一个对应的 Value。