aoc

commit 2f0e522317972711b5e03ad25b38bf03af17e8d5
parent 58800a74af2c9f3578d3e62b3226aaf7c73aef87
Author: Sebastiano Tronto <sebastiano@tronto.net>
Date:   Mon,  7 Jul 2025 09:51:31 +0200

Day 20 2022

Diffstat:
A
2022/20/a.rs
 | 
8
++++++++
A
2022/20/b.rs
 | 
8
++++++++
A
2022/20/common.rs
 | 
37
+++++++++++++++++++++++++++++++++++++

3 files changed, 53 insertions(+), 0 deletions(-)
diff --git a/2022/20/a.rs b/2022/20/a.rs
@@ -0,0 +1,8 @@
+mod common;
+use common::*;
+
+fn main() {
+    let mut a = read_from_stdin(1);
+    mix(&mut a);
+    println!("{}", groove_sum(&a));
+}
diff --git a/2022/20/b.rs b/2022/20/b.rs
@@ -0,0 +1,8 @@
+mod common;
+use common::*;
+
+fn main() {
+    let mut a = read_from_stdin(811589153);
+    for _ in 0..10 { mix(&mut a); }
+    println!("{}", groove_sum(&a));
+}
diff --git a/2022/20/common.rs b/2022/20/common.rs
@@ -0,0 +1,37 @@
+const MAX_OVERFLOW: i64 = 10000000000;
+
+pub fn read_from_stdin(key: i64) -> Vec<(usize, i64)> {
+    let mut line = String::new();
+    let mut v = vec![];
+    while std::io::stdin().read_line(&mut line).unwrap() > 0 {
+        v.push((v.len(), line[..line.len()-1].parse::<i64>().unwrap() * key));
+        line.clear();
+    }
+    v
+}
+
+fn get_newpos(pos: i64, k: i64, n: i64) -> i64 {
+    (pos + k + MAX_OVERFLOW * (n-1)) % (n-1)
+}
+
+fn move_to(a: &mut [(usize, i64)], i: usize, j: usize) {
+    if i < j {
+        for k in i..j { a.swap(k, k+1); }
+    } else {
+        for k in (j+1..=i).rev() { a.swap(k, k-1); }
+    }
+}
+
+pub fn mix(a: &mut Vec<(usize, i64)>) {
+    for i in 0..a.len() {
+        let pos = a.iter().position(|x| x.0 == i).unwrap();
+        let newpos = get_newpos(pos as i64, a[pos].1, a.len() as i64) as usize;
+        move_to(a, pos, newpos);
+    }
+}
+
+pub fn groove_sum(a: &Vec<(usize, i64)>) -> i64 {
+    let n = a.len();
+    let z = a.iter().position(|x| x.1 == 0).unwrap();
+    a[(z + 1000) % n].1 +  a[(z + 2000) % n].1 + a[(z + 3000) % n].1
+}