深搜:给定整数a1、a2、...an,判断是否可以从中选出若干数,使它们的和恰好为K。

部分和问题

时间限制: 1000 ms  |  内存限制: 65535 KB
难度: 2
描述
给定整数a1、a2、.......an,判断是否可以从中选出若干数,使它们的和恰好为K。
输入
首先,n和k,n表示数的个数,k表示数的和。
接着一行n个数。
(1<=n<=20,保证不超int范围)
输出
如果和恰好可以为k,输出“YES”,并按输入顺序依次输出是由哪几个数的和组成,否则“NO”
样例输入
4 13
1 2 4 7
样例输出
YES
2 4 7


很明显,这是一道简单的dfs搜索,直接理由dfs的定义解决,不过最让人头痛的是如何实现 剪枝,减少不必要的时间浪费,剪枝还不熟练,这道题需要减掉的是

1.从当前状态如何转移都不会存在解

2.当sum超过k时,也没必要继续搜索


#include <bits/stdc++.h>
using namespace std;
//int a[MAX_N];
int n,k,a[22],b[22];
bool dfs(int i,int sum){
    if(sum>k)return false;
    if(i==n)return sum==k;
    //not add a[i]
    if(dfs(i+1,sum)){
            b[i]=0;
        return true;
    }
    if(dfs(i+1,sum+a[i])){
        b[i]=1;
        return true;
    }
    return false;
}
int main(){
    cin>>n;
    for(int i=0;i<n;i++)cin>>a[i];
    cin>>k;
    if(dfs(0,0)){
        cout<<"Yes"<<endl;

    for(int i=0;i<n;i++)
    {
        if(b[i])cout<<a[i]<<" ";

    }
    cout<<endl;
    }
}


猜你喜欢

转载自blog.csdn.net/qq_36294146/article/details/79882699