Add problem 849: Maximize Distance to Closest Person

Author: EFanZh

src/lib.rs

@@ -693,6 +693,7 @@ pub mod problem_0844_backspace_string_compare;
 pub mod problem_0845_longest_mountain_in_array;
 pub mod problem_0846_hand_of_straights;
 pub mod problem_0848_shifting_letters;
+pub mod problem_0849_maximize_distance_to_closest_person;
 pub mod problem_0867_transpose_matrix;
 pub mod problem_1143_longest_common_subsequence;
 pub mod problem_1192_critical_connections_in_a_network;

src/problem_0849_maximize_distance_to_closest_person/iterative.rs

@@ -0,0 +1,55 @@
+pub struct Solution;
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl Solution {
+    pub fn max_dist_to_closest(seats: Vec<i32>) -> i32 {
+        let i = seats.iter().position(|&x| x != 0).unwrap();
+        let mut iter = seats[i + 1..].iter().map(|&x| x != 0);
+        let mut result = i as i32;
+
+        loop {
+            loop {
+                if let Some(value) = iter.next() {
+                    if !value {
+                        break;
+                    }
+                } else {
+                    return result;
+                }
+            }
+
+            let mut length = 1;
+
+            loop {
+                if let Some(value) = iter.next() {
+                    if value {
+                        result = result.max((length + 1) / 2);
+
+                        break;
+                    }
+
+                    length += 1;
+                } else {
+                    return result.max(length);
+                }
+            }
+        }
+    }
+}
+
+// ------------------------------------------------------ snip ------------------------------------------------------ //
+
+impl super::Solution for Solution {
+    fn max_dist_to_closest(seats: Vec<i32>) -> i32 {
+        Self::max_dist_to_closest(seats)
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    #[test]
+    fn test_solution() {
+        super::super::tests::run::<super::Solution>();
+    }
+}

src/problem_0849_maximize_distance_to_closest_person/mod.rs

@@ -0,0 +1,23 @@
+pub mod iterative;
+
+pub trait Solution {
+    fn max_dist_to_closest(seats: Vec<i32>) -> i32;
+}
+
+#[cfg(test)]
+mod tests {
+    use super::Solution;
+
+    pub fn run<S: Solution>() {
+        let test_cases = [
+            (&[1, 0, 0, 0, 1, 0, 1] as &[_], 2),
+            (&[1, 0, 0, 0], 3),
+            (&[0, 1], 1),
+            (&[1, 1, 0, 1, 1], 1),
+        ];
+
+        for (seats, expected) in test_cases {
+            assert_eq!(S::max_dist_to_closest(seats.to_vec()), expected);
+        }
+    }
+}