結果

問題 No.1740 Alone 'a'
ユーザー StrorkisStrorkis
提出日時 2021-11-12 23:33:16
言語 Rust
(1.77.0)
結果
AC  
実行時間 9 ms / 2,000 ms
コード長 2,507 bytes
コンパイル時間 1,096 ms
コンパイル使用メモリ 157,524 KB
実行使用メモリ 4,384 KB
最終ジャッジ日時 2023-08-17 04:19:10
合計ジャッジ時間 3,155 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 9 ms
4,376 KB
testcase_04 AC 9 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 1 ms
4,376 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,376 KB
testcase_09 AC 1 ms
4,380 KB
testcase_10 AC 1 ms
4,380 KB
testcase_11 AC 2 ms
4,380 KB
testcase_12 AC 8 ms
4,376 KB
testcase_13 AC 7 ms
4,380 KB
testcase_14 AC 9 ms
4,380 KB
testcase_15 AC 1 ms
4,380 KB
testcase_16 AC 8 ms
4,376 KB
testcase_17 AC 1 ms
4,376 KB
testcase_18 AC 2 ms
4,376 KB
testcase_19 AC 7 ms
4,376 KB
testcase_20 AC 4 ms
4,384 KB
testcase_21 AC 6 ms
4,380 KB
testcase_22 AC 6 ms
4,380 KB
testcase_23 AC 7 ms
4,380 KB
testcase_24 AC 4 ms
4,384 KB
testcase_25 AC 4 ms
4,380 KB
testcase_26 AC 6 ms
4,380 KB
testcase_27 AC 6 ms
4,380 KB
testcase_28 AC 5 ms
4,380 KB
testcase_29 AC 9 ms
4,376 KB
testcase_30 AC 6 ms
4,380 KB
testcase_31 AC 7 ms
4,380 KB
testcase_32 AC 5 ms
4,380 KB
testcase_33 AC 7 ms
4,380 KB
testcase_34 AC 3 ms
4,380 KB
testcase_35 AC 5 ms
4,376 KB
testcase_36 AC 4 ms
4,380 KB
testcase_37 AC 2 ms
4,376 KB
testcase_38 AC 3 ms
4,376 KB
testcase_39 AC 2 ms
4,380 KB
testcase_40 AC 2 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

pub mod io {
    pub fn scan<R: std::io::BufRead>(r: &mut R) -> Vec<u8> {
        let mut res = Vec::new();
        loop {
            let buf = match r.fill_buf() {
                Ok(buf) => buf,
                Err(e) if e.kind() == std::io::ErrorKind::Interrupted => continue,
                Err(e) => panic!("{}", e), 
            };
            let (done, used, buf) = match buf.iter().position(u8::is_ascii_whitespace) {
                Some(i) => (i > 0 || res.len() > 0, i + 1, &buf[..i]),
                None => (buf.is_empty(), buf.len(), buf),
            };
            res.extend_from_slice(buf);
            r.consume(used);
            if done { return res; }
        }
    }

    #[macro_export]
    macro_rules! scan {
        ($r:expr, [$t:tt; $n:expr]) => ((0..$n).map(|_| scan!($r, $t)).collect::<Vec<_>>());
        ($r:expr, [$t:tt]) => (scan!($r, [$t; scan!($r, usize)]));
        ($r:expr, ($($t:tt),*)) => (($(scan!($r, $t)),*));
        ($r:expr, Usize1) => (scan!($r, usize) - 1);
        ($r:expr, String) => (String::from_utf8(scan!($r, Bytes)).unwrap());
        ($r:expr, Bytes) => (io::scan($r));
        ($r:expr, $t:ty) => (scan!($r, String).parse::<$t>().unwrap());
    }

    #[macro_export]
    macro_rules! input {
        ($r:expr, $($($v:ident)* : $t:tt),* $(,)?) => {
            $(let $($v)* = scan!($r, $t);)*
        };
    }
}

const MOD: u64 = 998_244_353;

fn run<R: std::io::BufRead, W: std::io::Write>(reader: &mut R, writer: &mut W) {
    input! {
        reader,
        _n: usize,
        s: Bytes,
    }

    // inner, inner_a, border, border_a
    let mut dp = vec![0; 4];
    dp[2] = 1;
    for b in s {
        let mut ndp = vec![0; 4];
        ndp[0] += dp[0] * 25;
        ndp[1] += dp[0] + dp[1] * 25;
        if b == b'a' {
            if dp[2] == 1 {
                ndp[2] = 0;
                ndp[3] = 1;    
            } else {
                ndp[2] = 0;
                ndp[3] = 0;
            }
        } else {
            ndp[0] += dp[2] * (b - b'b') as u64;
            ndp[1] += dp[2] + dp[3] * (b - b'b') as u64;
            ndp[2] = dp[2];
            ndp[3] = dp[3];    
        }
        dp = ndp;
        dp.iter_mut().for_each(|dp| *dp %= MOD);
    }
    writeln!(writer, "{}", dp[1]).ok();
}

fn main() {
    let (stdin, stdout) = (std::io::stdin(), std::io::stdout());
    let ref mut reader = std::io::BufReader::new(stdin.lock());
    let ref mut writer = std::io::BufWriter::new(stdout.lock());
    run(reader, writer);
}
0