結果

問題 No.1276 3枚のカード
コンテスト
ユーザー akakimidori
提出日時 2021-07-23 19:04:18
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 1,534 bytes
コンパイル時間 11,322 ms
コンパイル使用メモリ 378,868 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-18 14:06:47
合計ジャッジ時間 16,009 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 61
権限があれば一括ダウンロードができます

ソースコード

diff # 

fn read() -> u32 {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).unwrap();
    s.trim().parse().unwrap()
}

// n を与えると [l, r], q を返す
// 順序はようわからん
// f(l, r, q): x \in [l, r] で floor(n / x) = q
fn quot_range<F>(n: u32, mut f: F)
where
    F: FnMut(u32, u32, u32)
{
    let mut p = Vec::with_capacity((n as f64).sqrt().ceil() as usize + 1);
    for d in 1.. {
        let q = n / d;
        p.push((d, q));
        if q < d {
            break;
        }
        f(d, d, q);
        if q == d {
            break;
        }
    }
    for p in p.windows(2).rev() {
        let (l, r, q) = (p[1].1 + 1, p[0].1, p[0].0);
        f(l, r, q);
    }
}

fn run() {
    let n = read();
    const MOD: u64 = 1_000_000_007;
    let mut ans = 0;
    quot_range(n, |l, r, q| {
        let l = l as u64;
        let r = r as u64;
        let q = q as u64;
        let n = n as u64;
        ans += (r - l + 1) * (q - 1) * (n - 2) % MOD;
        ans += MOD - (r - l + 1) * (q - 1) * (q - 2) % MOD;
    });
    quot_range(n, |l, r, q| {
        if q <= 2 {
            return;
        }
        let w = (r - l + 1) as u64;
        let mut s = 0;
        quot_range(q, |l, r, q| {
            if l > 1 {
                let l = l as u64;
                let r = r as u64;
                let q = q as u64;
                s += (r - l + 1) * (q - 1);
            }
        });
        ans += MOD - s % MOD * w % MOD;
    });
    ans %= MOD;
    println!("{}", ans);
}

fn main() {
    run();
}
0