結果
問題 | No.837 Noelちゃんと星々2 |
ユーザー |
|
提出日時 | 2023-11-20 18:18:36 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 17 ms / 2,000 ms |
コード長 | 3,483 bytes |
コンパイル時間 | 11,450 ms |
コンパイル使用メモリ | 391,880 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-09-26 06:40:56 |
合計ジャッジ時間 | 13,504 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 29 |
ソースコード
#![allow(unused_imports)]// use itertools::Itertools;// use proconio::input;// use proconio::marker::*;use std::cmp::Reverse;use std::collections::*;fn main() {input! {n:usize,y:[i64;n],}if y.iter().all(|yi| *yi == y[0]) {println!("1");return;}let mut y = y;y.sort();let tyuo = |y: &[i64]| {if y.len() % 2 == 0 {(y[y.len() / 2] + y[y.len() / 2 - 1]) / 2} else {y[y.len() / 2]}};let acc_y = y.cum();const INF: i64 = std::i64::MAX;let mut ans = INF;for i in 1..n {let tyuo0 = tyuo(&y[..i]);let tyuo1 = tyuo(&y[i..]);if tyuo0 == tyuo1 {continue;}let cost0 = if i % 2 == 0 {-acc_y[i / 2] + acc_y[i] - acc_y[i / 2]} else {-acc_y[i / 2] + acc_y[i] - acc_y[i / 2 + 1]};let cost1 = if (n - i) % 2 == 0 {-(acc_y[i + (n - i) / 2] - acc_y[i]) + acc_y[n] - acc_y[i + (n - i) / 2]} else {-(acc_y[i + (n - i) / 2] - acc_y[i]) + acc_y[n] - acc_y[i + (n - i) / 2 + 1]};ans = ans.min(cost0 + cost1);}println!("{}", ans);}use cumulative::*;mod cumulative {use std::ops::Add;pub trait Cumulative<T: Default + Copy + Add> {fn cum(&self) -> Vec<T>;fn cum_rev(&self) -> Vec<T>;}impl<T> Cumulative<T> for [T]whereT: Default + Copy + Add<Output = T>,{//cum[i]:=sum[0,i)fn cum(&self) -> Vec<T> {let n = self.len();let mut res = vec![T::default(); n + 1];for i in 0..n {res[i + 1] = res[i] + self[i];}res}//cum_rev[i]:=sum[i,n)fn cum_rev(&self) -> Vec<T> {let n = self.len();let mut res = vec![T::default(); n + 1];for i in (0..n).rev() {res[i] = res[i + 1] + self[i];}res}}}mod input {#[macro_export]macro_rules! input {(source = $s:expr, $($r:tt)*) => {let mut iter = $s.split_whitespace();input_inner!{iter, $($r)*}};($($r:tt)*) => {let s = {use std::io::Read;let mut s = String::new();std::io::stdin().read_to_string(&mut s).unwrap();s};let mut iter = s.split_whitespace();input_inner!{iter, $($r)*}};}#[macro_export]macro_rules! input_inner {($iter:expr) => {};($iter:expr, ) => {};($iter:expr, $var:ident : $t:tt $($r:tt)*) => {let $var = read_value!($iter, $t);input_inner!{$iter $($r)*}};}#[macro_export]macro_rules! read_value {($iter:expr, ( $($t:tt),* )) => {( $(read_value!($iter, $t)),* )};($iter:expr, [ $t:tt ; $len:expr ]) => {(0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>()};($iter:expr, chars) => {read_value!($iter, String).chars().collect::<Vec<char>>()};($iter:expr, usize1) => {read_value!($iter, usize) - 1};($iter:expr, $t:ty) => {$iter.next().unwrap().parse::<$t>().expect("Parse error")};}}