LintCode-k数和 II

给定n个不同的正整数，整数k（1<= k <= n）以及一个目标数字。　　　　

在这n个数里面找出K个数，使得这K个数的和等于目标数字，你需要找出所有满足要求的方案。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出[1,2,3,4]，k=2， target=5，返回 [[1,4],[2,3]]

标签  Expand  

相关题目  Expand 

分析：就是简单的递归练习，注意剪枝

代码：

class Solution {
public:
    /**
     * @param A: an integer array.
     * @param k: a positive integer (k <= length(A))
     * @param target: a integer
     * @return a list of lists of integer 
     */
    vector<vector<int> > kSumII(vector<int> A, int k, int target) {
        // write your code here
        vector<int> cur;
        vector<vector<int> > ret;
        deal(A,0,k,target,0,cur,ret);
        return ret;
    }
    void deal(vector<int>& A,int index,int k,int target,int sum,vector<int> cur,vector<vector<int> > &ret)
    {
        if(cur.size()==k&&sum==target)
        {
            ret.push_back(cur);
            return;
        }
        if(index==A.size())
            return;
        if(sum>target)
            return;
        cur.push_back(A[index]);
        deal(A,index+1,k,target,sum+A[index],cur,ret);
        cur.pop_back();
        deal(A,index+1,k,target,sum,cur,ret);
    }
};