119 Pascal's Triangle II

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> result(rowIndex + 1,1);
        if(rowIndex < 2) return result;
        
        for(int i = 2;i <= rowIndex;++i){
            for(int j = i - 1;j >= 0;--j){
                result[j] += result[j - 1];
            }
        }
        
        return result;
    }
};

 

这道题跟Pascal's Triangle很类似，只是这里只需要求出某一行的结果。Pascal's Triangle中因为是求出全部结果，所以我们需要上一行的数据就很自然的可以去取。而这里我们只需要一行数据，就得考虑一下是不是能只用一行的空间来存储结果而不需要额外的来存储上一行呢？这里确实是可以实现的。对于每一行我们知道如果从前往后扫，第i个元素的值等于上一行的res[i]+res[i+1]，可以看到数据是往前看的，如果我们只用一行空间，那么需要的数据就会被覆盖掉。所以这里采取的方法是从后往前扫，这样每次需要的数据就是res[i]+res[i-1]，我们需要的数据不会被覆盖，因为需要的res[i]只在当前步用，下一步就不需要了。这个技巧在动态规划省空间时也经常使用，主要就是看我们需要的数据是原来的数据还是新的数据来决定我们遍历的方向。时间复杂度还是O(n^2)，而空间这里是O(k)来存储结果，仍然不需要额外空间。代码如下：
[java] view plaincopy在CODE上查看代码片派生到我的代码片

    public ArrayList<Integer> getRow(int rowIndex) {  
        ArrayList<Integer> res = new ArrayList<Integer>();  
        if(rowIndex<0)  
            return res;  
        res.add(1);  
        for(int i=1;i<=rowIndex;i++)  
        {  
            for(int j=res.size()-2;j>=0;j--)  
            {  
                res.set(j+1,res.get(j)+res.get(j+1));  
            }  
            res.add(1);  
        }  
        return res;  
    }  

这道题相比于Pascal's Triangle其实更有意思一些，因为有一个考点就是能否省去额外空间，在面试中出现的可能性大一些，不过总体比较简单，电面中比较合适。