From 6acc2899beb0f91dd9a08bf72f610f0a0aa2b234 Mon Sep 17 00:00:00 2001
From: Alpha Chen <alpha.chen@gmail.com>
Date: Sat, 2 Dec 2017 12:37:53 -0800
Subject: [PATCH] [2017][rust][1.1]

---
 2017/rust/src/day_01.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/2017/rust/src/day_01.rs b/2017/rust/src/day_01.rs
index 12b3a14..875e5a1 100644
--- a/2017/rust/src/day_01.rs
+++ b/2017/rust/src/day_01.rs
@@ -8,8 +8,9 @@ pub fn solve(input: &str) -> Result<String, Error> {
         .collect::<Result<_, _>>()?;
 
     // 1
-    let offset = 1;
-    let offset_iter = input.iter().skip(offset).cycle().take(input.len());
+    let offset = input.len() / 2;
+
+    let offset_iter = input.iter().cycle().skip(offset).take(input.len());
     Ok(
         input
             .iter()
@@ -20,3 +21,12 @@ pub fn solve(input: &str) -> Result<String, Error> {
             .to_string(),
     )
 }
+
+#[test]
+fn part_two() {
+    assert_eq!(solve("1212").unwrap(), "6".to_string());
+    assert_eq!(solve("1221").unwrap(), "0".to_string());
+    assert_eq!(solve("123425").unwrap(), "4".to_string());
+    assert_eq!(solve("123123").unwrap(), "12".to_string());
+    assert_eq!(solve("12131415").unwrap(), "4".to_string());
+}