使用ArrayList解决杨辉三角问题

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

在杨辉三角中,每个数是它左上方和右上方的数的和。
算法思想

  1. 用for循环创建numRows个动态链表,代表杨辉三角的前numRows行;
  2. 设置变量i和j,分别代表行和列,因为第0列和对角线数据为1,所以将j为0和j与i相等的位置设为1;
  3. 设置其他位置:当前位置=前一行正对数据与前一个数据的和。
    注意:在使用set(index,element)向ArrayList中设置元素时,index必须在size范围内,不能越界。
    Java代码
public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> list = new ArrayList<>();
        for(int i = 0;i < numRows;i++){
            List<Integer> l = new ArrayList<>();
            for(int j = 0;j <= i;j++){
                if(i == j || j == 0){
                    l.add(1);
                }else{
                    l.add(0);
                }
            }
            list.add(l);
        }
        for(int i = 2;i < numRows;i++){
            List<Integer> pre = list.get(i - 1);
            List<Integer> cur = list.get(i);
            for(int j = 1;j < i;j++){
                cur.set(j,pre.get(j - 1) + pre.get(j));
            }
        }
        return list;
    }

猜你喜欢

转载自blog.csdn.net/qq_43452252/article/details/104435088