上海市计算机学会竞赛平台2024年7月月赛丙组子集归零

题目描述

给定 nn 个数字 a1,a2,,…,ana1​,a2​,,…,an​，请统计能从 11 到 nn 中，选出多少种不同的下标子集，使得这些下标对应的数字之和等于 00。

注意空集与全集也是子集中的一种。

输入格式

• 第一行，单个整数表示 nn
• 第二行，nn 个整数表示 a1,a2,…,ana1​,a2​,…,an​

输出格式

• 单个整数：表示归零子集的数量。

数据范围

• 对于 30%30% 的数据，1≤n≤51≤n≤5
• 对于 60%60% 的数据，1≤n≤101≤n≤10
• 对于 100%100% 的数据，1≤n≤221≤n≤22
• 对于 100%100% 的数据，−1,000,000≤ai≤1,000,000−1,000,000≤ai​≤1,000,000

样例数据

输入:

4
2 -1 -2 1

输出:

4

说明:

{}
{1 -1}
{2 -2}
{1 2 -1 -2}

详见代码:

#include<bits/stdc++.h>
using namespace std;
int n;
int a[25];
int ans = 0;
void dfs(int step, int sum)
{
    if(step > n)
    {
        if (sum == 0) ans++;
        return;
    }
    dfs(step + 1, sum); 
    dfs(step + 1, sum + a[step]); 
}
int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
    {
        cin >> a[i];
    }
    dfs(1, 0);
    cout << ans;
    return 0;
}