119. Pascal’s Triangle II

119. 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].

杨辉三角,主要是要在O(n)的空间复杂度实现
如果直接用组合数公式推导,阶乘计算会溢出
所以还是逐行递推下去,不过每次都从末尾开始更新,因为ans[j]=ans[j]+ans[j-1],不会擦除掉下一个要用的数据,再处理下边界就行。

代码

vector<int> getRow(int rowIndex)
{
    if (rowIndex == 0)
    {
        vector<int> ans;
        ans.push_back(0);
        return ans;
    }
    else
    {
        vector<int> ans(rowIndex + 1);
        ans[0] = 1;
        for (int i = 1; i <= rowIndex; i++)
        {
            for (int j = i; j >= 0; j--)
            {               //1 1
                if (j == i || j == 0)
                {
                    ans[j] = 1;
                }//1 2 1
                else
                {
                    ans[j] = ans[j] + ans[j - 1];
                }
            }
        }
        return ans;
    }
}

你可能感兴趣的:(LeetCode)