結果
問題 | No.1653 Squarefree |
ユーザー | akakimidori |
提出日時 | 2021-08-20 22:10:35 |
言語 | Rust (1.77.0 + proconio) |
結果 |
AC
|
実行時間 | 728 ms / 2,000 ms |
コード長 | 4,223 bytes |
コンパイル時間 | 12,622 ms |
コンパイル使用メモリ | 392,160 KB |
実行使用メモリ | 77,184 KB |
最終ジャッジ日時 | 2024-10-14 08:18:23 |
合計ジャッジ時間 | 32,233 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 705 ms
77,184 KB |
testcase_01 | AC | 5 ms
5,248 KB |
testcase_02 | AC | 5 ms
5,248 KB |
testcase_03 | AC | 6 ms
5,248 KB |
testcase_04 | AC | 6 ms
5,248 KB |
testcase_05 | AC | 6 ms
5,248 KB |
testcase_06 | AC | 6 ms
5,248 KB |
testcase_07 | AC | 6 ms
5,248 KB |
testcase_08 | AC | 5 ms
5,248 KB |
testcase_09 | AC | 5 ms
5,248 KB |
testcase_10 | AC | 6 ms
5,248 KB |
testcase_11 | AC | 393 ms
73,600 KB |
testcase_12 | AC | 392 ms
73,600 KB |
testcase_13 | AC | 700 ms
76,672 KB |
testcase_14 | AC | 692 ms
76,544 KB |
testcase_15 | AC | 702 ms
77,184 KB |
testcase_16 | AC | 706 ms
76,800 KB |
testcase_17 | AC | 728 ms
76,928 KB |
testcase_18 | AC | 703 ms
77,184 KB |
testcase_19 | AC | 697 ms
76,772 KB |
testcase_20 | AC | 700 ms
77,184 KB |
testcase_21 | AC | 695 ms
77,184 KB |
testcase_22 | AC | 706 ms
77,148 KB |
testcase_23 | AC | 691 ms
76,416 KB |
testcase_24 | AC | 698 ms
76,728 KB |
testcase_25 | AC | 694 ms
77,104 KB |
testcase_26 | AC | 689 ms
76,544 KB |
testcase_27 | AC | 697 ms
76,996 KB |
testcase_28 | AC | 698 ms
76,544 KB |
testcase_29 | AC | 704 ms
77,056 KB |
testcase_30 | AC | 704 ms
76,372 KB |
testcase_31 | AC | 722 ms
77,056 KB |
testcase_32 | AC | 706 ms
76,928 KB |
testcase_33 | AC | 705 ms
76,668 KB |
testcase_34 | AC | 705 ms
77,056 KB |
testcase_35 | AC | 695 ms
76,928 KB |
testcase_36 | AC | 6 ms
5,248 KB |
testcase_37 | AC | 6 ms
5,248 KB |
testcase_38 | AC | 6 ms
5,248 KB |
testcase_39 | AC | 6 ms
5,248 KB |
testcase_40 | AC | 6 ms
5,248 KB |
ソースコード
// ---------- begin enumerate prime ---------- fn enumerate_prime<F>(n: usize, mut f: F) where F: FnMut(usize), { assert!(1 <= n && n <= 5 * 10usize.pow(8)); let batch = (n as f64).sqrt().ceil() as usize; let mut is_prime = vec![true; batch + 1]; for i in (2..).take_while(|p| p * p <= batch) { if is_prime[i] { let mut j = i * i; while let Some(p) = is_prime.get_mut(j) { *p = false; j += i; } } } let mut prime = vec![]; for (i, p) in is_prime.iter().enumerate().skip(2) { if *p && i <= n { f(i); prime.push(i); } } let mut l = batch + 1; while l <= n { let r = std::cmp::min(l + batch, n + 1); is_prime.clear(); is_prime.resize(r - l, true); for &p in prime.iter() { let mut j = (l + p - 1) / p * p - l; while let Some(is_prime) = is_prime.get_mut(j) { *is_prime = false; j += p; } } for (i, _) in is_prime.iter().enumerate().filter(|p| *p.1) { f(i + l); } l += batch; } } // ---------- end enumerate prime ---------- // floor(a ^ (1 / k)) fn kth_root(a: u64, k: u64) -> u64 { assert!(k > 0); if a == 0 { return 0; } if k >= 64 { return 1; } if k == 1 { return a; } let valid = |x: u64| -> bool { let mut t = x; for _ in 1..k { let (val, ok) = t.overflowing_mul(x); if !(!ok && val <= a) { return false; } t = val; } true }; let mut ok = 1; let mut ng = 2; while valid(ng) { ok = ng; ng *= 2; } while ng - ok > 1 { let mid = ok + (ng - ok) / 2; if valid(mid) { ok = mid; } else { ng = mid; } } ok } // ---------- begin input macro ---------- // reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8 macro_rules! input { (source = $s:expr, $($r:tt)*) => { let mut iter = $s.split_whitespace(); input_inner!{iter, $($r)*} }; ($($r:tt)*) => { let 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, bytes) => { read_value!($iter, String).bytes().collect::<Vec<u8>>() }; ($iter:expr, usize1) => { read_value!($iter, usize) - 1 }; ($iter:expr, $t:ty) => { $iter.next().unwrap().parse::<$t>().expect("Parse error") }; } // ---------- end input macro ---------- fn run() { input!(l: usize, r: usize); let mut factor = vec![vec![]; r - l + 1]; enumerate_prime(1000000, |p| { let mut x = (l + p - 1) / p * p - l; while x < factor.len() { factor[x].push(p); x += p; } }); let mut ans = 0; for (i, f) in factor.iter().enumerate() { let mut n = i + l; if n == 1 { ans += 1; continue; } let mut add = true; for &p in f.iter() { add &= n % (p * p) != 0; while n % p == 0 { n /= p; } } if n > 1 { let r = kth_root(n as u64, 2) as usize; add &= r * r != n; } if add { ans += 1; } } println!("{}", ans); } fn main() { run(); }