C练题笔记之:Leetcode-1337. 方阵中战斗力最弱的 K 行

    科技2022-08-29  107

    题目:

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

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

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

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

     

    示例 1:

    输入: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] 示例 2:

    输入: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]

    提示:

    m == mat.length n == mat[i].length 2 <= n, m <= 100 1 <= k <= m matrix[i][j] 不是 0 就是 1

    来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/the-k-weakest-rows-in-a-matrix 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    结果:

    解题思路:

    通过一个二维数组,【0】放如行号,【1】放入军人数量,按照【1】的数量进行二维数组排序。再把前几个【0】的数字放入返回的数组中。

    代码:

    /** * Note: The returned array must be malloced, assume caller calls free(). */ int compar(const void *a, const void *b) { return ((int *)a)[1] - ((int *)b)[1]; } int* kWeakestRows(int** mat, int matSize, int* matColSize, int k, int* returnSize){ int temp[100][2] = {}; int *ret = (int *)malloc(sizeof(int) * k); for(int i = 0; i < matSize; i++) { temp[i][0] = i; temp[i][1] = 0; for(int j = 0; j < *matColSize; j++) { if(mat[i][j]==1) { temp[i][1] += 1; } else { break; } } } qsort(temp, matSize, sizeof(int *), compar); for(int i = 0; i < k; i++) { ret[i] = temp[i][0]; } *returnSize = k; return ret; }

     

    Processed: 0.010, SQL: 9