杨辉三角(Java)

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

示例:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/pascals-triangle
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解:
杨辉三角的特点:
1. 第一行固定为 1
2. 第二行也是固定为1,也就是两个 1
3. 第 i 行,首尾元素都固定为1
4. 第 i 行有 i 个元素
5. 对于第 i 行来说,第 j 列的值是 i - 1 行 j - 1 列的值 加上 i - 1 行 j 列的值

import java.util.ArrayList;
import java.util.List;

public class YangHui {
    public List<List<Integer>> generate(int numRows) {
        if (numRows <= 0) {
            return new ArrayList<>();
        }
        List<List<Integer>> result = new ArrayList<>();
        //1.插入第一行
        List<Integer> firstLine = new ArrayList<>();
        firstLine.add(1);
        result.add(firstLine);
        if  (numRows == 1) {
            return result;
        }
        //2.插入第二行,只有两个元素
        List<Integer> secondLine = new ArrayList<>();
        secondLine.add(1);
        secondLine.add(1);
        result.add(secondLine);
        if (numRows == 2) {
            return result;
        }
        //3.处理第 i 行的情况
        // a)第 i 行有 i 列
        // b)(i, j) = (i - 1, j) + (i - 1, j - 1);
        // c)第一个元素和最后一个元素都固定为 1
        for (int row = 3; row <= numRows; row++) {
            //如果想知道第 row 行的情况,就得先知道 row - 1 行
            List<Integer> prevLine = result.get((row - 1) -1);
            //第一次减一是为了得到 row - 1 行
            //第二次是为了得到 row - 1 行的下标
            //下标是从 0 开始的
            List<Integer> curLine = new ArrayList<>();//当前行
            curLine.add(1);
            //第 row 行应为row列, 下面这个循环相当于循环了 row - 2 次
            //因为第一列和最后一列都是 固定的 1 ,不参与循环
            //针对 差一 问题, 最好的办法就是套入具体数字来验证是否合理
            //如果row 为 5 ,第5 行应该有第五列 , 下面的循环应执行3次
            for (int col = 2; col < row; col++) {
                //col 是从 1 开始计数的, 需要转换成下标 ,所以 - 1
                int curNum = prevLine.get(col - 1 - 1) + prevLine.get(col - 1);
                curLine.add(curNum);
            }
            //处理该行最后一个 1
            curLine.add(1);
            result.add(curLine);
        }
        return result;
    }
}

原创文章 11 获赞 6 访问量 302

猜你喜欢

转载自blog.csdn.net/weixin_45122262/article/details/105987867