leetcode Pascal's Triangle II 2.23 难度系数2

Question:

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 Solution {
    public ArrayList<Integer> getRow(int rowIndex) {
		ArrayList<Integer> prev = new ArrayList<Integer>();
		ArrayList<Integer> current = new ArrayList<Integer>();
		
		for (int i = 0; i <= rowIndex; i++) {
			if (i == 0) {
				current.add(1);
			} else {
				for (int j = 0; j <= i; j++) {
					if (j == 0 || j == i) {
						current.add(1);
					} else {
						current.add(prev.get(j - 1) + prev.get(j));
					}
				}
			}

			ArrayList<Integer> temp = prev;
			prev = current;
			current = temp;
			current.clear();
		}
		return prev;
	}
}

或者在前一个题的基础上直接拿取最后一个:

public class Solution {
    public ArrayList<Integer> getRow(int rowIndex) {
	 ArrayList<ArrayList<Integer>> results = new ArrayList<>();
	    ArrayList<Integer> ini = new ArrayList<>();
	    ini.add(1);
	    results.add(ini);
	    for(int i=1;i<=rowIndex;i++){
	        ArrayList<Integer> prev = results.get(i-1);
	        ArrayList<Integer> cur = new ArrayList<>();
	        cur.add(1);
	        for(int j=1;j<i;j++){
	            cur.add(prev.get(j)+prev.get(j-1));
	        }
	        cur.add(1);
	        results.add(cur);
	    }
	    return results.get(results.size()-1);
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值