当前位置:   article > 正文

LeetCode73:矩阵置零_给定一个 × n×m 的矩阵,如果一个元素为 0 0,则将其所在行和列的所有

给定一个 × n×m 的矩阵,如果一个元素为 0 0,则将其所在行和列的所有

题目:

给定一个 m x n 的矩阵,如果一个元素为 ,则将其所在行和列的所有元素都设为 0 。请使用 原地 算法

示例:

提示:

m == matrix.length
n == matrix[0].length
1 <= m, n <= 200
-2^31 <= matrix[i][j] <= 2^31 - 1

本道题的题目并不难理解,最重要的就是需要在原来的矩阵中进行修改

第一步先定义vector数组来记录0元素的横纵坐标

  1. vector<int> x;
  2. vector<int> y;

下一步利用两个for()来把矩阵所有的元素遍历一遍。如果遇到0元素,就将横坐标push_back()到x中,纵坐标push_back( )到y中。

  1. int m = matrix.size();
  2. int n = matrix[0].size();
  3. for (int i = 0; i < m; i++)
  4. {
  5. for (int j = 0; j < n; j++)
  6. {
  7. if (matrix[i][j] == 0)
  8. {
  9. x.push_back(i);
  10. y.push_back(j);
  11. }
  12. }
  13. }

然后再用sort()排序函数对x,y数组中的元素进行排序

  1. sort(x.begin(), x.end());
  2. sort(y.begin(), y.end());

然后最后一步就该在原矩阵中进行修改

利用for()遍历所有的矩阵元素时

binary_search( )二分查找i,j是否在x,y数组中出现过,返回true,则将其元素修改为0;

  1. for (int i = 0; i < m; i++)
  2. {
  3. for (int j = 0; j < n; j++)
  4. {
  5. bool ret = binary_search(x.begin(), x.end(), i);
  6. bool res = binary_search(y.begin(), y.end(), j);
  7. if (ret == true || res == true)
  8. {
  9. matrix[i][j] = 0;
  10. }
  11. }
  12. }

ps:binary_search()必须是bool类型定义,而且查找方法是二分查找,所以必须是有序数组,这也是提前用sort()函数的意图所在。

完整代码:

  1. class Solution
  2. {
  3. public:
  4. void setZeroes(vector<vector<int>> &matrix)
  5. {
  6. vector<int> x;
  7. vector<int> y;
  8. int m = matrix.size();
  9. int n = matrix[0].size();
  10. for (int i = 0; i < m; i++)
  11. {
  12. for (int j = 0; j < n; j++)
  13. {
  14. if (matrix[i][j] == 0)
  15. {
  16. x.push_back(i);
  17. y.push_back(j);
  18. }
  19. }
  20. }
  21. sort(x.begin(), x.end());
  22. sort(y.begin(), y.end());
  23. for (int i = 0; i < m; i++)
  24. {
  25. for (int j = 0; j < n; j++)
  26. {
  27. bool ret = binary_search(x.begin(), x.end(), i);
  28. bool res = binary_search(y.begin(), y.end(), j);
  29. if (ret == true || res == true)
  30. {
  31. matrix[i][j] = 0;
  32. }
  33. }
  34. }
  35. }
  36. };

本道题的思路就是这样了。

欢迎各位读者的补充!
 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/257029?site
推荐阅读
相关标签
  

闽ICP备14008679号