結果

問題 No.1495 パンの仕入れ
ユーザー akakimidori
提出日時 2021-03-31 12:58:46
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 186 ms / 2,000 ms
コード長 2,991 bytes
コンパイル時間 13,562 ms
コンパイル使用メモリ 378,492 KB
実行使用メモリ 11,252 KB
最終ジャッジ日時 2024-12-16 05:42:43
合計ジャッジ時間 16,631 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

const MAX_T: usize = 100;
const MAX_N: usize = 200_000;
const MAX_M: usize = 200_000;
const MAX_K: i64 = 1_000_000_000;
const MAX_Y: i64 = 1_000_000_000;
fn read() -> Vec<(usize, i64, Vec<(usize, i64)>)> {
    let mut s = String::new();
    use std::io::Read;
    std::io::stdin().read_to_string(&mut s).unwrap();
    let mut it = s.trim().split_whitespace();
    let mut next = || it.next().unwrap().parse::<i64>().unwrap();
    let t = next() as usize;
    assert!(1 <= t && t <= MAX_T);
    let mut test = vec![];
    for _ in 0..t {
        let n = next() as usize;
        assert!(1 <= n && n <= MAX_N);
        let m = next() as usize;
        assert!(1 <= m && m <= MAX_M);
        let k = next();
        assert!(1 <= k && k <= MAX_K);
        let mut p = vec![(0usize, 0i64); m];
        for p in p.iter_mut() {
            p.0 = next() as usize;
            assert!(1 <= p.0 && p.0 <= n);
            p.0 -= 1;
            p.1 = next();
            assert!(1 <= p.1 && p.1 <= MAX_Y);
        }
        assert!(
            m.saturating_mul(k.max(p.iter().map(|p| p.1).max().unwrap()) as usize)
                <= 10usize.pow(18)
        );
        test.push((n, k, p));
    }
    test
}
fn solve(n: usize, k: i64, p: &[(usize, i64)]) -> i64 {
    let mut dp = vec![(0, 0, 0); n];
    for &(x, y) in p.iter() {
        let po = &mut dp[x];
        po.0 += y * y;
        po.1 -= 2 * y;
        po.2 += 1;
    }
    let eval = |m: i64| -> (i64, i64) {
        if m >= 0 {
            let mut cnt = 0;
            let mut cost = 0;
            let mut zero = false;
            for &(c, b, a) in dp.iter() {
                cost += c;
                zero |= a == 0;
                if a > 0 && 0 >= b - a {
                    let x = (0 - (b - a)) / (2 * a);
                    cnt += x;
                    cost += 2 * a * (x + 1) * x / 2 + (b - a) * x;
                }
            }
            if zero {
                cnt += k;
            }
            if cnt >= k {
                return (cnt, cost);
            }
        }
        let mut cnt = 0;
        let mut cost = 0;
        let mut p = vec![0; n];
        for (p, &(c, b, a)) in p.iter_mut().zip(dp.iter()) {
            cost += c;
            if a > 0 && m >= b - a {
                let x = (m - (b - a)) / (2 * a);
                cnt += x;
                *p = x;
                cost += 2 * a * (x + 1) * x / 2 + (b - a) * x;
            }
        }
        (cnt, cost)
    };
    let mut ng = dp.iter().map(|&(_, b, a)| 2 * a + b - a - 1).min().unwrap();
    let mut ok = dp.iter().map(|&(_, b, a)| 2 * a * k + b - a).min().unwrap();
    while ok - ng > 1 {
        let mid = (ok + ng) / 2;
        if eval(mid).0 >= k {
            ok = mid;
        } else {
            ng = mid;
        }
    }
    let (x, y) = eval(ok);
    let ans = y - ok * (x - k);
    ans
}
fn main() {
    let test = read();
    for (n, k, p) in test {
        let ans = solve(n, k, &p);
        println!("{}", ans);
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0