基础算法-搜索与回溯：子集和问题(subsum)

admin_ms17_01

于 2024-02-19 14:41:51 发布

阅读量880

点赞数 9

文章标签：算法

本文链接：https://blog.csdn.net/admin_ms17_01/article/details/136169963

版权

前言：

今天我给大家讲一讲子集和问题，通过是在我的”锲而不舍“下，唉............

超时原因只有一个：剪枝不是最优化。

下面我们来看一下：

正文

网址：4249 - 基础算法-搜索与回溯：子集和问题(subsum)

描述：

子集和问题的一个实例为〈S,t〉。其中，S={ x1， x2，…， xn}是一个正整数的集合，c是一个正整数。子集和问题判定是否存在S的一个子集S1，使得子集S1和等于c。对于给定的正整数的集合S={ x1， x2，…， xn}和正整数c，编程计算S 的一个子集S1，使得子集S1和等于c。

输入
第1行有2个正整数n和c，n表示S的个数，c是子集和的目标值。接下来的1 行中，有n个正整数，表示集合S中的元素。

输出
当问题有解时输出满足条件的一个子集，当问题无解时，输出“No Solution!”。

样例
输入
5 10
2 2 6 5 4
输出
2 2 6
提示
S中的元素个数不超过10000，C的值int范围

分析：

这道题要用两个数组，一个储存数据，一个来打标记。选择过程中，要注意过程优化，就成了。

好，上代码：

#include<bits/stdc++.h>
using namespace std;
int data[9999999] ;
bool visit[9999999] ;
int n, c ;
bool choose(int n) {
	int p = 0, sum = 0;
	while (p >= 0) {
		if (!visit[p]) {
			visit[p] = true;
			sum += data[p];
			if (c == sum)
				return true;
			else if ( c < sum) {
				visit[p] = false ;
				sum -= data[p] ;
			}
			p++;
		}
		if (p >= n) {
			while ( visit[p - 1] ) {
				p-- ;
				visit[p] = false ;
				if (p < 1) return false ;
			}
			while ( !visit[p - 1]) {
				p-- ;
				if (p < 1) return false ;
			}
			sum -= data[p - 1] ;
			visit[p - 1] = false ;
		}
	}
	return false ;
}
signed main() {
	scanf("%d%d", &n,&c) ;
	for (int i = 0 ; i < n ; i++)
		scanf("%d", &data[i]) ;
	if (choose(n)) {
		int first = 1 ;
		for (int i = 0 ; i < n; i++)
			if (visit[i]) {
				if (first)
					first = 0 ;
				else
					printf(" ") ;
				printf("%d",data[i]) ;
			}
		printf("\n") ;
	} else
		printf("No Solution!\n") ;
}

OK，提交一下，AC了。