leetcode 119. Pascal's Triangle II 杨辉三角形2

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

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

就是中国最伟大的杨辉三角形。

代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution 
{
    public List<Integer> getRow(int rowIndex)
    {
        List<List<Integer>> res=new ArrayList<>();
        if(rowIndex==0)
        {
            List<Integer> one=new ArrayList<>();
            one.add(1);
            res.add(one);
            return res.get(0);
        }else
        {
            List<Integer> one=new ArrayList<>();
            one.add(1);
            one.add(0);
            res.add(one);
            for(int i=1;i<rowIndex+1;i++)
            {
                List<Integer> tt=res.get(i-1);
                List<Integer> rr=new ArrayList<>();
                rr.add(tt.get(0));
                for(int j=1;j<tt.size();j++)
                    rr.add(tt.get(j) + tt.get(j-1));

                rr.add(0);
                res.add(rr);
            }
            for(int i=0;i<res.size();i++)
                res.get(i).remove(res.get(i).size()-1);
            return res.get(rowIndex);
        }
    }
}

下面是C++的做法，和上面的一道题一模一样，就是做一个迭代计算即可完成

代码如下：

#include <iostream>
#include <vector>

using namespace std;

class Solution 
{
public:
    vector<int> getRow(int rowIndex) 
    {
        if (rowIndex == 0)
            return vector<int>(1,1);
        else
        {
            vector<vector<int>> res;
            vector<int> one{ 1 ,0 };
            res.push_back(one);
            for (int i = 1; i < rowIndex+1; i++)
            {
                vector<int> one{ res[i - 1][0] };
                for (int j = 1; j < res[i - 1].size(); j++)
                    one.push_back(res[i - 1][j] + res[i - 1][j - 1]);
                one.push_back(0);
                res.push_back(one);
            }
            for (int i = 0; i < res.size(); i++)
                res[i].erase(res[i].end() - 1);
            return res[rowIndex];
        }
    }
};