結果

問題 No.2060 AND Sequence
ユーザー 1234512345
提出日時 2022-08-26 23:17:13
言語 Rust
(1.77.0)
結果
WA  
実行時間 -
コード長 3,272 bytes
コンパイル時間 521 ms
コンパイル使用メモリ 160,896 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-04-22 01:24:45
合計ジャッジ時間 1,566 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 AC 1 ms
5,376 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 AC 1 ms
5,376 KB
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
権限があれば一括ダウンロードができます
コンパイルメッセージ
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

ソースコード

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