部分和为特定的值

来源:互联网 发布:java常见异常类型 编辑:程序博客网 时间:2024/05/21 06:47

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


  1. #include 
  2. using namespace std;
  3.  
  4.  int n, k;
  5.  int a[25], b[25];
  6.  
  7.  bool dfs(int i, int sum)   //已经从前i项得到了sum,然后对于i项之后的进行分支
  8.  {
  9.    if(i==n)                               //如果前i项都计算过了,则返回sum是否与k相等
  10.                        return  sum=k;    
  11.                                b[i]=0;
  12.    if(dfs(i+1, sum))                       //不加上a[i]的情况
  13.                         return true;      
  14.                                b[i]=1;
  15.    if(dfs(i+1, sum+a[i]))                  //加上a[i]的情况
  16.                          return true;
  17.  
  18.     return false;
  19.   }
  20.  
  21. int main()
  22. {
  23.     while(cin>>n>>k)
  24.     {
  25.        for(int i=0; i<n; i++)
  26.             cin>>a[i];
  27.   
  28.         if(dfs(0, 0))
  29.          {
  30.             cout<<"YES"<<endl;

  31.             for(int i=0; i<n; i++)
  32.              {
  33.                if(b[i])
  34.                 cout<<a[i]<<" ";
  35.               }
  36.               cout<<endl;
  37.           }
  38.         else
  39.              cout<<"NO"<<endl;
  40.    }
  41.    return 0;
  42. }





0 0
原创粉丝点击