119. Pascal's Triangle II (Graph; WFS)

帕斯卡三角形求特定行

weixin_30271335

于 2015-10-04 15:43:00 发布

阅读量52

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/qionglouyuyu/p/4854608.html

本文介绍了一种高效算法来求解帕斯卡三角形的特定行，特别是使用了仅O(k)的额外空间优化算法。通过从右到左遍历并更新三角形的图结构，该算法能快速准确地计算出给定行的所有元素。

Given an index k, return the k^th 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?

思路：从右往左遍历并填写三角形的图结构

class Solution {
public:
    vector<int> getRow(int rowIndex) {
        vector<int> a(rowIndex + 1);
        
        a[0] = 1;
        for(int i = 1; i <= rowIndex; i++) //遍历行
            for(int j = i; j >= 0; j--) //遍历行中的元素，第k行有k个元素。注意行中元素需要从右往左遍历，否则下一行填写的结果会覆盖上一行还未使用的元素
                if (j == i)
                    a[j] = a[j-1];
                else if (j == 0)
                    a[j] = a[j];
                else
                    a[j] = a[j-1] + a[j];
                    
        return a;                    
    }
};