神奇的口袋-dfs

题目描述

有一个神奇的口袋，总的容积是40，用这个口袋可以变出一些物品，这些物品的总体积必须是40。John现在有n个想要得到的物品，每个物品的体积分别是a1，a2……an。John可以从这些物品中选择一些，如果选出的物体的总体积是40，那么利用这个神奇的口袋，John就可以得到这些物品。现在的问题是，John有多少种不同的选择物品的方式。

输入描述:

输入的第一行是正整数n (1 <= n <= 20)，表示不同的物品的数目。接下来的n行，每行有一个1到40之间的正整数，分别给出a1，a2……an的值。

输出描述:

输出不同的选择物品的方式的数目。

示例1

输入

3
20
20
20

输出

3

#include <iostream>
#include <cstring>
using namespace std;

int count=0;           //不同的方式数目
int n;                 //总数
void dfs(int arr[], int i, bool vis[], int num){
    if(i==n+1){
       if(num==40)
           ++count;
       return ;
    }
    vis[i]=true;
    num += arr[i];
    dfs(arr, i+1, vis, num);
    num -= arr[i];
    vis[i]=false;
    dfs(arr, i+1, vis, num);
}

int main(){
    cin >> n;
    int arr[25];
    bool vis[25];
    memset(vis, false, sizeof(vis));
    for(int i=1; i<=n; ++i){
        cin >> arr[i];
    }
    dfs(arr, 1, vis, 0);
    cout << count << endl;
    return 0;
}