Alpha Chen
7 years ago
parent
bb8f1c5da3
commit
71575ca08a
@ -0,0 +1,90 @@
|
||||
use std::convert::TryFrom;
|
||||
use failure::*;
|
||||
use regex::Regex;
|
||||
|
||||
pub fn solve(input: &str) -> Result<String, Error> {
|
||||
// let input = "s1,x3/4,pe/b";
|
||||
let moves: Vec<_> = input
|
||||
.split(',')
|
||||
.map(Move::try_from)
|
||||
.collect::<Result<_, _>>()?;
|
||||
let mut programs = (b'a'..=b'p').map(|c| c as char).collect::<Vec<_>>();
|
||||
dance(&moves, &mut programs);
|
||||
Ok(programs.iter().collect())
|
||||
}
|
||||
|
||||
fn dance(moves: &[Move], programs: &mut [char]) {
|
||||
for m in moves {
|
||||
match m {
|
||||
Move::Spin(size) => {
|
||||
let len = programs.len();
|
||||
programs.rotate(len - size);
|
||||
}
|
||||
Move::Exchange(a, b) => {
|
||||
programs.swap(*a, *b);
|
||||
}
|
||||
Move::Partner(a, b) => {
|
||||
let a = programs.iter().position(|x| x == a).unwrap();
|
||||
let b = programs.iter().position(|x| x == b).unwrap();
|
||||
programs.swap(a, b);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_dance() {
|
||||
let moves = vec![Move::Spin(1), Move::Exchange(3, 4), Move::Partner('e', 'b')];
|
||||
let mut programs: Vec<_> = (b'a'..=b'e').map(|c| c as char).collect();
|
||||
|
||||
dance(&moves, &mut programs);
|
||||
|
||||
assert_eq!(programs, vec!['b', 'a', 'e', 'd', 'c']);
|
||||
}
|
||||
|
||||
#[derive(Debug, Eq, PartialEq)]
|
||||
enum Move {
|
||||
Spin(usize),
|
||||
Exchange(usize, usize),
|
||||
Partner(char, char),
|
||||
}
|
||||
|
||||
lazy_static! {
|
||||
static ref RE_S: Regex = { Regex::new(r"s(\d+)").unwrap() };
|
||||
static ref RE_X: Regex = { Regex::new(r"x(\d+)/(\d+)").unwrap() };
|
||||
static ref RE_P: Regex = { Regex::new(r"p(\w)/(\w)").unwrap() };
|
||||
}
|
||||
|
||||
impl<'a> TryFrom<&'a str> for Move {
|
||||
type Error = Error;
|
||||
|
||||
fn try_from(value: &str) -> Result<Self, Self::Error> {
|
||||
RE_S.captures(value)
|
||||
.and_then(|c| {
|
||||
let size = c.get(1).unwrap().as_str().parse().ok()?;
|
||||
Some(Move::Spin(size))
|
||||
})
|
||||
.or_else(|| {
|
||||
RE_X.captures(value).and_then(|c| {
|
||||
let a = c.get(1).unwrap().as_str().parse().ok()?;
|
||||
let b = c.get(2).unwrap().as_str().parse().ok()?;
|
||||
Some(Move::Exchange(a, b))
|
||||
})
|
||||
})
|
||||
.or_else(|| {
|
||||
RE_P.captures(value).and_then(|c| {
|
||||
let a = c.get(1).unwrap().as_str().chars().next()?;
|
||||
let b = c.get(2).unwrap().as_str().chars().next()?;
|
||||
Some(Move::Partner(a, b))
|
||||
})
|
||||
})
|
||||
.ok_or_else(|| format_err!("invalid move: {}", value))
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_move_try_from() {
|
||||
assert_eq!(Move::try_from("s1").unwrap(), Move::Spin(1));
|
||||
assert_eq!(Move::try_from("x3/4").unwrap(), Move::Exchange(3, 4));
|
||||
assert_eq!(Move::try_from("pe/b").unwrap(), Move::Partner('e', 'b'));
|
||||
}
|
||||
Loading…
Reference in new issue