結果
問題 | No.1531 く2 |
ユーザー |
![]() |
提出日時 | 2021-05-15 17:35:43 |
言語 | Rust (1.83.0 + proconio) |
結果 |
TLE
|
実行時間 | - |
コード長 | 9,770 bytes |
コンパイル時間 | 10,798 ms |
コンパイル使用メモリ | 401,936 KB |
実行使用メモリ | 13,636 KB |
最終ジャッジ日時 | 2024-10-03 05:09:10 |
合計ジャッジ時間 | 17,191 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 24 TLE * 2 -- * 18 |
コンパイルメッセージ
warning: unused imports: `Leaf`, `Tuple`, `VecLen` --> src/main.rs:38:27 | 38 | multi_token::{Leaf, Parser, ParserTuple, RawTuple, Tuple, VecLen}, | ^^^^ ^^^^^ ^^^^^^ | = note: `#[warn(unused_imports)]` on by default warning: unused import: `with_str` --> src/main.rs:294:35 | 294 | pub use self::i::{with_stdin, with_str}; | ^^^^^^^^ warning: unused imports: `ParserTuple`, `Parser`, `RawTuple`, `Token`, `Usize1` --> src/main.rs:297:28 | 297 | pub use super::i::{Parser, ParserTuple, RawTuple, Token, Usize1}; | ^^^^^^ ^^^^^^^^^^^ ^^^^^^^^ ^^^^^ ^^^^^^
ソースコード
use crate::ngtio::with_stdin;const LEN: usize = 64;fn main() {let mut buf = with_stdin();let n = buf.u64();let k = buf.u64();let mut cnt = vec![0; LEN + 2];let mut ans = 0;let mut bs = n;loop {cnt[(bs + k).count_ones() as usize] += 1;if bs == 0 {break;} else {bs = (bs - 1) & n;}}for i in 0..=LEN + 1 {ans += cnt[i] * 2u128.pow(i as u32);}ans %= 998244353;println!("{}", ans);}// ngtio {{{#[allow(dead_code)]mod ngtio {mod i {use std::{io::{self, BufRead},iter,};pub use self::{multi_token::{Leaf, Parser, ParserTuple, RawTuple, Tuple, VecLen},token::{Token, Usize1},};pub fn with_stdin() -> Tokenizer<io::BufReader<io::Stdin>> {io::BufReader::new(io::stdin()).tokenizer()}pub fn with_str(src: &str) -> Tokenizer<&[u8]> {src.as_bytes().tokenizer()}pub struct Tokenizer<S: BufRead> {queue: Vec<String>, // FIXME: String のみにすると速そうです。scanner: S,}macro_rules! prim_method {($name:ident: $T:ty) => {pub fn $name(&mut self) -> $T {<$T>::leaf().parse(self)}};($name:ident) => {prim_method!($name: $name);};}macro_rules! prim_methods {($name:ident: $T:ty; $($rest:tt)*) => {prim_method!($name:$T);prim_methods!($($rest)*);};($name:ident; $($rest:tt)*) => {prim_method!($name);prim_methods!($($rest)*);};() => ()}impl<S: BufRead> Tokenizer<S> {pub fn token(&mut self) -> String {self.load();self.queue.pop().expect("入力が終了したのですが。")}pub fn new(scanner: S) -> Self {Self {queue: Vec::new(),scanner,}}fn load(&mut self) {while self.queue.is_empty() {let mut s = String::new();let length = self.scanner.read_line(&mut s).unwrap(); // 入力が UTF-8 でないときにエラーだそうです。if length == 0 {break;}self.queue = s.split_whitespace().rev().map(str::to_owned).collect();}}pub fn skip_line(&mut self) {assert!(self.queue.is_empty(),"行の途中で呼ばないでいただきたいです。現在のトークンキュー: {:?}",&self.queue);self.load();}pub fn end(&mut self) {self.load();assert!(self.queue.is_empty(), "入力はまだあります!");}pub fn parse<T: Token>(&mut self) -> T::Output {T::parse(&self.token())}pub fn parse_collect<T: Token, B>(&mut self, n: usize) -> BwhereB: iter::FromIterator<T::Output>,{iter::repeat_with(|| self.parse::<T>()).take(n).collect()}pub fn tuple<T: RawTuple>(&mut self) -> <T::LeafTuple as Parser>::Output {T::leaf_tuple().parse(self)}pub fn vec<T: Token>(&mut self, len: usize) -> Vec<T::Output> {T::leaf().vec(len).parse(self)}pub fn vec_tuple<T: RawTuple>(&mut self,len: usize,) -> Vec<<T::LeafTuple as Parser>::Output> {T::leaf_tuple().vec(len).parse(self)}pub fn vec2<T: Token>(&mut self, height: usize, width: usize) -> Vec<Vec<T::Output>> {T::leaf().vec(width).vec(height).parse(self)}pub fn vec2_tuple<T>(&mut self,height: usize,width: usize,) -> Vec<Vec<<T::LeafTuple as Parser>::Output>>whereT: RawTuple,{T::leaf_tuple().vec(width).vec(height).parse(self)}prim_methods! {u8; u16; u32; u64; u128; usize;i8; i16; i32; i64; i128; isize;f32; f64;char; string: String;}}mod token {use super::multi_token::Leaf;use std::{any, fmt, marker, str};pub trait Token: Sized {type Output;fn parse(s: &str) -> Self::Output;fn leaf() -> Leaf<Self> {Leaf(marker::PhantomData)}}impl<T> Token for TwhereT: str::FromStr,<Self as str::FromStr>::Err: fmt::Debug,{type Output = Self;fn parse(s: &str) -> Self::Output {s.parse().unwrap_or_else(|_| {panic!("Parse error!: ({}: {})", s, any::type_name::<Self>(),)})}}pub struct Usize1 {}impl Token for Usize1 {type Output = usize;fn parse(s: &str) -> Self::Output {usize::parse(s).checked_sub(1).expect("Parse error! (Zero substruction error of Usize1)")}}}mod multi_token {use super::{Token, Tokenizer};use std::{io::BufRead, iter, marker};pub trait Parser: Sized {type Output;fn parse<S: BufRead>(&self, server: &mut Tokenizer<S>) -> Self::Output;fn vec(self, len: usize) -> VecLen<Self> {VecLen { len, elem: self }}}pub struct Leaf<T>(pub(super) marker::PhantomData<T>);impl<T: Token> Parser for Leaf<T> {type Output = T::Output;fn parse<S: BufRead>(&self, server: &mut Tokenizer<S>) -> T::Output {server.parse::<T>()}}pub struct VecLen<T> {pub len: usize,pub elem: T,}impl<T: Parser> Parser for VecLen<T> {type Output = Vec<T::Output>;fn parse<S: BufRead>(&self, server: &mut Tokenizer<S>) -> Self::Output {iter::repeat_with(|| self.elem.parse(server)).take(self.len).collect()}}pub trait RawTuple {type LeafTuple: Parser;fn leaf_tuple() -> Self::LeafTuple;}pub trait ParserTuple {type Tuple: Parser;fn tuple(self) -> Self::Tuple;}pub struct Tuple<T>(pub T);macro_rules! impl_tuple {($($t:ident: $T:ident),*) => {impl<$($T),*> Parser for Tuple<($($T,)*)>where$($T: Parser,)*{type Output = ($($T::Output,)*);#[allow(unused_variables)]fn parse<S: BufRead >(&self, server: &mut Tokenizer<S>) -> Self::Output {match self {Tuple(($($t,)*)) => {($($t.parse(server),)*)}}}}impl<$($T: Token),*> RawTuple for ($($T,)*) {type LeafTuple = Tuple<($(Leaf<$T>,)*)>;fn leaf_tuple() -> Self::LeafTuple {Tuple(($($T::leaf(),)*))}}impl<$($T: Parser),*> ParserTuple for ($($T,)*) {type Tuple = Tuple<($($T,)*)>;fn tuple(self) -> Self::Tuple {Tuple(self)}}};}impl_tuple!();impl_tuple!(t1: T1);impl_tuple!(t1: T1, t2: T2);impl_tuple!(t1: T1, t2: T2, t3: T3);impl_tuple!(t1: T1, t2: T2, t3: T3, t4: T4);impl_tuple!(t1: T1, t2: T2, t3: T3, t4: T4, t5: T5);impl_tuple!(t1: T1, t2: T2, t3: T3, t4: T4, t5: T5, t6: T6);impl_tuple!(t1: T1, t2: T2, t3: T3, t4: T4, t5: T5, t6: T6, t7: T7);impl_tuple!(t1: T1,t2: T2,t3: T3,t4: T4,t5: T5,t6: T6,t7: T7,t8: T8);}trait Scanner: BufRead + Sized {fn tokenizer(self) -> Tokenizer<Self> {Tokenizer::new(self)}}impl<R: BufRead> Scanner for R {}}pub use self::i::{with_stdin, with_str};pub mod prelude {pub use super::i::{Parser, ParserTuple, RawTuple, Token, Usize1};}}// }}}