蓝桥杯 ALGO-115 算法训练 和为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 <stdio.h>

int n;
long long int a[25];
long long int T;
long long int current_sum;
long long int chosen[25];
int size_chosen;
long long int num_solutions;

void search(int idx)
{
    if (idx < 0)
    {
        if (current_sum == T && size_chosen)
        {
            num_solutions++;
            for (int i = size_chosen - 1; i >= 0; --i)
                printf("%lld ", chosen[i]);
            printf("\n");
        }
        return;
    }

    search(idx - 1);
    current_sum += a[idx];
    chosen[size_chosen++] = a[idx];
    search(idx - 1);
    size_chosen--;
    current_sum -= a[idx];
}

int main()
{
    scanf("%d", &n);
    for (int i = 0; i < n; ++i)
        scanf("%lld", &a[i]);
    scanf("%lld", &T);

    current_sum = 0;
    size_chosen = 0;
    num_solutions = 0;
    search(n-1);
    printf("%lld", num_solutions);

    return 0;
}