結果

問題 No.1766 Tatsujin Remix
ユーザー first_vilfirst_vil
提出日時 2021-10-12 09:04:20
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 13 ms / 2,000 ms
コード長 1,248 bytes
コンパイル時間 4,125 ms
コンパイル使用メモリ 256,264 KB
最終ジャッジ日時 2025-01-25 00:12:57
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In lambda function:
main.cpp:17:66: warning: control reaches end of non-void function [-Wreturn-type]
   17 |     auto idx = [&](char c) { rep(i,3)if (notes[i] == c)return i; };
      |                                                                  ^

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using mint = atcoder::modint998244353;
#define rep(i,n) for(int i=0;i<n;++i)
int main() {
    int n; cin >> n;
    string s;
    rep(i,n) {
        string t; cin >> t;
        s += t;
    }
    n *= 16;
    s += '.';

    const string notes = ".dk";
    auto idx = [&](char c) { rep(i,3)if (notes[i] == c)return i; };
    vector<vector<mint>> dp(3, vector<mint>(n + 1));
    dp[0][0] = 1;
    rep(i,n) {
        if (s[i] == '.') {
            // . 
            {
                dp[0][i + 1] += dp[0][i];
                if (i % 2 == 1)dp[0][i + 1] += dp[1][i] + dp[2][i];
            }
            // d
            if (s[i + 1] != 'd') {
                if (i % 2 == 0)dp[1][i + 1] += dp[0][i];
                dp[1][i + 1] += dp[1][i] + dp[2][i];
            }
            // k
            if (s[i + 1] != 'k') {
                if (i % 2 == 0)dp[2][i + 1] += dp[0][i];
                dp[2][i + 1] += dp[1][i] + dp[2][i];
            }
        }
        else {
            if (i % 2 == 0)dp[idx(s[i])][i + 1] += dp[0][i];
            dp[idx(s[i])][i + 1] += dp[1][i] + dp[2][i];
        }
    }

    cout << (dp[0][n] + dp[1][n] + dp[2][n]).val() << "\n";
    
    return 0;
}
0