結果
問題 |
No.1251 絶対に間違ってはいけない最小化問題
|
ユーザー |
![]() |
提出日時 | 2020-10-09 21:48:21 |
言語 | Rust (1.83.0 + proconio) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,542 bytes |
コンパイル時間 | 25,450 ms |
コンパイル使用メモリ | 379,352 KB |
実行使用メモリ | 11,320 KB |
最終ジャッジ日時 | 2024-07-20 10:32:25 |
合計ジャッジ時間 | 37,141 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 13 WA * 29 |
ソースコード
#![allow(unused_imports)] use std::cmp::*; use std::collections::*; use std::io::Write; use std::ops::Bound::*; #[allow(unused_macros)] macro_rules! debug { ($($e:expr),*) => { #[cfg(debug_assertions)] $({ let (e, mut err) = (stringify!($e), std::io::stderr()); writeln!(err, "{} = {:?}", e, $e).unwrap() })* }; } fn main() { let n = read::<usize>(); let a = read_vec::<i64>(); let b = read_vec::<i64>(); let b_sum = b.iter().sum::<i64>(); let mut pairs = vec![]; for i in 0..n { pairs.push((a[i], b[i])); } pairs.sort(); let mut cur = 0; let mut cand = 0; for i in 0..n { cur += b[i]; if cur * 2 >= b_sum { cand = a[i]; break; } } let mut ans = pairs .iter() .map(|&(anum, bnum)| (anum - cand).abs() * bnum) .sum::<i64>(); let mut ii = cand; for i in -500..501 { let temp = pairs .iter() .map(|&(anum, bnum)| (anum - (cand + i)).abs() * bnum) .sum::<i64>(); if temp < ans { ans = temp; ii = cand + i; } } println!("{} {}", ii, ans); } fn read<T: std::str::FromStr>() -> T { let mut s = String::new(); std::io::stdin().read_line(&mut s).ok(); s.trim().parse().ok().unwrap() } fn read_vec<T: std::str::FromStr>() -> Vec<T> { read::<String>() .split_whitespace() .map(|e| e.parse().ok().unwrap()) .collect() }