Pascal's Triangle II

来源:互联网 发布:韵达快运网络招商 编辑:程序博客网 时间:2024/06/14 02:22

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行的数据。

Note:最好让算法只用O(k)的额外空间。

直接初始化第n行大小的空间,然后从后往前求下一行的数据就可以只用O(k)的额外空间了,代码如下:

class Solution {public:    vector<int> getRow(int rowIndex) {        vector<int> vec(rowIndex+1);         vec[0]=1;         for(int i=1;i<=rowIndex;i++)         {             for(int j=i;j>=0;j--)             {                 if(j==i)                     vec[j]=vec[j-1];                 else if(j==0)                     vec[j]=vec[j];                 else                     vec[j]=vec[j-1]+vec[j];             }         }         return vec;      }};


0 0
原创粉丝点击