day8:矩阵

将一维数组转变为二维数组

func construct2DArray(original []int, m int, n int) [][]int {
    if len(original) != n*m {
        return [][]int{}
    }
    res := make([][]int, m)
    for i:=0;i<m;i++ {
        res[i] = make([]int, 0)
    }

    i, j, col := 0, n, 0
    for col < m {
        res[col] = append(res[col], original[i:j]...)
        i = j
        j = j + n
        col++
    }
    return res
}

重塑矩阵

func matrixReshape(mat [][]int, r int, c int) [][]int {
    if len(mat)*len(mat[0]) != r*c {
        return mat
    }

    res := make([][]int, r)
    for i:=0;i<r;i++ {
        res[i] = make([]int, c)
    }

    for i:=0;i<len(mat)*len(mat[0]);i++ {
        res[i/c][i%c] = mat[i/len(mat[0])][i%len(mat[0])]
    }
    return res
}

反转图像

class Solution {
public:
    vector<vector<int>> flipAndInvertImage(vector<vector<int>>& image) {
        for(int i=0;i<image.size();i++) {
            reverse(image[i].begin(),image[i].end());
            for (int j=0;j<image[i].size();j++) {
                image[i][j] ^= 1;
            }
        }
        return image;
    }
};

转置矩阵

func transpose(matrix [][]int) [][]int {
    res := make([][]int, len(matrix[0]))
    for i:=0;i<len(matrix[0]);i++ {
        res[i] = make([]int, len(matrix))
    }

    for i:=0;i<len(matrix);i++ {
        for j:=0;j<len(matrix[0]);j++ {
            res[j][i] = matrix[i][j]
        }
    }
    return res
}

二维网格迁移

func shiftGrid(grid [][]int, k int) [][]int {
    for i:=k;i>0;i-- {
        pre := grid[len(grid)-1][len(grid[0])-1]
        for i:=0;i<len(grid);i++ {
            for j:=0;j<len(grid[0]);j++ {
                grid[i][j] = grid[i][j]^pre
                pre = grid[i][j]^pre
                grid[i][j] = grid[i][j]^pre
            }
        }
    }
    return grid
}