c++制作简单支票计算脚本(回溯+剪枝)

前些天我的亲戚找到我,说找我给他做一个简单支票计算脚本,要求如下:
输入不同数量的支票,每张支票只能用一次,给定一个目标数,求出用已给支票组合出目标数的所有组合方式,如果无解,给出可解的低于该目标数的最大数字的所有解。
例如:
在这里插入图片描述
也就是求数字的所有组合问题。
思路是回溯和剪枝。
算法如下:

class Solution {
private:
    vector<int> candidates, path;
    vector<vector<int> > ans;
public:
    bool dfs(int start, int target) {
        if (target == 0) {
            ans.push_back(path);
            return 1;
        }
        for (int i = start; i < candidates.size() && target - candidates[i] >= 0; ++i)
        {
            if (i > start&& candidates[i] == candidates[i - 1])//剪枝
                continue;
            path.push_back(candidates[i]);
            dfs(i + 1, target - candidates[i]);
            path.pop_back();
        }
        return 0;
    }

    vector<vector<int> > combinationSum2(vector<int>& candidates, int target) {
        ans={};
        sort(candidates.begin(), candidates.end());
        this->candidates = candidates;
        dfs(0, target);
        return ans;
    }
};

求解时记得每次要初始化ans数组,不然会出现赘余导致问题。
完整代码如下:

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
private:
    vector<int> candidates, path;
    vector<vector<int> > ans;
public:
    bool dfs(int start, int target) {
        if (target == 0) {
            ans.push_back(path);
            return 1;
        }
        for (int i = start; i < candidates.size() && target - candidates[i] >= 0; ++i)
        {
            if (i > start&& candidates[i] == candidates[i - 1])
                continue;
            path.push_back(candidates[i]);
            dfs(i + 1, target - candidates[i]);
            path.pop_back();
        }
        return 0;
    }

    vector<vector<int> > combinationSum2(vector<int>& candidates, int target) {
        ans={};
        sort(candidates.begin(), candidates.end());
        this->candidates = candidates;
        dfs(0, target);
        return ans;
    }
};
int main()
{
    cout<<endl;
    cout<<"||——————————————————————欢迎使用支票计算程序———————————————————————||"<<endl;
    cout<<endl;
    int flag_u=1;
    while (1) {
        cout<<"=============>开始处理第"<<flag_u<<"组数据"<<endl;
        int c_num = 0;
        cout << "请输入支票的个数(仅一组支票的数量,多组支票请分开处理):" << endl;
        cin >> c_num;
        Solution solution = Solution();
        cout << "请依次输入每张支票的面额:" << endl;
        vector<int> candidates;
        for (int i = 0; i < c_num; ++i) {
            int ci = 0; cin >> ci;
            candidates.push_back(ci);
        }
        cout << "请输入目标数字的个数:" << endl;
        int num_ans=0;
        cin>>num_ans;
        while(num_ans>0){
        cout << "请输入目标数字:" << endl;
        int target = 0; cin >> target;
        vector<vector<int> > res= solution.combinationSum2(candidates, target);
        if (res.size() != 0) {
            cout << "目标数字:" << target << "的所有解法为:" << endl;
            for (int i = 0; i < res.size(); ++i) {
                for (int j = 0; j < res[i].size(); ++j) {
                    if (j < res[i].size() - 1) cout << res[i][j] << " + ";
                    else cout << res[i][j] << " = " << target;
                }
                cout << "" << endl;
            }            cout << endl;
        }
        else {
            if (target > candidates.back()) {
                int target_down = target - 1; Solution solution_down = Solution();
                vector<vector<int> > res_down= solution_down.combinationSum2(candidates, target_down);
                while (res_down.size() == 0) {
                    target_down--; res_down = solution_down.combinationSum2(candidates, target_down);
                }
                cout << "目标数字无解!低于且最接近目标数的可解整数 " << target_down << " 的所有解法为:" << endl;
                for (int i = 0; i < res_down.size(); ++i) {
                    for (int j = 0; j < res_down[i].size(); ++j) {
                        if (j < res_down[i].size() - 1) cout << res_down[i][j] << " + ";
                        else cout << res_down[i][j] << " = " << target_down;
                    }
                    cout << "" << endl;
                }
                                int target_up = target + 1; Solution solution_up = Solution();
                /*vector<vector<int> > res_up = solution_up.combinationSum2(candidates, target_up);
                while (res_up.size() == 0&&target_up<target+candidates.back()) {
                    target_up++; res_up = solution_up.combinationSum2(candidates, target_up);
                }
                if(res_up.size()!=0){
                cout << "目标数字无解!高于且最接近目标数的可解整数 " << target_up << " 的所有解法为:" << endl;
                for (int i = 0; i < res_up.size(); ++i) {
                    for (int j = 0; j < res_up[i].size(); ++j) {
                        if (j < res_up[i].size() - 1) cout << res_up[i][j] << " + ";
                        else cout << res_up[i][j] << " = " << target_up;
                    }
                    cout << "" << endl;
                }
                }*/

            }
            cout << endl;
        }
        num_ans--;
        }

    cout<<"=============>第"<<flag_u<<"组数据处理完毕!"<<endl;
    cout << endl;
    flag_u++;
    }
    return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值