Pascal's Triangle II @LeetCode

最新推荐文章于 2025-11-27 10:55:11 发布

最新推荐文章于 2025-11-27 10:55:11 发布 · 63 阅读

文章标签：

#java

本文介绍了一种基于前一行结果计算帕斯卡三角形特定行的方法，并通过Java实现了一个具体的例子。该方法使用了额外的空间来存储中间结果，尽管如此，作者认为存在更高效的算法仅使用O(k)的额外空间。

用了笨方法，基于前一题的结果，来取值。应该有用数学公式直接计算出来的办法。

package Level2;

import java.io.ObjectInputStream.GetField;
import java.util.ArrayList;

/**
 * Pascal's Triangle II 
 * 
 * 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?
 */
public class S119 {

	public static void main(String[] args) {
		System.out.println(getRow(0));
	}

	public static ArrayList<Integer> getRow(int rowIndex) {
		ArrayList<ArrayList<Integer>> list = generate(rowIndex+1);
		return list.size()==0 ? null : list.get(rowIndex);
    }
	
	private static ArrayList<ArrayList<Integer>> generate(int numRows) {
		ArrayList<ArrayList<Integer>> ret = new ArrayList<ArrayList<Integer>>();
		int i, j;
		
		// i用来走列
        for(i=0; i<numRows; i++){
        	ArrayList<Integer> list = new ArrayList<Integer>();
        	// j用来走行，先走前半行
        	for(j=0; j<=i/2; j++){
        		if(j == 0){			// 对首位特殊处理
            		list.add(1);
            	}else if(i>0){		// 累加前面结果
            		list.add(ret.get(i-1).get(j-1)+ret.get(i-1).get(j));
            	}
        	}
        	// i再走后半行
        	for(; j<=i; j++){
        		if(j == i){		// 对末位特殊处理
        			list.add(1);
        		}else if(i>0){	// 累加前面结果
        			list.add(ret.get(i-1).get(j-1)+ret.get(i-1).get(j));
        		}
        	}
        	ret.add(list);
        }
        return ret;
    }
}