重塑矩阵

要求

MATLAB 中,有一个非常有用的函数 reshape ,它可以将一个 m x n 矩阵重塑为另一个大小不同r x c的新矩阵,但保留其原始数据。

给你一个由二维数组 mat 表示的 m x n 矩阵,以及两个正整数 rc ,分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。

如果具有给定参数的 reshape 操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。

示例

输入:mat = [[1,2],[3,4]], r = 1, c = 4
输出:[[1,2,3,4]]

解答

首先判断矩阵是否可以按照rc进行变换,不行则返回原矩阵,否则新建二维数组,从左到右从上到下依次遍历原矩阵的每个元素,对应到新矩阵的[i / c][i % c]元素中,最后返回新矩阵

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int m = mat.length;
        int n = mat[0].length;
        if(r*c != m*n){
            return mat;
        }
        int[][] num = new int[r][c];
        for (int i = 0; i < m * n; ++i) {
            num[i / c][i % c] = mat[i / n][i % n];
        }
        return num;
    }
}

杨辉三角

要求

给定一个非负整数 numRows,生成「杨辉三角」的前 numRows 行。

在「杨辉三角」中,每个数是它左上方和右上方的数的和。

示例

输入: numRows = 5
输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]

解答

class Solution {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> ret= new ArrayList<List<Integer>>();
        for(int i = 0 ; i < numRows ; i++){
            List<Integer> row = new ArrayList<Integer>();
            for(int j = 0 ; j <= i ; j++){
                if( j==0 || i == j){
                    row.add(1);
                }else{
                    row.add(ret.get(i-1).get(j-1)+ret.get(i-1).get(j));
                }
            }
            ret.add(row);
        }
        return ret;
    }
}