結果

問題 No.93 ペガサス
ユーザー krotonkroton
提出日時 2014-10-28 12:51:48
言語 C++11
(gcc 13.3.0)
結果
AC  
実行時間 37 ms / 5,000 ms
コード長 1,585 bytes
コンパイル時間 876 ms
コンパイル使用メモリ 62,384 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-30 13:51:57
合計ジャッジ時間 1,914 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;

const ll MOD = 1e9 + 7;
ll dp[2][1050][2][2];

void add(ll &a, ll b){
    a = (a + b % MOD) % MOD;
}

ll solve(int n){
    if(n == 1)return 1;
    
    int curr = 0, next = 1;
    memset(dp[curr], 0, sizeof(dp[curr]));
    
    int segs = 3;
    dp[curr][0][0][0] = 2;
    
    for(int i=3;i<=n;i++){
        memset(dp[next], 0, sizeof(dp[next]));
        
        for(int b=0;b<=segs;b++){
            int s = segs - b;
            
            for(int p2=0;p2<=1;p2++)for(int p1=0;p1<=1;p1++){
                int pb = p1 + p2;
                int ps = 3 - pb;
                
                int nb = b - pb;
                int ns = s - ps;
                
                if(nb < 0 || ns < 0)continue;
                
                ll now = dp[curr][b][p2][p1];
                
                add(dp[next][b+1][p1][1], now);
                add(dp[next][b+1-p2][p1][1], now);
                
                add(dp[next][b-p1][0][0], now);
            
                if(nb > 0){
                    add(dp[next][b-1][p1][0], nb * now);
                }
                
                if(ns > 0){
                    add(dp[next][b][p1][0], ns * now);
                }
            }
        }
        
        swap(curr, next);
        ++segs;
    }
    
    return dp[curr][0][0][0];
}


int main(){
    int N;
    cin >> N;
    
    cout << solve(N) << endl;
    return 0;
}
0