adt_easy_20240307_2 e

bouzuya · Mar 11, 2024 · baed17b · baed17b
1 parent 2c0f388
commit baed17b
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/cargo-atcoder-1.70.0/contests/adt_easy_20240307_2/src/bin/e.rs b/cargo-atcoder-1.70.0/contests/adt_easy_20240307_2/src/bin/e.rs
@@ -1,11 +1,20 @@
-// TODO
-use proconio::{input, marker::Usize1};
+use proconio::input;
 
 fn main() {
     input! {
         n: usize,
-        a: [Usize1; n],
+        mut a: [usize; n],
     };
-    let ans = n - a.len();
+    a.sort();
+    let sum = a.iter().sum::<usize>();
+    let avg = sum / n;
+    let avg2 = sum - avg * n;
+    let ans = vec![avg; n - avg2]
+        .into_iter()
+        .chain(vec![avg + 1; avg2])
+        .zip(a)
+        .map(|(x, y)| (x as i64 - y as i64).unsigned_abs() as usize)
+        .sum::<usize>()
+        / 2;
     println!("{}", ans);
 }