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