PTA7-3 子集和问题 (30 分) 回溯法

Posted 2021-12-18 karshey

输入：

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的原因是：数组开小了。
我大裂开，这应该显示段错误啊！
老师你欠我的用什么还！！！！！