結果
問題 | No.1514 Squared Matching |
ユーザー |
|
提出日時 | 2022-12-01 21:37:50 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 1,830 ms / 4,000 ms |
コード長 | 932 bytes |
コンパイル時間 | 17,519 ms |
コンパイル使用メモリ | 379,428 KB |
実行使用メモリ | 392,888 KB |
最終ジャッジ日時 | 2024-10-09 02:25:51 |
合計ジャッジ時間 | 44,044 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
fn get_factors(n: usize) -> Vec<u32> { let mut flgs = vec![true; n+1]; flgs[0] = false; flgs[1] = false; let mut ret = (0..=n as u32).collect::<Vec<_>>(); for i in 2..=(n as f64).sqrt().ceil() as usize { if !flgs[i] { continue; } let base = i * i; let base = base as u32; for j in i..=n/i { let idx = i*j; flgs[idx] = false; while ret[idx] % base == 0 { ret[idx] /= base; } } } ret } fn main() { let mut n = String::new(); std::io::stdin().read_line(&mut n).ok(); let n: usize = n.trim().parse().unwrap(); let factors = get_factors(n); let mut cnts = vec![0u32; n+1]; for &v in factors.iter() { cnts[v as usize] += 1; } let mut result = 0usize; for a in 1..=n { result += cnts[factors[a] as usize] as usize; } println!("{}", result); }