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