【Leetcode】418. Sentence Screen Fitting

记录算法题解

已于 2023-01-29 16:40:51 修改

阅读量401

点赞数

分类专栏： LC 栈、队列、串及其他数据结构文章标签： leetcode 算法 c++

于 2023-01-24 19:27:16 首次发布

本文链接：https://blog.csdn.net/qq_46105170/article/details/128757649

版权

LC 栈、队列、串及其他数据结构专栏收录该内容

678 篇文章

订阅专栏

文章讲述了如何解决LeetCode上的一个问题，即给定单词列表和屏幕尺寸，计算可以填满多少轮。解决方案是使用哈希表存储每个单词作为行首时能填的单词数，然后遍历行来计算总填词数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

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

给定一个长 $n$ 的单词列表，再给定 $r$ 行 $c$ 列的屏幕，我们要将单词列表顺次填写到每一行中，所有单词填完之后再填一轮，如此下去。除了每行第一个单词之外，其余单词填的时候必须前面加一个空格，并且同行两个单词之间最多一个空格。问能填多少轮。

开一个哈希表，存这样的信息：当下标 $i$ 的单词为某一行的第一个单词的时候，这一行能填多少个单词。接下来枚举每一行，如果已经知道了当前行的第一个单词下标，则当前行有多少个单词已经知道了，直接累加单词数再枚举下一行；如果不知道，则暴力求出当前行填多少个单词，并且将信息存入哈希表。最后只需要返回总共的单词数除以 $n$ 即可。代码如下：

class Solution {
 public:
  int wordsTyping(vector<string>& a, int rs, int cs) {
    unordered_map<int, int> mp;
    int c = 0, n = a.size();
    for (int i = 0; i < rs; i++) {
      // 求一下当前行的第一个单词的下标
      int st = c % n;
      if (!mp.count(st)) {
        // 暴力求一下以a[st]开头的行能存多少个单词
        int cnt = 0, len = 0;
        for (int j = st; len < cs; j = (j + 1) % n, cnt++) {
          if (len + a[j].size() > cs) break;
          len += a[j].size() + 1;
        }
        c += cnt;
        mp[st] = cnt;
      } else c += mp[st];
    }
    return c / n;
  }
};

时间复杂度 $O (r n c)$ ，空间 $O (n)$ 。