【PTA】7-3 子集和问题 (30 分) 回溯法

输入：

5 10
2 2 6 5 4

输出：

2 2 6 

代码1：dfs，d就完事了。

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
int n;
int c;
int a[20000];
vector<int>ans;
int flag=0;
void dfs(int u,int rest,int sumrest)//当前第几个 离目标差几 还剩的机会 
{
	if(rest==0)
	{
		flag=1;
		for(auto x:ans) cout<<x<<" ";
		return;
	}
	else
	{
		if(u<=n&&!flag)
		{
			//拿
			if(rest-a[u]>=0)
			{
				ans.pb(a[u]);
				dfs(u+1,rest-a[u],sumrest-a[u]);
				ans.pop_back();
			} 
			//不拿 
			if(rest>=0&&sumrest>=rest) dfs(u+1,rest,sumrest-a[u]); 
		}
	}
}
int main()
{
	cin>>n>>c;
	int summ=0;
	for(int i=1;i<=n;i++) 
	{
		cin>>a[i];
		summ+=a[i];
	}
	if(summ<c) 
	{
		cout<<"No Solution!";
		return 0;
	}
	dfs(1,c,summ);
	if(flag==0) cout<<"No Solution!";
	return 0;
}

代码2：（z同学的代码，比我的规范多了）

#include <iostream>
using namespace std;
int n,v,a[1000],ans[1000],cv;
bool flag;

void Backsearch(int t) {
	if(cv==0) flag=1;
	if(t>n||flag==1) return;
	if(flag==0&&a[t]<=cv) {
		cv-=a[t];
		ans[t]=a[t];
		Backsearch(t+1);
		if(flag==0) cv+=a[t];
		if(flag==0) ans[t]=0;
	}
	if(flag==0)
		Backsearch(t+1);
}

int main() {
	cin>>n>>v;
	for(int i=1; i<=n; i++) {
		cin>>a[i];
	}
	flag=0;
	cv=v;
	Backsearch(1);
	if(flag==1) {
		for(int i=1; i<=n; i++) {
			if(ans[i]!=0)
				cout<<ans[i]<<" ";
		}
	} else
		cout<<"No Solution!";
	return 0;
}

ps：这道题提交了起码100发，后来发现第五个测试点WA的原因是：数组开小了。
我大裂开，这应该显示段错误啊！
老师你欠我的用什么还！！！！！