算法训练 和为T

算法训练 和为T  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　从一个大小为n的整数集中选取一些元素，使得它们的和等于给定的值T。每个元素限选一次，不能一个都不选。

输入格式

　　第一行一个正整数n，表示整数集内元素的个数。
　　第二行n个整数，用空格隔开。
　　第三行一个整数T，表示要达到的和。

输出格式

　　输出有若干行，每行输出一组解，即所选取的数字，按照输入中的顺序排列。
　　若有多组解，优先输出不包含第n个整数的；若都包含或都不包含，优先输出不包含第n-1个整数的，依次类推。
　　最后一行输出总方案数。

样例输入

5
-7 -3 -2 5 9
0

样例输出

-3 -2 5
-7 -2 9
2

数据规模和约定

　　1<=n<=22
　　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;
}