249. Group Shifted Strings

Given a string, we can "shift" each of its letter to its successive letter, for example: "abc" -> "bcd". We can keep "shifting" which forms the sequence:

"abc" -> "bcd" -> ... -> "xyz"

Given a list of non-empty strings which contains only lowercase alphabets, group all strings that belong to the same shifting sequence.


Input: ["abc", "bcd", "acef", "xyz", "az", "ba", "a", "z"],

Rust Solution

struct Solution;
use std::collections::HashMap;

impl Solution {
    fn group_strings(strings: Vec<String>) -> Vec<Vec<String>> {
        let mut hm: HashMap<Vec<u8>, Vec<String>> = HashMap::new();
        for s in strings {
            let mut v: Vec<u8> = s.bytes().collect();
            v = v.iter().map(|b| (b + 26 - v[0]) % 26).collect();
        hm.into_iter().map(|(_, v)| v).collect()

fn test() {
    let strings = vec_string!["abc", "bcd", "acef", "xyz", "az", "ba", "a", "z"];
    let mut res = vec_vec_string![["abc", "bcd", "xyz"], ["az", "ba"], ["acef"], ["a", "z"]];
    let mut ans = Solution::group_strings(strings);
    assert_eq!(ans, res);

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