#793202 (Rust) No.2060 AND Sequence

• 問題: No.2060 AND Sequence
• ユーザー: 12345
• 提出日時: 2022-08-26 23:18:10
• 言語: Rust (1.77.0)
• 結果: AC
• 実行時間: 1 ms / 2,000 ms
• コード長: 3,272 bytes
• コンパイル時間: 1,234 ms
• コンパイル使用メモリ: 173,428 KB
• 実行使用メモリ: 6,944 KB
• 最終ジャッジ日時: 2024-04-22 01:26:25
• 合計ジャッジ時間: 2,000 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,812 KB
testcase_01	AC	1 ms; 6,940 KB
testcase_02	AC	1 ms; 6,940 KB
testcase_03	AC	1 ms; 6,944 KB
testcase_04	AC	1 ms; 6,940 KB
testcase_05	AC	1 ms; 6,944 KB
testcase_06	AC	1 ms; 6,944 KB
testcase_07	AC	1 ms; 6,944 KB
testcase_08	AC	1 ms; 6,940 KB
testcase_09	AC	1 ms; 6,940 KB
testcase_10	AC	1 ms; 6,944 KB
testcase_11	AC	0 ms; 6,944 KB
testcase_12	AC	0 ms; 6,944 KB
testcase_13	AC	1 ms; 6,940 KB
testcase_14	AC	1 ms; 6,940 KB
testcase_15	AC	1 ms; 6,944 KB
testcase_16	AC	1 ms; 6,944 KB
testcase_17	AC	0 ms; 6,940 KB
testcase_18	AC	1 ms; 6,940 KB
testcase_19	AC	1 ms; 6,944 KB
testcase_20	AC	1 ms; 6,940 KB
testcase_21	AC	1 ms; 6,944 KB
testcase_22	AC	1 ms; 6,944 KB
testcase_23	AC	1 ms; 6,944 KB
testcase_24	AC	1 ms; 6,944 KB
testcase_25	AC	1 ms; 6,944 KB
testcase_26	AC	1 ms; 6,940 KB
testcase_27	AC	1 ms; 6,944 KB
testcase_28	AC	1 ms; 6,940 KB
testcase_29	AC	1 ms; 6,940 KB
testcase_30	AC	1 ms; 6,940 KB
testcase_31	AC	1 ms; 6,940 KB
testcase_32	AC	1 ms; 6,944 KB
testcase_33	AC	1 ms; 6,944 KB
testcase_34	AC	1 ms; 6,940 KB
testcase_35	AC	0 ms; 6,940 KB
testcase_36	AC	1 ms; 6,944 KB
testcase_37	AC	1 ms; 6,940 KB
testcase_38	AC	1 ms; 6,944 KB
testcase_39	AC	1 ms; 6,940 KB
testcase_40	AC	1 ms; 6,940 KB
testcase_41	AC	0 ms; 6,940 KB
testcase_42	AC	1 ms; 6,940 KB
testcase_43	AC	1 ms; 6,944 KB
testcase_44	AC	1 ms; 6,940 KB
testcase_45	AC	1 ms; 6,940 KB

コンパイルメッセージ

warning: variable does not need to be mutable
  --> 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
   --> 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)

warning: 2 warnings emitted

ソースコード

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();
}