題目大意:給出 n 種不同種類的石子,每種石子的個數(shù)是 a[ i ] 個,記 sum 為石子的總數(shù),問用 k ∈ [ 1 ,? sum ] 個石子組成的不同排列組合有多少種
題目分析:考慮 dp,dp[ i ][ j ] 代表的是到了第 i 堆為止,已經(jīng)選了 j 個石子的答案,那么轉(zhuǎn)移方程就是,dp[ i ][ j ] += dp[ i - 1 ][ j - k ] * C[ j ][ k ] ,理解起來也比較簡單,k 的意思是從第 i 堆選擇了 k 個石子,當從第 i 堆選擇了 k 個石子后,總共就選擇了 j 個石子,原本的 j - k 個石子的相對位置不變,新加入的石子需要從 j 個位置中選擇 k 個位置來放置,因為這 k 個石子都是種類一樣的,所以只需要選擇出位置就可以了
代碼: ?
//#pragma GCC optimize(2)
//#pragma GCC optimize("Ofast","inline","-ffast-math")
//#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
#include<bitset>
#include<unordered_map>
using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=110;const int mod=1e9+7;LL C[N*N][N],dp[N][N*N];int a[N];void init()//預處理組合數(shù)
{C[0][0]=1;for(int i=1;i<N*N;i++){C[i][0]=1;for(int j=1;j<=min(i,100);j++)C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;}
}int main()
{
#ifndef ONLINE_JUDGE
// freopen("data.in.txt","r",stdin);
// freopen("data.out.txt","w",stdout);
#endif
// ios::sync_with_stdio(false);init();int n;int kase=0;while(scanf("%d",&n)!=EOF){memset(dp,0,sizeof(dp));int sum=0;for(int i=1;i<=n;i++){scanf("%d",a+i);sum+=a[i]; }dp[0][0]=1;for(int i=1;i<=n;i++)for(int k=0;k<=a[i];k++)//第i輪選了k個石子 for(int j=k;j<=sum;j++)dp[i][j]=(dp[i][j]+dp[i-1][j-k]*C[j][k])%mod;LL ans=0;for(int i=1;i<=sum;i++)ans=(ans+dp[n][i])%mod;printf("Case %d: %lld\n",++kase,ans);}return 0;
}