kageru/advent-of-code
1
0
Fork 0
Code Issues 1 Pull Requests Releases Wiki Activity

refactor execution into function

Browse Source
This commit is contained in:
kageru 2019-12-07 08:02:10 +01:00
parent b2e5521dc0
commit 55bf996963
Signed by: kageru
GPG Key ID: 8282A2BEA4ADA3D2
1 changed files with 49 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
2019/07/src/main.rs
View File

@ -160,66 +160,60 @@ pub fn main() {
.map(|n| n.parse().unwrap()) .map(|n| n.parse().unwrap())
.collect(); .collect();
let p1: i64 = (0..5).permutations(5).map(|amps| { let p1: i64 = (0..5)
let mut last_output = 0; .permutations(5)
for amp in amps { .map(|amps| {
let mut pos = 0; let mut last_output = 0;
let mut part1_input = input.clone(); for amp in amps {
let mut inputs = vec![last_output, amp]; let mut pos = 0;
while let Some(op) = next_operation(&part1_input, &mut pos, &mut inputs) { let mut tape = input.clone();
if let Some(o) = execute(op, &mut part1_input, &mut pos) { let mut params = vec![last_output, amp];
last_output = o; match execute_machine(&mut tape, &mut pos, &mut params) {
break; Ok(o) => last_output = o,
Err(o) => last_output = o,
} }
} }
} last_output
last_output })
}).max().unwrap(); .max()
.unwrap();
println!("Part 1: {}", p1); println!("Part 1: {}", p1);
let p2: i64 = (5..10).permutations(5).map(|mut amps| { let p2: i64 = (5..10)
//for mut amps in (5..10).permutations(5) { .permutations(5)
let mut last_output = 0; .map(|amps| {
let mut inputs = vec![ let mut last_output = 0;
vec![amps.pop().unwrap()], let mut inputs: Vec<_> = amps.into_iter().map(move |amp| vec![amp]).collect();
vec![amps.pop().unwrap()], let mut positions = vec![0, 0, 0, 0, 0];
vec![amps.pop().unwrap()], let mut states: Vec<_> = (0..5).map(|_| input.clone()).collect();
vec![amps.pop().unwrap()], for state in (0..5).cycle() {
vec![amps.pop().unwrap()], let mut part1_input = states.get_mut(state).unwrap();
]; let mut pos = positions.get_mut(state).unwrap();
let mut positions = vec![0, 0, 0, 0, 0]; let mut params = inputs.get_mut(state).unwrap();
let mut states = vec![ params.insert(0, last_output);
input.clone(), match execute_machine(&mut part1_input, &mut pos, &mut params) {
input.clone(), Err(output) => last_output = output,
input.clone(), Ok(_) => break,
input.clone(),
input.clone(),
];
let mut halted = false;
for state in (0..5).cycle() {
let mut part1_input = states.get_mut(state).unwrap();
let mut pos = positions.get_mut(state).unwrap();
let mut params = inputs.get_mut(state).unwrap();
params.insert(0, last_output);
loop {
match next_operation(&part1_input, &mut pos, &mut params) {
Some(op) => {
if let Some(o) = execute(op, &mut part1_input, &mut pos) {
last_output = o;
break;
}
}
None => {
halted = true;
break;
}
} }
} }
if halted { last_output
break; })
} .max()
} .unwrap();
last_output
}).max().unwrap();
println!("Part 2: {}", p2); println!("Part 2: {}", p2);
} }
fn execute_machine(tape: &mut Vec<i64>, pos: &mut i64, params: &mut Vec<i64>) -> Result<i64, i64> {
loop {
match next_operation(tape, pos, params) {
Some(op) => {
if let Some(o) = execute(op, tape, pos) {
return Err(o);
}
}
None => {
return Ok((params.pop().unwrap()));
}
}
}
}