2018: day03: implement FromStr instead of fn parse

This commit is contained in:
Antoine Martin 2019-11-18 14:27:45 +01:00
parent 8de2538d0e
commit a6ff14b260

View file

@ -1,6 +1,7 @@
use std::collections::HashMap; use std::collections::HashMap;
use std::collections::HashSet; use std::collections::HashSet;
use std::error::Error; use std::error::Error;
use std::str::FromStr;
use super::err; use super::err;
use super::Result; use super::Result;
@ -22,42 +23,48 @@ struct Claim {
id: usize, id: usize,
} }
/// Parses a claim from a line impl FromStr for Claim {
/// type Err = Box<dyn Error>;
/// Fails if the line is badly formatted. The expected format is:
///
/// ```text
/// #ID @ X,Y: WxH
/// ```
fn parse(line: &str) -> Option<Claim> {
// skip '#' in line
let line = &line[1..];
// find ' @ ' separator /// Parses a claim from a line
let at = line.find(" @ ")?; ///
let id = line[..at].parse().ok()?; /// Fails if the line is badly formatted. The expected format is:
let line = &line[(at + 3)..]; ///
/// ```text
/// #ID @ X,Y: WxH
/// ```
fn from_str(s: &str) -> Result<Self> {
// skip '#' in line
let s = &s[1..];
// parse 'X,Y: WxH // find ' @ ' separator
let comma = line.find(',')?; let at = s
let colon = line.find(':')?; .find(" @ ")
let x = line[..comma].parse().ok()?; .ok_or_else(|| err!("` @ ` delimiter not found"))?;
let y = line[(comma + 1)..colon].parse().ok()?; let id = s[..at].parse()?;
let s = &s[(at + 3)..];
// reduce line to 'WxH' // parse 'X,Y: WxH
let line = &line[(colon + 2)..]; let comma = s.find(',').ok_or_else(|| err!("`,` delimiter not found"))?;
let colon = s.find(':').ok_or_else(|| err!("`:` delimiter not found"))?;
let x = s[..comma].parse()?;
let y = s[(comma + 1)..colon].parse()?;
let sep = line.find('x')?; // reduce line to 'WxH'
let width = line[..sep].parse().ok()?; let s = &s[(colon + 2)..];
let height = line[(sep + 1)..].parse().ok()?;
Some(Claim { let sep = s.find('x').ok_or_else(|| err!("`x` delimiter not found"))?;
x, let width = s[..sep].parse()?;
y, let height = s[(sep + 1)..].parse()?;
width,
height, Ok(Claim {
id, x,
}) y,
width,
height,
id,
})
}
} }
fn part1(input: &str) -> Result<u64> { fn part1(input: &str) -> Result<u64> {
@ -65,7 +72,9 @@ fn part1(input: &str) -> Result<u64> {
let mut map: HashMap<(usize, usize), u64> = HashMap::new(); let mut map: HashMap<(usize, usize), u64> = HashMap::new();
for line in input.lines() { for line in input.lines() {
let claim = parse(line).ok_or_else(|| err!("Couldn't parse line: {}", line))?; let claim: Claim = line
.parse()
.or_else(|e| Err(err!("couldn't parse line: `{}`, {}", line, e)))?;
for i in 0..claim.width { for i in 0..claim.width {
for j in 0..claim.height { for j in 0..claim.height {
@ -92,7 +101,9 @@ fn part2(input: &str) -> Result<usize> {
let mut set = HashSet::new(); let mut set = HashSet::new();
for line in input.lines() { for line in input.lines() {
let claim = parse(line).ok_or_else(|| err!("Couldn't parse line: {}", line))?; let claim: Claim = line
.parse()
.or_else(|e| Err(err!("couldn't parse line: `{}`, {}", line, e)))?;
set.insert(claim.id); set.insert(claim.id);
for i in 0..claim.width { for i in 0..claim.width {