题解 | #杨辉三角(二)#

import java.util.*;


public class Solution {
    /**
     * 返回杨辉三角中从上到下第 num 层,时间复杂度O(n2),空间复杂度O(n)
     *
     * @param num int整型 
     * @return int整型一维数组
     */
    public int[] getRow (int num) {
        if(num==0){
            return new int[] {1};
        }
        int[] lastLine=new int[]{1,1};
        if(num==1){
            return lastLine;
        }
        int[] res=null;
        for(int i=2;i<=num;i++){
            res=new int[i+1];
            for(int j=0;j<=i;j++){
                if(j==0||j==i){
                    res[j]=1;
                }else{
                    res[j]=lastLine[j-1]+lastLine[j];
                }
            }
            lastLine=res;
        }
        return res;
    }
}