hdu 1258 Sum it up dfs+剪枝

点击打开链接

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int M =20;
int a[M],s,n,res[M],visit[M],flag;
bool check(int x,int y)
{
	for(int i=x;i<y;i++)
	{
		if(a[i]==a[y])
		return 1;
	}
	return 0;
}
void dfs(int cur,int k,int last) // cur 当前和 , k为个数  last为上次选的 
{
	if(cur==s)
	{
		flag=1;
		for(int i=1;i<=k;i++)
		{
			if(i==k)
			cout<<res[i]<<endl;
			else
			cout<<res[i]<<"+";
		
		}
		return;
		
		
	}							// 4 3 2 2 1 1 s=4 n=6  
	for(int i=last+1;i<=n;i++) // 从last+1开始搜 保证 只出现3+1 
	{
		if(!visit[i])
		{
			if(a[i]+cur<=s)
			{		// 4 3 2 2 1 1 s=4 n=6  保证 3+1 只出现一次 
				if(check(last+1,i)) //  以a[i]"开头"的在这一层已经搜索过一遍时 cut 
				continue;
				else 
				{
					res[++k]=a[i];
					visit[i]=1; 
					dfs(a[i]+cur,k,i);
					visit[i]=0; 
					--k;//回溯 
				}
			}
		}
	}
}
int main()
{
	
	while(cin>>s>>n&&(s+n))
	{
		flag=0;
		memset(visit,0,sizeof(visit));
		for(int i=1;i<=n;i++)
		{
			cin>>a[i];
		}
		printf("Sums of %d:\n",s);
		dfs(0,0,0);
		if(!flag)
		{
			cout<<"NONE"<<endl;
		}
	} 
	
	return 0;
}