Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。
・問題 "Balls and Boxes 12"
https://onlinejudge.u-aizu.ac.jp/problems/DPL_5_L
・ボールと箱 12
僕が作成、提出したコードは、以下のとおりです。
Aizu Online Judge in C++ #DPL_5_L : Balls and Boxes 12
/* Aizu Online Judge in C++ #DPL_5_L : Balls and Boxes 12 https://onlinejudge.u-aizu.ac.jp/problems/DPL_5_L 提出コードの解答例 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<<(n<k ? 0 : dp[n-k][k])<<endl; return 0; }
設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/