bouzuya.hatenablog.com

ぼうずやのにっき

ABC307 E を解いた

東京海上日動プログラミングコンテスト2023（ABC307: AtCoder Beginner Contest 307）

use modint::ModInt998244353 as ModInt;
use proconio::input;

fn main() {
    input! {
        n: usize,
        m: usize,
    };

    let mut dp = (ModInt::new(1), ModInt::new(0));
    for _ in 1..n {
        dp = (dp.1, dp.0 * (m - 1) + dp.1 * (m - 2));
    }
    let ans = dp.1 * m;
    println!("{}", ans);
}

// modint

今日のコミット。