806. 写字符串需要的行数 : 简单模拟题

一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第12天，点击查看活动详情。

题目描述

这是 LeetCode 上的 806. 写字符串需要的行数 ，难度为 简单。

Tag : 「模拟」

我们要把给定的字符串 S 从左到右写到每一行上，每一行的最大宽度为 $100$ 个单位，如果我们在写某个字母的时候会使这行超过了 $100$ 个单位，那么我们应该把这个字母写到下一行。

我们给定了一个数组 widths，这个数组  $widths[0]$ 代表 'a' 需要的单位，  $widths[1]$ 代表 'b' 需要的单位，...，  $widths[25]$ 代表 'z' 需要的单位。

现在回答两个问题：至少多少行能放下 S，以及最后一行使用的宽度是多少个单位？

将你的答案作为长度为 $2$ 的整数列表返回。

示例 1:

输入: 
widths = [10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10]
S = "abcdefghijklmnopqrstuvwxyz"

输出: [3, 60]

解释: 
所有的字符拥有相同的占用单位10。所以书写所有的26个字母，
我们需要2个整行和占用60个单位的一行。
复制代码

示例 2:

输入: 
widths = [4,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10,10]
S = "bbbcccdddaaa"

输出: [2, 4]

解释: 
除去字母'a'所有的字符都是相同的单位10，并且字符串 "bbbcccdddaa" 将会覆盖 9 * 10 + 2 * 4 = 98 个单位.
最后一个字母 'a' 将会被写到第二行，因为第一行只剩下2个单位了。
所以，这个答案是2行，第二行有4个单位宽度。
复制代码

注:

• 字符串 S 的长度在  $[1, 1000]$ 的范围。
• S 只包含小写字母。
• widths 是长度为  $26$ 的数组。
• $widths[i]$ 值的范围在  $[2, 10]$。

模拟

根据题意进行模拟即可。

使用变量 a 代指当前有多少行是满的，使用变量 b 代指当前填充光标所在的位置。

代码：

class Solution {
    public int[] numberOfLines(int[] widths, String s) {
        int a = 0, b = 0;
        for (char c : s.toCharArray()) {
            int t = widths[c - 'a'];
            if (b + t > 100 && ++a >= 0) b = t;
            else b += t;         
        }
        if (b != 0) a++;
        return new int[]{a, b};
    }
}
复制代码

• 时间复杂度： $O(n)$
• 空间复杂度：不使用 toCharArray 为 $O(1)$，否则为 $O(n)$

最后

这是我们「刷穿 LeetCode」系列文章的第 No.804 篇，系列开始于 2021/01/01，截止于起始日 LeetCode 上共有 1916 道题目，部分是有锁题，我们将先把所有不带锁的题目刷完。

在这个系列文章里面，除了讲解解题思路以外，还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码，我建立了相关的仓库：github.com/SharingSour… 。

在仓库地址里，你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。