899. Orderly Queue

A string S of lowercase letters is given.  Then, we may make any number of moves.

In each move, we choose one of the first K letters (starting from the left), remove it, and place it at the end of the string.

Return the lexicographically smallest string we could have after any number of moves.

 

Example 1:

Input: S = "cba", K = 1
Output: "acb"
Explanation: 
In the first move, we move the 1st character ("c") to the end, obtaining the string "bac".
In the second move, we move the 1st character ("b") to the end, obtaining the final result "acb".

Example 2:

Input: S = "baaca", K = 3
Output: "aaabc"
Explanation: 
In the first move, we move the 1st character ("b") to the end, obtaining the string "aacab".
In the second move, we move the 3rd character ("c") to the end, obtaining the final result "aaabc".

 

Note:

  1. 1 <= K <= S.length <= 1000
  2. S consists of lowercase letters only.

Rust Solution

struct Solution;

impl Solution {
    fn orderly_queue(s: String, k: i32) -> String {
        let mut s: Vec<char> = s.chars().collect();
        let n = s.len();
        if k > 1 {
            s.sort_unstable();
            s.into_iter().collect()
        } else {
            let mut res = s.to_vec();
            for i in 0..n {
                let mut t = vec![];
                for j in i..n {
                    t.push(s[j]);
                }
                for j in 0..i {
                    t.push(s[j]);
                }
                if t < res {
                    res = t;
                }
            }
            res.into_iter().collect()
        }
    }
}

#[test]
fn test() {
    let s = "cba".to_string();
    let k = 1;
    let res = "acb".to_string();
    assert_eq!(Solution::orderly_queue(s, k), res);
    let s = "baaca".to_string();
    let k = 3;
    let res = "aaabc".to_string();
    assert_eq!(Solution::orderly_queue(s, k), res);
}

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