原题:
试题4 :篮球队(共10个测试点,每个点2分)
源代码:team.c / team.cpp / team.pas / team.bas
输入文件:team.in
输出文件:team.out
【问题描述】
小红的班级里有n名爱打篮球的同学。已知这n名同学的身高。现在体育老师要从他们当中选择m名同学组成一支篮球队,老师希望蓝球队的平均身高不小于h。
问:一共有多少种不同选择方法使得m名篮球队员的平均身高不小于h?
【输入文件】
文件的第一行是三个整数n, m,h(1≤m≤n≤30, 100≤h≤200)。
第二行是n个用空格分开的整数a1,……,an,依次表示n名同学的身高。ai是不超过200的正整数。
这题原题数据n,m<=30,枚举组合爆搜就可以AC。
但cjr神犇问我,如果数据再大呢,搜索肯定就T了。假如N<=500,m<=100呢。这类题吧。。看多了我们就知道。。我们可以考虑DP。
不同选择方案是求方案数,肯定和max,min没关系。有n名同学选m个。我们可以想到前i个中选j个。然后二维费用维护几个人和身高。f i j k 表示前i个人中选了j个 身高和是k的方案数。
但是这样无疑会MLE的。所以滚动数组优化压掉一维。
#include<bits/stdc++.h>
using namespace std;
/*f[0][0][0]=1;
f[1][0][0]=1;
f[n][0][0]=1;
f[i][j][k]+=f[i-1][j-1][k-k[i]]
const int MAXN=2e5+5;
int n,m,h,w[MAXN];
int f[1000][MAXN];
int sum=0;
int main(){
// freopen("1.in","r",stdin);
// freopen("1.out","w",stdout);
memset(f,0,sizeof(f));
scanf("%d%d%d",&n,&m,&h);
for(int i=1;i<=n;i++){
scanf("%d",&w[i]);
sum+=w[i];
}
f[0][0]=1;
for(int i=1;i<=n;i++)
for(int j=m;j>=1;j--){
for(int k=w[i];k<=sum;k++){
f[j][k]+=f[j-1][k-w[i]];
}
}
int ans=0;
for(int i=m*h;i<=sum;i++)ans+=f[m][i];
printf("%d",ans);
return 0;
}
/*
6 3 153
158 155 152 144 149 161
*/