nyoj ACM:部分和問題(DFS 回溯 遞歸)

部分和問題
時間限制：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 ，遞歸 的解釋可以參考我另一篇博客素數環

我的代碼：

#include<iostream>
#include<cstdio>
#include<cstring>

int n,k;
int a[30];
bool visited[30];
int sum=0;
int flag=0;
void DFS(int i)//從i開始選，但是i還沒有被選 
{
    //遞歸形式的第一部分：遞歸在簡單情境下的退出
    if(i>n || flag==1)
        return;

    //遞歸形式的第二部分：由於此時還不是簡單情形，那麼就要繼續將改問題減少成更小的子問題 
    for(int j=i;j<=n;j++)
    {
        visited[j]=true;
         if(k==sum+a[j] && flag==0 )
        {
            flag=1;
            visited[j]=true;
            printf("YES\n");
            for(int p=1;p<=n;p++)
            {
                if(visited[p]==true)
                    printf("%d ",a[p]);
            }
            printf("\n");
            return;
        }
        else if(k!=sum+a[j] && flag==0 )
        {
            sum+=a[j];
            visited[j+1]=true;
            DFS(j+1);
            sum-=a[j];
            visited[j+1]=false;
        }
        visited[j]=false;
    }


    return; 
}

int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",a+i);
        }

        memset(visited,false,sizeof(visited));
        sum=0;
        flag=0;
        DFS(1);
        if(flag==0)
            printf("NO\n");
    }

    return 0;
}