算法训练 和为T
时间限制:1.0s 内存限制:256.0MB
问题描述
从一个大小为n的整数集中选取一些元素,使得它们的和等于给定的值T。每个元素限选一次,不能一个都不选。
输入格式
第一行一个正整数n,表示整数集内元素的个数。
第二行n个整数,用空格隔开。
第三行一个整数T,表示要达到的和。
第二行n个整数,用空格隔开。
第三行一个整数T,表示要达到的和。
输出格式
输出有若干行,每行输出一组解,即所选取的数字,按照输入中的顺序排列。
若有多组解,优先输出不包含第n个整数的;若都包含或都不包含,优先输出不包含第n-1个整数的,依次类推。
最后一行输出总方案数。
若有多组解,优先输出不包含第n个整数的;若都包含或都不包含,优先输出不包含第n-1个整数的,依次类推。
最后一行输出总方案数。
样例输入
5
-7 -3 -2 5 9
0
-7 -3 -2 5 9
0
样例输出
-3 -2 5
-7 -2 9
2
-7 -2 9
2
数据规模和约定
1<=n<=22
T<=maxlongint
集合中任意元素的和都不超过long的范围
T<=maxlongint
集合中任意元素的和都不超过long的范围
#include <cstdio>
#include <vector>
#include <algorithm>
#define MAX 22
using namespace std;
int a[MAX];
int visit[MAX];
int N, T;
typedef struct {
int v[MAX];
vector<int> path;
} Result;
vector<int> temp;
vector<Result> result;
int cmp( Result a, Result b ) {
for( int i = MAX - 1; i >= 0; i-- ) {
if( a.v[i] != b.v[i] ) {
return a.v[i] < b.v[i] ? 1 : 0;
}
}
}
void dfs( int curIndex, int curSum ) {
for( int i = curIndex; i < N; i++ ) {
temp.push_back( a[i] );
visit[i] = 1;
//printf( "%d入栈\n", a[i] );
dfs( i + 1, curSum + a[i] );
temp.pop_back();
visit[i] = 0;
//printf( "%d出栈\n", a[i] );
}
if( curSum == T && temp.size() != 0 ) {
Result r;
r.path = temp;
for( int i = 0; i < MAX; i++ ) {
r.v[i] = visit[i];
}
result.push_back( r );
}
}
int main() {
//freopen( "123.txt", "r", stdin );
scanf( "%d", &N );
for( int i = 0; i < N; i++ ) scanf( "%d", &a[i] );
scanf( "%d", &T );
dfs( 0, 0 );
sort( result.begin(), result.end(), cmp );
for( int i = 0; i < result.size(); i++ ) {
for( int j = 0; j < result[i].path.size(); j++ ) {
printf( "%d ", result[i].path[j] );
}
printf( "\n" );
}
printf( "%d", result.size() );
return 0;
}