Problem 418: Sentence Screen Fitting

https://leetcode.com/problems/sentence-screen-fitting/

思路

复杂度

• Time

  $O(m + n)$

  其中 m 是 rows，n 是 cols

public class Solution {
    public int wordsTyping(String[] sentence, int rows, int cols) {
        String s = String.join(" ", sentence) + " ";
        int len = s.length(), count = 0;
        int[] index = new int[len];

        for (int i = 1; i < len; i++) {
            index[i] = s.charAt(i) == ' ' ? 1 : index[i - 1] - 1;
        }
        for (int i = 0; i < rows; i++) {
            count += cols;
            count += index[count % len];
        }

        return count / len;
    }
}

易错点

1. index[] 的计数是从 1 开始的

   for (int i = 1; i < len; i++) {
        index[i] = s.charAt(i) == ' ' ? 1 : index[i - 1] - 1;
   }

2. String join 函数

   String.join(" ", sentence)

   join 函数实际上，第一个 parameter 是分隔符，比如 “-”，这里我们用的是 space。也就是说，字符串数组里的元素会被 space 连接起来。