java实现杨辉三角

朱雀 2024-03-22 15:32 20阅读 0赞

力扣原题链接  
\[https://leetcode.cn/problems/pascals-triangle/

首先我们必须要知道杨辉三角用什么样的数据结构存储时最合适的  
![在这里插入图片描述][850245f783d5405a9f95480d68048af4.png]从这里看出来应该是二维数组比较合适  
我们函数形参设置为杨辉三角这个二维数组的行数rowNumber，该函数的返回值是`ArrayList<ArrayList<Integer>>`，这是一个二维数组的Integer泛型.

import java.util.ArrayList;
    
    //杨辉三角
    public class Test {
        
        public static ArrayList<ArrayList<Integer>> YangHuiTriangle(int rowNember) {
        
    
    
            ArrayList<Integer> preArrayRow = new ArrayList<>();//上一行
            ArrayList<ArrayList<Integer>> array = new ArrayList<>();
    
            //后面的行
            for (int i = 0; i < rowNember; i++) {
        
                //第一行
                if (i == 0) {
        
                    ArrayList<Integer> firstRow = new ArrayList<>();
                    firstRow.add(1);
                    array.add(firstRow);
                } else {
        
                    ArrayList<Integer> curArrayRow = new ArrayList<>();//现在的行
                    preArrayRow = array.get(i - 1);//获取前一行
    
                    //每一行第一个元素是1
                    curArrayRow.add(1);
    
                    //中间元素满足[i][j] =[i-1][j-1] + [i-1][j]
                    for (int j = 1; j < i; j++) {
        
                        curArrayRow.add(preArrayRow.get(j) + preArrayRow.get(j - 1));
                    }
    
                    //每一行的最后一个元素
                    curArrayRow.add(1);
                    array.add(curArrayRow);
                }
            }
            return array;
        }
    
        public static void main(String[] args) {
        
    
            ArrayList<ArrayList<Integer>> arrayList = new ArrayList<>();
            arrayList = YangHuiTriangle(5);
            for (int i = 0; i < arrayList.size(); i++) {
        
                for (int j = 0; j < arrayList.get(i).size(); j++) {
        
                    System.out.print(arrayList.get(i).get(j) + " ");
                }
                System.out.println();
            }
    
        }
    }

![在这里插入图片描述][3307d010ce0f48cd838abd56d4ab6ffd.png]

[850245f783d5405a9f95480d68048af4.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/18/ce088ee5fb9647bdbcb2409e36ff9a90.png
[3307d010ce0f48cd838abd56d4ab6ffd.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/18/0086e75bd18a4bf1b58bafc62f55790d.png