結果

問題 No.1330 Multiply or Divide
ユーザー Strorkis
提出日時 2021-01-09 11:35:53
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 3,792 bytes
コンパイル時間 13,560 ms
コンパイル使用メモリ 399,644 KB
実行使用メモリ 8,176 KB
最終ジャッジ日時 2025-06-20 01:24:30
合計ジャッジ時間 15,805 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff # 

struct Scanner<R> { r: R, buf: Vec<u8> }

impl<R: std::io::BufRead> Scanner<R> {
    fn new(r: R) -> Scanner<R> {
        Scanner { r, buf: Vec::with_capacity(8000) }
    }

    fn next(&mut self) -> &str {
        self.buf.clear();
        loop {
            let (done, used) = {
                let available = self.r.fill_buf().unwrap();
                match available.iter().position(|b| !b.is_ascii_whitespace()) {
                    Some(i) => (true, i),
                    None => (false, available.len()),
                }
            };
            self.r.consume(used);
            if done || used == 0 { break; }
        }
        loop {
            let (done, used) = {
                let available = self.r.fill_buf().unwrap();
                match available.iter().position(|b| b.is_ascii_whitespace()) {
                    Some(i) => {
                        self.buf.extend_from_slice(&available[..i]);
                        (true, i)
                    }
                    None => {
                        self.buf.extend_from_slice(&available);
                        (false, available.len())
                    }
                }
            };
            self.r.consume(used);
            if done || used == 0 {
                unsafe {
                    return std::str::from_utf8_unchecked(&self.buf);
                }
            }
        }
    }
}

macro_rules! parse {
    ($s:expr, Chars) => ($s.chars().collect::<Vec<_>>());
    ($s:expr, Bytes) => ($s.bytes().collect::<Vec<_>>());
    ($s:expr, Usize1) => (parse!($s, usize) - 1);
    ($s:expr, $t:ty) => ($s.parse::<$t>().unwrap());
}

macro_rules! read {
    ($sc:expr, [$t:tt; $n:expr]) => (
        (0..$n).map(|_| read!($sc, $t)).collect::<Vec<_>>()
    );
    ($sc:expr, ($($t:tt),*)) => (($(read!($sc, $t)),*));
    ($sc:expr, $t:ident) => (parse!($sc.next(), $t));
    ($sc:expr, $($t:tt),*) => (($(read!($sc, $t)),*));
}

fn main() {
    use std::io::Write;
    let (stdin, stdout) = (std::io::stdin(), std::io::stdout());
    let mut sc = Scanner::new(stdin.lock());
    let mut wr = std::io::BufWriter::new(stdout.lock());
    macro_rules! scan {
        ($($t:tt)*) => (read!(sc, $($t)*));
    }
    macro_rules! println {
        ($($arg:tt)*) => (writeln!(wr, $($arg)*).ok());
    }

    let (n, m, p) = scan!(usize, usize, usize);
    let a = scan!([usize; n]);

    let max_a = a.iter().max().unwrap();
    if m < *max_a {
        println!("1");
        return;
    }
    let m = m / max_a;

    let mut b = Vec::with_capacity(n);
    for mut a in a {
        let mut c: usize = 1;
        while a % p == 0 {
            a /= p;
            c += 1;
        }
        if a == 1 { continue; }
        b.push((c, a));
    }
    b.sort_by_key(|key| (key.0, std::cmp::Reverse(key.1)));

    let mut v = Vec::new();
    let mut max_a = 0;
    for (c, a) in b {
        if a <= max_a { continue; }
        v.push((c, a));
        max_a = max_a.max(a);
    }

    use std::cmp::Reverse;

    let mut ans = std::usize::MAX;
    let mut map = std::collections::BTreeMap::new();
    let mut heap = std::collections::BinaryHeap::new();
    map.insert(0, 1);
    heap.push(Reverse(0));
    while let Some(Reverse(cost)) = heap.pop() {
        let x = map[&cost];
        for (c, a) in &v {
            let next_cost = cost + c;
            let next_x = x * a;
            if next_x > m {
                ans = ans.min(next_cost + 1);
            } else {
                if let Some(value) = map.get_mut(&next_cost) {
                    *value = std::cmp::max(*value, next_x);
                } else {
                    map.insert(next_cost, next_x);
                    heap.push(Reverse(next_cost));
                }
            }
        }
    }
    println!("{}", ans as isize);
}
0