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];        }    }};