沈师PTA7-1 子集和

//讲真，要不是明天要考我真不想写pta

//本题和昨天类似，但今天是回溯+剪枝

问题描述

设集合S={x1,x2,…,xn}是一个正整数集合，c是一个正整数，子集和问题判定是否存在S的一个子集S1，使S1中的元素之和为c。试设计一个解子集和问题的回溯法，并输出利用回溯法在搜索树（按输入顺序建立）中找到的第一个解。

输入格式:

输入数据第1行有2个正整数n和c，n表示S的大小，c是子集和的目标值。接下来的1行中，有n个正整数，表示集合S中的元素。
是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。

输出格式:

输出利用回溯法找到的第一个解，以空格分隔，最后一个输出的后面有空格。当问题无解时，输出“No Solution!”。

输入样例:

在这里给出一组输入。例如：

5 10
2 2 6 5 4

输出样例:

在这里给出相应的输出。例如：

2 2 6 

先看张图

 

 

#include<iostream>
#include<vector>
using namespace std;
const int N=10000;
int path[N];//目标数组，从里面选target
int visit[N];//与目标数组相互映射，用于表示我们是否选中了目标数组中的某个数
int n,target,sum,sub_sum;
/*
 * 主要讲下sum和sub_sum
 * sum表示我们当前选的总和，如果sum>target就不能选了，再选也不是target，用于左剪枝
 * sub_sum表示余下的总和 如果 sub_sum+sum<target 再选也不够达到target 用于右剪枝
 */
vector<int> ans;
void dfs(int u);
int main(){
    scanf("%d%d",&n,&target);
    for(int i=0;i<n;i++){
        scanf("%d",&path[i]);
        sub_sum+=target;
    }
    dfs(0);
    printf("No Solution");
    return 0;
}
void dfs(int u){
    if(sum>target) return ;
    else if(sum==target){
        for(int i=0;i<ans.size();i++) printf("%d ",ans[i]);
        exit(0);
    }
    if(sum+sub_sum<target) return ;
    for(int i=u;i<n;i++){//从当前的位置u开始选
        visit[i]=true;
        sum+=path[i];
        sub_sum-=path[i];
        ans.push_back(path[i]);//这里要加一个ans，因为path是目标数组，解向量要另外存在一个数组里面
        dfs(i+1);
        ans.pop_back();
        visit[i]=false;
        sum-=path[i];
        sub_sum+=path[i];
    }
}

最后一个点过不去，以后再看吧。