깊은 수색 트리

정수 a1, a2,.......an, 그 중에서 몇 개의 수를 골라서 그것들의 합을 K로 할 수 있는지 판단한다.
입력:
n=4
a={1,2,4,7}
k=13
출력:
Yes(13=2+4+7)
클래식 심층 검색 문제: 귀속 코드:
#include <cstdio>
int a[30];int n,k;
bool dfs(int i,int sum)  // i sum
{
    if(i==n)
        return sum==k;
    if(dfs(i+1,sum))  // i 
        return true;
    if(dfs(i+1,sum+a[i]))  // i 
        return true;
}
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        if(dfs(0,0))
            printf("YES
"); else printf("NO
"); } return 0; }

추가된 각 수를 저장해야 하는 경우 다음을 수행할 수 있습니다.
#include <iostream>
#include <vector>
using namespace std;
int a[4]={1,2,4,7},k;
vector<int> ans;
void dfs(int i,int sum)
{
	if(i==4){
		if(sum==k)
	   	{
   			for(int i=0;i!=ans.size();i++)cout<<ans[i]<<" ";
			cout<<endl;
			return ;
   		}
   		else 
		 return ;
	}
	dfs(i+1,sum);
	ans.push_back(a[i]);
	dfs(i+1,sum+a[i]);
	ans.pop_back();
	return ;
}
int main()
{
	 k=7;
	 dfs(0,0);
}

좋은 웹페이지 즐겨찾기