《leetCode》： 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?

思路一

由于题目只需要得到杨辉三角的第k层的元素。由于上篇博文可以得到我们整个杨辉三角每层的元素，因此，直接调用上篇博文的函数即可完成

public class Solution {    public List<Integer> getRow(int rowIndex) {        if(rowIndex<0){            return  new ArrayList<Integer>();        }        List<List<Integer>> result=generate( rowIndex+1);//rowIndex是从零开始的        return result.get(result.size()-1);    }    /*     * 思路：根据层数来算出杨辉三角的值     * 杨辉三角的特性：第i层有i个元素，且每层的元素等于上一层的两个元素之后     * */    public List<List<Integer>> generate(int numRows) {        List<List<Integer>> res=new ArrayList<List<Integer>>();        if(numRows<1){            return res;        }        List<Integer> temp=new ArrayList<Integer>();        temp.add(new Integer(1));        res.add(temp);        for(int i=2;i<=numRows;i++){//从第二层开始            List<Integer> I_Res=new ArrayList<Integer>();            I_Res.add(1);//每层的第一个元素均是1            for(int j=1;j<i-1;j++){//算每层中间的元素，等于上一层的两个元素之和                int num=temp.get(j-1)+temp.get(j);//要注意的是j出界,为避免出界，因此将每层的最后一个元素分开加入                I_Res.add(num);            }            I_Res.add(1);//每层的最后一个元素也为1;            res.add(I_Res);            //I_Res.clear();//清零            temp=new ArrayList<Integer>(I_Res);//更新，供下一层的计算        }        return res;    }}

但是，由于上篇借助的空间不是O(k)，因此算法上需要进行改变，不能将杨辉三角的每层的元素值进行保存，然后返回最后一层的元素；而是只保存上一层的元素即可

实现代码如下：

public class Solution {       public List<Integer> getRow(int rowIndex) {            if(rowIndex<0){                return  new ArrayList<Integer>();            }            List<Integer> result=generate( rowIndex+1);//rowIndex是从零开始的            return result;        }    //@param rowIndex是从1开始    private List<Integer> generate(int rowIndex) {        List<Integer> curFloor=new ArrayList<Integer>();        List<Integer> nextFloor=new ArrayList<Integer>();        curFloor.add(1);        if(rowIndex==1){//第一行单独拿出来考虑            return curFloor;        }        for(int i=2;i<=rowIndex;i++){            nextFloor.clear();//清零            nextFloor.add(1);            for(int j=1;j<i-1;j++){//注意j是从1开始，到i-2                int num=curFloor.get(j-1)+curFloor.get(j);                nextFloor.add(num);            }            nextFloor.add(1);            curFloor.clear();            curFloor=new ArrayList<Integer>(nextFloor);        }        return nextFloor;    }}