Day-4 旋转&矩阵置零

 

 

AcWing 3174. 旋转

图片旋转是对图片最简单的处理方式之一，在本题中，你需要对图片顺时针旋转 9090 度。

我们用一个 n×mn×m 的二维数组来表示一个图片，例如下面给出一个 3×43×4 的图片的例子：

1 3 5 7
9 8 7 6
3 5 9 7

这个图片顺时针旋转 9090 度后的图片如下：

3 9 1
5 8 3
9 7 5
7 6 7

给定初始图片，请计算旋转后的图片。

输入格式

输入的第一行包含两个整数 nn 和 mm，分别表示行数和列数。

接下来 nn 行，每行 mm 个整数，表示给定的图片。图片中的每个元素（像素）为一个值为 00 至 255255 之间的整数（包含 00 和 255255）。

输出格式

输出 mm 行 nn 列，表示旋转后的图片。

数据范围

1≤n,m≤1001≤n,m≤100

输入样例：

3 4
1 3 5 7
9 8 7 6
3 5 9 7

输出样例：

3 9 1
5 8 3
9 7 5
7 6 7

#include <iostream>

using namespace std;

const int N = 110;

int a[N][N];
int n,m;

int main()
{
    cin >> n >> m;
    for(int i = 0; i < n; i ++ )
        for(int j = 0; j < m; j ++ )
            cin >> a[i][j];
            
    for(int j = 0; j < m; j ++ )
    {
        for(int i = n - 1; i >= 0; i -- )
            cout << a[i][j] << ' ';
        
        cout << endl;
    }
    
    return 0;
}

 

LeetCode 73.矩阵置零

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

进阶：

一个直观的解决方案是使用  O(mn) 的额外空间，但这并不是一个好的解决方案。
一个简单的改进方案是使用 O(m + n) 的额外空间，但这仍然不是最好的解决方案。
你能想出一个仅使用常量空间的解决方案吗？
 

示例 1：

输入：matrix = [[1,1,1],[1,0,1],[1,1,1]]
输出：[[1,0,1],[0,0,0],[1,0,1]]
示例 2：

输入：matrix = [[0,1,2,0],[3,4,5,2],[1,3,1,5]]
输出：[[0,0,0,0],[0,4,5,0],[0,3,1,0]]
 

提示：

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

思路 

class Solution {
public:
    void setZeroes(vector<vector<int>>& matrix) {
        int col = 1, row = 1;
        int n = matrix.size(), m = matrix[0].size();

        for(int i = 0; i < n; i ++ )
            if(matrix[i][0] == 0) col = 0;
        for(int i = 0; i < m; i ++ )
            if(matrix[0][i] == 0) row = 0;
        
        for(int i = 1; i < n; i ++ )
        {
            for(int j = 1; j < m; j ++ )
            {
                if(matrix[i][j] == 0)
                {
                    matrix[i][0] = 0;
                    matrix[0][j] = 0;
                }
            }
        }

        for(int i = 1; i < n; i ++ )
            if(matrix[i][0] == 0)
            {
                for(int j = 1 ; j < m; j ++ )
                    matrix[i][j] = 0;
            }

        for(int j = 1; j < m; j ++ )
            if(matrix[0][j] == 0)
            {
                for(int i = 1; i < n; i ++ )
                    matrix[i][j] = 0;
            }
        
        if(!col) 
            for(int i = 0; i < n; i ++ )
                matrix[i][0] = 0;
        if(!row)
            for(int j = 0; j < m; j ++ )
                matrix[0][j] = 0;
    }
};