418. Sentence Screen Fitting

Given a rows x cols screen and a sentence represented by a list of non-empty words, find how many times the given sentence can be fitted on the screen.

Note:

  1. A word cannot be split into two lines.
  2. The order of words in the sentence must remain unchanged.
  3. Two consecutive words in a line must be separated by a single space.
  4. Total words in the sentence won't exceed 100.
  5. Length of each word is greater than 0 and won't exceed 10.
  6. 1 ≤ rows, cols ≤ 20,000.

Example 1:

Input:
rows = 2, cols = 8, sentence = ["hello", "world"]

Output: 
1

Explanation:
hello---
world---

The character '-' signifies an empty space on the screen.

Example 2:

Input:
rows = 3, cols = 6, sentence = ["a", "bcd", "e"]

Output: 
2

Explanation:
a-bcd- 
e-a---
bcd-e-

The character '-' signifies an empty space on the screen.

Example 3:

Input:
rows = 4, cols = 5, sentence = ["I", "had", "apple", "pie"]

Output: 
1

Explanation:
I-had
apple
pie-I
had--

The character '-' signifies an empty space on the screen.

Rust Solution

struct Solution;

impl Solution {
    fn words_typing(sentence: Vec<String>, rows: i32, cols: i32) -> i32 {
        let size: Vec<usize> = sentence.into_iter().map(|s| s.len()).collect();
        let n = size.len();
        let sum: usize = size.iter().sum::<usize>() + n;
        let mut i = 0;
        let rows = rows as usize;
        let cols = cols as usize;
        let mut res = 0;
        for _ in 0..rows {
            if size[i] <= cols {
                let m = cols / sum;
                res += m;
                let mut start = m * sum;
                if start + size[i] <= cols {
                    start += size[i];
                    i += 1;
                    if i == n {
                        i = 0;
                        res += 1;
                    }
                } else {
                    continue;
                }
                while start + size[i] < cols {
                    start += size[i] + 1;
                    i += 1;
                    if i == n {
                        i = 0;
                        res += 1;
                    }
                }
            } else {
                break;
            }
        }
        res as i32
    }
}

#[test]
fn test() {
    let rows = 2;
    let cols = 8;
    let sentence = vec_string!["hello", "world"];
    let res = 1;
    assert_eq!(Solution::words_typing(sentence, rows, cols), res);
    let rows = 3;
    let cols = 6;
    let sentence = vec_string!["a", "bcd", "e"];
    let res = 2;
    assert_eq!(Solution::words_typing(sentence, rows, cols), res);
    let rows = 4;
    let cols = 5;
    let sentence = vec_string!["I", "had", "apple", "pie"];
    let res = 1;
    assert_eq!(Solution::words_typing(sentence, rows, cols), res);
}

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