結果

問題 No.2060 AND Sequence
コンテスト
ユーザー 12345
提出日時 2022-08-26 23:18:10
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 1 ms / 2,000 ms
コード長 3,272 bytes
コンパイル時間 24,801 ms
コンパイル使用メモリ 377,756 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-14 00:02:05
合計ジャッジ時間 15,837 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 43
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable does not need to be mutable
  --> src/main.rs:83:13
   |
83 |         let mut finv: Vec<_> = fact.iter().map(|x| modinv(*x, M)).collect();
   |             ----^^^^
   |             |
   |             help: remove this `mut`
   |
   = note: `#[warn(unused_mut)]` on by default

warning: variable does not need to be mutable
   --> src/main.rs:7:13
    |
7   |           let mut s = {
    |               ----^
    |               |
    |               help: remove this `mut`
...
130 | /     input! {
131 | |         n: i64,
132 | |         m: i64,
133 | |     };
    | |_____- in this macro invocation
    |
    = note: this warning originates in the macro `input` (in Nightly builds, run with -Z macro-backtrace for more info)

ソースコード

diff # 

macro_rules! input {
    (source = $s:expr, $($r:tt)*) => {
        let mut iter = $s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
    ($($r:tt)*) => {
        let mut s = {
            use std::io::Read;
            let mut s = String::new();
            std::io::stdin().read_to_string(&mut s).unwrap();
            s
        };
        let mut iter = s.split_whitespace();
        input_inner!{iter, $($r)*}
    };
}

macro_rules! input_inner {
    ($iter:expr) => {};
    ($iter:expr, ) => {};

    ($iter:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($iter, $t);
        input_inner!{$iter $($r)*}
    };
}

macro_rules! read_value {
    ($iter:expr, ( $($t:tt),* )) => {
        ( $(read_value!($iter, $t)),* )
    };

    ($iter:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()
    };

    ($iter:expr, chars) => {
        read_value!($iter, String).chars().collect::<Vec<char>>()
    };

    ($iter:expr, usize1) => {
        read_value!($iter, usize) - 1
    };

    ($iter:expr, $t:ty) => {
        $iter.next().unwrap().parse::<$t>().expect("Parse error")
    };
}

fn modpow(x: i64, p: i64, m: i64) -> i64 {
    if p == 0 {
        1
    } else if p == 1 {
        x
    } else if p % 2 == 1 {
        x * modpow(x, p - 1, m) % m
    } else {
        let t = modpow(x, p / 2, m);
        t * t % m
    }
}

fn modinv(x: i64, m: i64) -> i64{
    modpow(x, m - 2, m)
}

const K: i64 = 40;
const M: i64 = 998244353;

struct Solver {
    n: i64,
    m: i64,
    fact: Vec<i64>,
    finv: Vec<i64>,
}

impl Solver {
    fn new(n: i64, m: i64) -> Self {
        let mut fact = vec![1; (K + 1) as usize];
        for i in 1..(K + 1) {
            fact[i as usize] = fact[(i - 1) as usize] * (i as i64) % M;
        }
        let mut finv: Vec<_> = fact.iter().map(|x| modinv(*x, M)).collect();
        return Solver {
            n, m,
            fact, finv
        }
    }

    fn choose(&self, n: i64, m: i64) -> i64 {
        self.fact[n as usize] * self.finv[m as usize] % M * self.finv[(n - m) as usize] % M
    }

    fn countit(&self, popcount: i64) -> i64 {
        modpow(self.n, popcount, M)
    }

    fn search(&self, a: i64, k: i64) -> i64 {
        if k == -1 {
            // a自身のぶん。どこでも0にしなかったやつ。
            return self.countit(a.count_ones().into());
        }

        let mut ans = 0;
        if (a & (1_i64 << k)) != 0 {
            // kより上はa通り、k番目は0、kより下は2^k

            // i bits are one (lower k bits)
            for i in 0..(k + 1) {
                let t = self.choose(k, i);
                let f = (a >> (k + 1)).count_ones() as i64 + i;
                // dbg!((k, i, t, f));
                ans += self.countit(f) * t % M;
                ans %= M;
            }
        }

        ans += self.search(a, k - 1);
        ans %= M;
        return ans
    }

    fn solve(&mut self) {
        println!("{}", self.search(self.m, 35));
    }
}


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

    let mut solver = Solver::new(n, m);

    // dbg!(&solver.countit(0));
    // dbg!(&solver.countit(1));
    // dbg!(&solver.countit(2));

    solver.solve();
}
0