寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #Volume24 - 2439 : Hakone

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

・問題 "Hakone"
https://onlinejudge.u-aizu.ac.jp/problems/2439
箱根駅伝
僕が作成、提出したコードは、以下のとおりです。

・Aizu Online Judge in C++ #Volume24 - 2439 : Hakone
/*
Aizu Online Judge in C++ #Volume24 - 2439 : Hakone
 https://onlinejudge.u-aizu.ac.jp/problems/2439
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
ll n,dp[222][222];char c[222];
int main(void){
    cin>>n;
    for(ll i=1;i<=n;i++){
        cin>>c[i];
    }
    dp[0][0]=1;
    for(ll i=1;i<=n;i++){
        for(ll j=0;j<=n;j++){
            if(c[i]=='U'){
                dp[i][j]=dp[i-1][j]*j+(j>=1 ? dp[i-1][j-1] : 0);
            }
            if(c[i]=='D'){
                dp[i][j]=dp[i-1][j+1]*(j+1)*(j+1)+dp[i-1][j]*j;
            }
            if(c[i]=='-'){
                dp[i][j]=dp[i-1][j];
            }
            dp[i][j]%=1000000007;
        }
    }
    cout<<dp[n][0]<<endl;
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/