Combination Sum


class Solution {
public:
    vector<vector<int> > combinationSum(vector<int> &candidates, int target) {
        sort(candidates.begin(), candidates.end());
        vector<vector<int> > result;
        vector<int> path;
        int sum = 0;
        dfs(candidates, 0, sum, path, result, target);
        return result;
    }
    void dfs(vector<int> &candidates, int pos, int &sum, vector<int> &path, vector<vector<int> > &result, int target)
    {
        if(sum == target)
        {
            result.push_back(path);
            return;
        }
        else if(sum > target)
            return;
        else
        {
            for(int i=pos; i<candidates.size(); ++i)
            {
                path.push_back(candidates[i]);
                sum += candidates[i];
                dfs(candidates, i, sum, path, result, target);
                sum -= candidates[i];
                path.pop_back();
            }
        }
    }
};

9.06非递归实现: 参考 http://blog.csdn.net/starmsg/article/details/39085143

注意:如果每次可以push相同的元素入栈,则无法通过 

s.peek() == currentPath.peek()

区分是否有新的元素push到stack中,此时可以在path中记录stack top的index而非实际的元素,来标记是否有新的元素push。

新的判断条件变成了 

<span style="color:#333333;">(stack.size()-1) == path.back()</span>

class Solution {
public:
    vector<vector<int> > combinationSum(vector<int> &candidates, int target) {
        vector<vector<int> > result;
        if(candidates.size() == 0)
            return result;
        sort(candidates.begin(), candidates.end());
        
        for(int start = 0; start < candidates.size(); start++)
            dfsWithStack(candidates, result, start, target);
        return result;
    }
    
    void dfsWithStack(vector<int> &candidates, vector<vector<int> > &result, int startIdx, int target)
    {
        vector<int> path;
        vector<int> stack;
        int sum = 0;
        if(sum + candidates[startIdx] <= target)
        {
            stack.push_back(startIdx);
        }
    
        int count = 0;
    
        while(!stack.empty())
        {
            int idx = stack.back();
            path.push_back(stack.size()-1);
            sum += candidates[idx];
    
            if(sum == target)
            {
            	vector<int> res;
            	for(int i=0; i<path.size(); i++)
            		res.push_back(candidates[stack[path[i]]]);
                result.push_back(res);
            }
            else if(sum < target)
            {
                for(size_t i = idx; i < candidates.size(); i++)
                {
                    if(sum + candidates[i] <= target)
                    {
                        stack.push_back(i);
                    }
                }
            }
    
            while(!stack.empty() && (stack.size()-1) == path.back())
            {
                sum -= candidates[stack.back()];
                path.pop_back();
                stack.pop_back();
            }
            count++;
        }
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值