816. Ambiguous Coordinates

We had some 2-dimensional coordinates, like "(1, 3)" or "(2, 0.5)".  Then, we removed all commas, decimal points, and spaces, and ended up with the string S.  Return a list of strings representing all possibilities for what our original coordinates could have been.

Our original representation never had extraneous zeroes, so we never started with numbers like "00", "0.0", "0.00", "1.0", "001", "00.01", or any other number that can be represented with less digits.  Also, a decimal point within a number never occurs without at least one digit occuring before it, so we never started with numbers like ".1".

The final answer list can be returned in any order.  Also note that all coordinates in the final answer have exactly one space between them (occurring after the comma.)

Example 1:
Input: "(123)"
Output: ["(1, 23)", "(12, 3)", "(1.2, 3)", "(1, 2.3)"]
Example 2:
Input: "(00011)"
Output:  ["(0.001, 1)", "(0, 0.011)"]
Explanation: 
0.0, 00, 0001 or 00.01 are not allowed.
Example 3:
Input: "(0123)"
Output: ["(0, 123)", "(0, 12.3)", "(0, 1.23)", "(0.1, 23)", "(0.1, 2.3)", "(0.12, 3)"]
Example 4:
Input: "(100)"
Output: [(10, 0)]
Explanation: 
1.0 is not allowed.

 

Note:

  • 4 <= S.length <= 12.
  • S[0] = "(", S[S.length - 1] = ")", and the other elements in S are digits.

 

Rust Solution

struct Solution;
use std::iter::FromIterator;

trait Nums {
    fn nums(self) -> Vec<String>;
}

impl Nums for &[char] {
    fn nums(self) -> Vec<String> {
        let n = self.len();
        let mut res = vec![];
        for i in 1..=n {
            let left = String::from_iter(self[..i].iter());
            let right = String::from_iter(self[i..].iter());
            if left.starts_with('0') && left != "0" {
                continue;
            }
            if right.ends_with('0') {
                continue;
            }
            res.push(format!(
                "{}{}{}",
                left,
                if i == n { "" } else { "." },
                right
            ));
        }
        res
    }
}

impl Solution {
    fn ambiguous_coordinates(s: String) -> Vec<String> {
        let s: Vec<char> = s.chars().collect();
        let n = s.len();
        let mut res = vec![];
        for i in 2..n - 1 {
            let left = &s[1..i];
            let right = &s[i..n - 1];
            for l in left.nums() {
                for r in right.nums() {
                    res.push(format!("({}, {})", l, r));
                }
            }
        }
        res
    }
}

#[test]
fn test() {
    let s = "(123)".to_string();
    let mut res = vec_string!["(1, 23)", "(12, 3)", "(1.2, 3)", "(1, 2.3)"];
    let mut ans = Solution::ambiguous_coordinates(s);
    res.sort();
    ans.sort();
    assert_eq!(ans, res);
    let s = "(00011)".to_string();
    let mut res = vec_string!["(0.001, 1)", "(0, 0.011)"];
    let mut ans = Solution::ambiguous_coordinates(s);
    res.sort();
    ans.sort();
    assert_eq!(ans, res);
    let s = "(0123)".to_string();
    let mut res = vec_string![
        "(0, 123)",
        "(0, 12.3)",
        "(0, 1.23)",
        "(0.1, 23)",
        "(0.1, 2.3)",
        "(0.12, 3)"
    ];
    let mut ans = Solution::ambiguous_coordinates(s);
    res.sort();
    ans.sort();
    assert_eq!(ans, res);
    let s = "(100)".to_string();
    let mut res = vec_string!["(10, 0)"];
    let mut ans = Solution::ambiguous_coordinates(s);
    res.sort();
    ans.sort();
    assert_eq!(ans, res);
}

Having problems with this solution? Click here to submit an issue on github.