noj 1005 B 装载问题 (dfs暴力)

Problem B

装载问题

时限：1000ms 内存限制：10000K 总时限：3000ms

描述：

 

有两艘船，载重量分别是c1、 c2，n个集装箱，重量是wi (i=1…n)，且所有集装箱的总重量不超过c1+c2。确定是否有可能将所有集装箱全部装入两艘船。

输入：

 

多个测例，每个测例的输入占两行。第一行一次是c1、c2和n（n<=10）；第二行n个整数表示wi (i=1…n)。n等于0标志输入结束。

输出：

 

对于每个测例在单独的一行内输出Yes或No。

输入样例：

 

7 8 2
8 7
7 9 2
8 8
0 0 0

输出样例：

 

Yes
No

来源：

 

#include <iostream>
using namespace std;

int c1, c2, n;
int w[10];

bool flag;

void dfs(int k, int w1, int w2) 
{
	if(k == n)
	{
		if(w1 <= c1 && w2 <= c2)
		{
			flag = true;
		}
		return;
	}

	//分给船1(左子树)
	dfs(k+1, w1 + w[k], w2);
	//分给船2(右子树)
	dfs(k+1, w1, w2 + w[k]);
}


int main(int argc, char const *argv[])
{
	while(cin >> c1 >> c2 >> n && n!=0)
	{
		for (int i = 0; i < n; ++i)
		{
			cin >> w[i];
		}	

		flag = false;
		dfs(0,0,0);
		if(flag)
		{
			cout<<"Yes"<<endl;
		}
		else
		{
			cout<<"No"<<endl;
		}
	}
	return 0;
}