首页 > 代码库 > Pascal's Triangle & II 解题报告

Pascal's Triangle & II 解题报告

杨辉三角,分别求前n行和第n行。

【求杨辉三角前n行】

Given numRows, generate the first numRows of Pascal‘s triangle.

For example, given numRows = 5,
Return

[
     [1],
    [1,1],
   [1,2,1],
  [1,3,3,1],
 [1,4,6,4,1]
]
基础题,直接看代码,注意边界。

public class Solution {
    public List<List<Integer>> generate1(int numRows) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        
        if (numRows == 0) return ret;
        
        List<Integer> list0 = new ArrayList<Integer>(); 
        list0.add(1);
        ret.add(list0);
        
        for (int i = 2; i <= numRows; i++) {
            List<Integer> list = new ArrayList<Integer>(); 
            list.add(1);
            
            List<Integer> pre = ret.get(ret.size()-1);
            for (int j = 1; j < i-1; j++) {
                list.add(pre.get(j-1) + pre.get(j));
            }
            
            list.add(1);
            ret.add(list);
        }
        
        return ret;
    }
    
    // another style
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        
        for (int i = 0; i < numRows; i++) {
            List<Integer> list = new ArrayList<Integer>(); 
            
            if (i == 0) {
                list.add(1);
            } else {
                List<Integer> pre = ret.get(ret.size()-1);
                for (int j = 0; j <= i; j++) {
                    if (j == 0 || j == i) {
                        list.add(1);
                    } else {
                        list.add(pre.get(j-1) + pre.get(j));
                    }
                }
            }
            
            ret.add(list);
        }
        
        return ret;
    }
}

【求杨辉三角第n行】

Given an index k, return the kth row of the Pascal‘s triangle.

For example, given k = 3,
Return [1,3,3,1].

Note:
Could you optimize your algorithm to use only O(k) extra space?

注意 [1] 算第0行,[1, 1] 为第1行……

用O(k)的空间记录前一行就行了。

public class Solution {
    public List<Integer> getRow(int rowIndex) {
        List<Integer> list = new ArrayList<Integer>();
        list.add(1);
        
        List<Integer> pre = new ArrayList<Integer>(list);
        for (int i = 1; i <= rowIndex; i++) { 
        	list.clear();
            list.add(1);
        	for (int j = 1; j < pre.size(); j++) {
                list.add(pre.get(j) + pre.get(j-1));
            }
            list.add(1);
            pre.clear();
            pre.addAll(list);
        }
        
        return list;
    }
}


Pascal's Triangle & II 解题报告