結果

問題 No.1251 絶対に間違ってはいけない最小化問題
ユーザー Strorkis
提出日時 2020-10-09 21:57:55
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 63 ms / 2,000 ms
コード長 1,335 bytes
コンパイル時間 16,094 ms
コンパイル使用メモリ 394,112 KB
実行使用メモリ 10,520 KB
最終ジャッジ日時 2024-07-20 11:08:18
合計ジャッジ時間 23,474 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #

fn read_line<T: std::str::FromStr>() -> T {
    let mut input = String::new();
    std::io::stdin().read_line(&mut input).unwrap();
    input.trim_end().parse().ok().unwrap()
}

fn read_line_to_vec<T: std::str::FromStr>() -> Vec<T> {
    let mut input = String::new();
    std::io::stdin().read_line(&mut input).unwrap();
    let iter = input.split_whitespace();
    iter.map(|x| x.parse().ok().unwrap()).collect::<Vec<T>>()
}

fn main() {
    let n = read_line::<usize>();
    let a = read_line_to_vec::<i64>();
    let b = read_line_to_vec::<i64>();

    let mut order = (0..n).collect::<Vec<_>>();
    order.sort_by_key(|&i| a[i]);

    let mut left = vec![0; n];
    let mut sum = b[order[0]];
    for i in 1..n {
        let now = order[i];
        let prev = order[i - 1];
        left[i] = left[i - 1] + (a[now] - a[prev]) * sum;
        sum += b[now];
    }

    let mut right = vec![0; n];
    let mut sum = b[order[n - 1]];
    for i in (0..(n - 1)).rev() {
        let now = order[i];
        let prev = order[i + 1];
        right[i] = right[i + 1] + (a[prev] - a[now]) * sum;
        sum += b[now];
    }

    let (mut x, mut fx) = (0, std::i64::MAX);
    for i in 0..n {
        if left[i] + right[i] < fx {
            x = a[order[i]];
            fx = left[i] + right[i];
        }
    }
    println!("{} {}", x, fx);
}
0