寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #DPL_5_I : Balls and Boxes 9

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "Balls and Boxes 9"
https://onlinejudge.u-aizu.ac.jp/problems/DPL_5_I
・ボールと箱9

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #DPL_5_I : Balls and Boxes 9
/*
Aizu Online Judge in C++ #DPL_5_I : Balls and Boxes 9
https://onlinejudge.u-aizu.ac.jp/problems/DPL_5_I
 提出コードの解答例
 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=0;j<=k;j++){
            dp[i+1][j]+=j*dp[i][j];
            dp[i+1][j]%=m;
            dp[i+1][j+1]+=dp[i][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/