洛谷 P1566 加等式

题目传送门

解题思路:

其实本题可以看作在一个01背包里,所有物品可以组成价值的所有可能,找出那些某些物品价值和恰好等于背包中某一个物品的价值的所有方案数,所以跑01背包.

AC代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring> 

using namespace std;

int t,n,a[31],f[1001],ans,sum;
bool vis[31];

int main() {
    scanf("%d",&t);
    for(int i = 1;i <= t; i++) {
        scanf("%d",&n);
        memset(a,0,sizeof(a));
        memset(f,0,sizeof(f));
        ans = 0;
        for(int j = 1;j <= n; j++) {
            scanf("%d",&a[j]);
            sum = max(sum,a[j]);    
        }
        f[0] = 1;
        sort(a+1,a+n+1);
        for(int j = 1;j <= n; j++)
            for(int k = sum;k >= a[j]; k--)
                f[k] += f[k-a[j]];
        for(int j = 1;j <= n; j++)
            ans += f[a[j]];
        printf("%d\n",ans - n);
    }
    return 0;
}