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

最新推荐文章于 2024-05-19 10:30:55 发布

karshey

最新推荐文章于 2024-05-19 10:30:55 发布

阅读量2.6k

点赞数 3

分类专栏：我的ACM之路文章标签：回溯法算法 C++

本文链接：https://blog.csdn.net/karshey/article/details/121479586

版权

我的ACM之路专栏收录该内容

159 篇文章

订阅专栏

这篇博客主要讨论了一种使用回溯法解决数组元素组合问题的算法。作者提供了两种不同的实现，代码1和代码2，分别展示了如何在不超过给定总数的情况下，找到能够组成特定和的数组元素子集。在解决过程中，作者遇到了因数组大小不足导致的错误，并分享了解决此问题的经验。博客还提到这个问题至少被提交了100次，强调了调试和测试的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述
输入：

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;
}