每日算法题--杨辉三角(简单)

题目描述：

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

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

示例:

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

动态图片演示：

题解：

首先，我们换一种思路看一下杨辉三角。

1

1 1

1 2 1

1 3 3 1

1 4 6 4 1

这就是个二维数组的形式。

我们可以先设一个二维数组arr【x】【y】，然后根据图可知：

当y=0的时候,都为1，当x=y的时候，都为1。其他数是它左上方和右上方的数的和。

由此可以判断

 if(j == 0 || j==i){
		arr[i][j] = 1;
	}else{
		arr[i][j] = arr[i-1][j-1] + arr[i-1][j];
	}

又有题意可知

 [1],
[1,1],

[1,2,1],
[1,3,3,1],
[1,4,6,4,1]

每一行就是一个数组形式，所以要将二维数组arr【x】【y】对应的值存到一个数组temp里面，因为涉及的是一行数据，就是当i=0当做一行，添加到temp里面、就是当i=1当做一行，添加到temp里面…

所以可得下面代码

 for(int i = 0 ; i < numRows; i++){
	for (int j = 0 ; j <=i ;  j++){
		if(j == 0 || j==i){
			arr[i][j] = 1;
			}else{
				arr[i][j] = arr[i-1][j-1] + arr[i-1][j];
			}
                temp.add(arr[i][j]);
            }
 }

补充一下for (int j = 0 ; j <=i ; j++)，这里为什么 j <=i，因为性状限定了，最外围的永远是j=i=1.

这里的temp数组就存了每一行数据，也就是说 [1,1]就是temp的值了

又根据题意可得

[
[1],
[1,1],
[1,2,1],
[1,3,3,1],
[1,4,6,4,1]
]

在temp数组上，还封装了一个数组res，要将temp里面的值有一个一个加到res数组里面，所以就有了下面代码

又因为temp在进行下一行i的时候就得要归零，不然会将temp里面所有的值全都弄到res中，结果会出错。

 for(int i = 0 ; i < numRows; i++){
    List<Integer> temp = new ArrayList<>();
	for (int j = 0 ; j <=i ;  j++){
		if(j == 0 || j==i){
			arr[i][j] = 1;
			}else{
				arr[i][j] = arr[i-1][j-1] + arr[i-1][j];
			}
                temp.add(arr[i][j]);
            }
     res.add(temp);
 }

然后因为res.add(temp)是数组里面存了数组，所以就有了下面代码

List<List<Integer>> res = new ArrayList<>();

总体代码：

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()){
            int a = sc.nextInt();
           	System.out.println(generate(a));
        }   
    }

    public static List<List<Integer>> generate(int numRows) {
        List<List<Integer>> res = new ArrayList<>();      
        int[][] arr = new int[numRows][numRows];
        for(int i = 0 ; i < numRows; i++){
         List<Integer> temp = new ArrayList<>();
            for (int j = 0 ; j <=i ;  j++){
                if(j == 0 || j==i){
                    arr[i][j] = 1;
                }else{
                    arr[i][j] = arr[i-1][j-1] + arr[i-1][j];
                }
                temp.add(arr[i][j]);
            }
            res.add(temp);
        }
        return res;

    }
}

算法题来源：力扣
欢迎朋友们一起进步！！！
2020-12-08