Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。
・問題 "Balls and Boxes 3"
https://onlinejudge.u-aizu.ac.jp/problems/DPL_5_C
・ボールと箱3
僕が作成、提出したコードは、以下のとおりです。
Aizu Online Judge in C++ #DPL_5_C : Balls and Boxes 3
/* Aizu Online Judge in C++ #DPL_5_C : Balls and Boxes 3 https://onlinejudge.u-aizu.ac.jp/problems/DPL_5_C 提出コードの解答例 https://neguse-atama.hatenablog.com */ #include<bits/stdc++.h> using namespace std; using ll=long long; int main(void){ ll n,k,m=1e9+7; ll dp[1010][1010]={}; cin>>n>>k; dp[0][0]=1; for(ll i=0;i<n;i++){ for(ll j=0;j<=k;j++){ dp[i+1][j]+=dp[i][j]*j; dp[i+1][j]%=m; dp[i+1][j+1]+=dp[i][j]*(k-j); dp[i+1][j+1]%=m; } } cout<<dp[n][k]<<endl; return 0; }
設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/