#背包#AT2037 [ARC060A] 高橋君とカード / Tak and Cards

题目

有一个长度为\(n\)的数组\(a\)，选择若干个数使它们的平均数为\(A\)，问共有多少种方案

---

分析

设\(dp[i][j]\)表示选择\(i\)个数总和为\(j\)的方案数，那么答案就是\(\sum dp[i][A*i]\)

---

代码

#include <cstdio>
#define rr register
using namespace std;
int n,A,x; long long ans,dp[51][2511];
signed main(){
	scanf("%d%d",&n,&A),dp[0][0]=1;
	for (rr int i=1;i<=n;++i){
		scanf("%d",&x);
		for (rr int j=i;j;--j)
		for (rr int m=A*n;m>=x;--m)
		    dp[j][m]+=dp[j-1][m-x];
	}
	for (rr int i=1;i<=n;++i) ans+=dp[i][i*A];
	return !printf("%lld\n",ans);
}