原来城市中有n个建筑,每一个建筑有一个权值表示它的种类。现在,建筑全被风沙埋掉了,人们只从中挖出了K个建筑。告诉你哪些种类的建筑被挖出来了至少一个。人们想知道,他们挖出来的建筑究竟是n个建筑中的哪几个,问可能的方案有几个?(n<=50)
举个例子:原来的建筑为1 2 2 1,人们挖出了2个建筑,其中1、2两种种类的建筑都被挖出来至少一个,那么可能挖出来方案有4个:第一个1和第一个2,第一个1和第二个2,第二个1和第一个2,第二个1和第二个2。
思路
<1>该题满足dp的性质即用dp[i][j]表示搜到第i个已经有了j个建筑物的方案数
<2>dp[i][j]=dp[i-1][j-t]*c[has[i-1]][t];//从i-1中有的元素中挑出
ll c[55][55],dp[55][55];
int cnt[55],has[55],n,m;
void made(){
for(int i=0;i<=50;i++)
for(int j=0;j<=i;j++){
if(j==i||j==0)c[i][j]=1;
else c[i][j]=c[i-1][j-1]+c[i-1][j];
}
}
ll count(vector<int>A,vector<int>B,int k){
made();
int n=A.size(),m=B.size();
for(int i=0;i<n;i++)cnt[A[i]]++;
for(int i=1;i<=m;i++)has[i]=cnt[B[i-1]];
dp[0][0]=1;
for(int i=1;i<=m;i++)
for(int j=k;j>=0;j--)
for(int t=1;t<=min(has[i],j);t++){
dp[i][j]+=c[has[i]][t]*dp[i-1][j-t];
}return dp[m][k];
}