結果
問題 | No.2440 Accuracy of Integer Division Approximate Functions |
ユーザー |
👑 |
提出日時 | 2023-08-14 20:25:50 |
言語 | Rust (1.77.0) |
結果 |
AC
|
実行時間 | 8 ms / 2,000 ms |
コード長 | 2,807 bytes |
コンパイル時間 | 14,827 ms |
コンパイル使用メモリ | 398,292 KB |
実行使用メモリ | 6,944 KB |
最終ジャッジ日時 | 2024-06-02 09:57:40 |
合計ジャッジ時間 | 13,427 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 0 ms
6,812 KB |
testcase_01 | AC | 7 ms
6,812 KB |
testcase_02 | AC | 7 ms
6,940 KB |
testcase_03 | AC | 8 ms
6,940 KB |
testcase_04 | AC | 8 ms
6,944 KB |
testcase_05 | AC | 7 ms
6,944 KB |
testcase_06 | AC | 6 ms
6,944 KB |
testcase_07 | AC | 6 ms
6,940 KB |
testcase_08 | AC | 6 ms
6,944 KB |
testcase_09 | AC | 6 ms
6,944 KB |
testcase_10 | AC | 7 ms
6,944 KB |
testcase_11 | AC | 8 ms
6,940 KB |
testcase_12 | AC | 8 ms
6,940 KB |
testcase_13 | AC | 8 ms
6,944 KB |
testcase_14 | AC | 8 ms
6,940 KB |
testcase_15 | AC | 8 ms
6,944 KB |
testcase_16 | AC | 5 ms
6,944 KB |
testcase_17 | AC | 5 ms
6,940 KB |
testcase_18 | AC | 5 ms
6,944 KB |
testcase_19 | AC | 5 ms
6,940 KB |
testcase_20 | AC | 5 ms
6,944 KB |
ソースコード
// calc sum_{i=0}^{n-1} floor((ai + b) / m) (mod 2^64) pub fn floor_sum_unsigned(n: u64, mut m: u128, a: u64, b: u64) -> u64 { let (mut ans, mut n, mut a, mut b) = (0u64, n as u128, a as u128, b as u128); loop { if a >= m { ans = ans.wrapping_add(((n * (n - 1) >> 1).wrapping_mul(a / m)) as u64); a %= m } if b >= m { ans = ans.wrapping_add((n.wrapping_mul(b / m)) as u64); b %= m } let y_max = a * n + b; if y_max < m { return ans; } (n, b, m, a) = (y_max / m, y_max % m, a, m) } } // calc min(floor(((a + 1) << s) - 1) / b), 2^64 - 1) pub fn solve_div_helper(a: u64, b: u128, mut s: u32) -> u64 { assert!(b < (1u128 << 127)); let (mut ans, mut a) = (0u64, a as u128); loop { let t = s.min(a.leading_zeros()); a = (a.wrapping_add(1) << t).wrapping_sub(1); ans = if ans > 0 { if ans.leading_zeros() < t { !0u64 } else { ans << t } } else { 0 }; s -= t; ans = ans.saturating_add(u64::try_from(a / b).unwrap_or(!0u64)); a %= b; if s == 0 { break ans; } } } pub fn solve(mut n: u64, d: u64, m: u64, s: u32) -> u64 { assert!(n < (1u64 << 60)); assert!(d < (1u64 << 60) && d > 0); assert!(m < (1u64 << 60)); assert!(s < 121); let (pow2s, dm) = (1u128 << s, (d as u128) * (m as u128)); if pow2s < dm { n = n.min(solve_div_helper(d - 1, dm - pow2s, s)); n = n .wrapping_sub(floor_sum_unsigned(n + 1, pow2s, m, 0)) .wrapping_add(floor_sum_unsigned(n + 1, d as u128, 1, 0)); } else if pow2s > dm { n = n.min(solve_div_helper(d - 1, pow2s - dm, s)); n = n .wrapping_add(floor_sum_unsigned(n + 1, pow2s, m, 0)) .wrapping_sub(floor_sum_unsigned(n + 1, d as u128, 1, 0)); } n } fn main() { use std::io::prelude::*; let stdin = std::io::stdin(); let stdout = std::io::stdout(); let stdinlock = stdin.lock(); let stdoutlock = stdout.lock(); let mut bufwriter = std::io::BufWriter::new(stdoutlock); let mut lines = stdinlock.lines(); let q = lines.next().unwrap().unwrap().parse::<usize>().unwrap(); for _ in 0..q { let l = lines.next().unwrap().unwrap(); let mut token = l.split_ascii_whitespace(); let n = token.next().unwrap().parse::<u64>().unwrap(); let d = token.next().unwrap().parse::<u64>().unwrap(); let m = token.next().unwrap().parse::<u64>().unwrap(); let s = token.next().unwrap().parse::<u32>().unwrap(); writeln!(&mut bufwriter, "{}", solve(n, d, m, s)).unwrap(); } }