题目

给定一个非负索引 k,其中 k ≤ 33,返回杨辉三角的第 行。

leetcode 119 : 杨辉三角2

leetcode 119 : 杨辉三角2

 

算法思想 :首先注意的是第k行有k+1个元素,正常解法是用二维数组存储,返回第k行,下面我们用O(k)的空间复杂的求解,我们用一个大小k+1的一位数组来模拟二维数组的操作,为了防止覆盖需要的数据,所以从后往前计算。公式是a[j] = a[j]+a[j-1].

 

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

相关文章: