bert/adventofcode
1
0
Fork 0
mirror of https://github.com/bertptrs/adventofcode.git synced 2025-12-25 12:50:32 +01:00
Code Issues Projects Releases Wiki Activity

Clean up day 24.

Browse Source 
Still doesn't give the right answer, but now it's more readable.
This commit is contained in:
Bert Peters
2018-12-25 14:54:34 +01:00
parent c05b486033
commit 20e7117b6c
1 changed files with 9 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
2018/src/day24.rs
View File

@@ -1,4 +1,3 @@
use std::cmp::Ordering;
use std::cmp::Reverse;
use std::io::BufRead;
use std::io::BufReader;
@@ -43,19 +42,6 @@ impl Group {
}
}
impl Ord for Group {
fn cmp(&self, other: &Self) -> Ordering {
self.effective_power().cmp(&other.effective_power())
.then(other.initiative.cmp(&self.initiative))
}
}
impl PartialOrd for Group {
fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
Some(self.cmp(other))
}
}
#[derive(Default)]
pub struct Day24 {
units: Vec<Group>,
@@ -114,8 +100,7 @@ impl Day24 {
fn simulate(&mut self) -> bool {
let mut order: Vec<usize> = (0..self.units.len()).collect();
order.sort_unstable_by_key(|&x| &self.units[x]);
order.reverse();
order.sort_unstable_by_key(|&x| Reverse((self.units[x].effective_power(), self.units[x].initiative)));
// select targets
let mut targets: Vec<Option<usize>> = vec![None; self.units.len()];
@@ -130,11 +115,7 @@ impl Day24 {
let damage = self.units.iter().map(|x| unit.damage_to(x)).collect_vec();
let target = (0..self.units.len())
.filter(|&x| !is_targeted[x] && self.units[x].faction != unit.faction && self.units[x].is_alive())
.max_by(|&x, &y| {
damage[x].cmp(&damage[y])
.then(self.units[x].effective_power().cmp(&self.units[y].effective_power()))
.then(self.units[x].initiative.cmp(&self.units[y].initiative))
});
.max_by_key(|&x| (damage[x], self.units[x].effective_power(), self.units[x].initiative));
if let Some(target) = target {
targets[i] = Some(target);
@@ -208,23 +189,22 @@ impl Solution for Day24 {
fn part2(&mut self, input: &mut Read) -> String {
self.read_input(input);
let original_count = self.units.iter().map(|x| x.count).collect_vec();
let original = self.units.clone();
loop {
for (unit, count) in self.units.iter_mut().zip(original_count.iter()) {
unit.count = *count;
if unit.faction == 'D' {
unit.power += 1;
}
for boost in 1.. {
self.units = original.clone();
for unit in self.units.iter_mut().filter(|unit| unit.faction == 'D') {
unit.power += boost;
}
println!("{:?}", self.units);
self.full_simulation();
let result: u32 = self.units.iter().map(|x| x.count).sum();
if self.faction_won('D') {
let result: u32 = self.units.iter().map(|x| x.count).sum();
return result.to_string();
}
}
unreachable!();
}
}