comments | difficulty | edit_url | tags | ||
---|---|---|---|---|---|
true |
简单 |
|
我们要把给定的字符串 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]
。
我们定义两个变量 lines
和 last
,分别表示行数和最后一行的宽度,初始时 lines = 1
,last = 0
。
遍历字符串 last
上,否则行数 lines
加一,并且 last
重置为
最后返回 lines
和 last
构成的数组。
时间复杂度
class Solution:
def numberOfLines(self, widths: List[int], s: str) -> List[int]:
lines, last = 1, 0
for w in map(lambda c: widths[ord(c) - ord("a")], s):
if last + w <= 100:
last += w
else:
lines += 1
last = w
return [lines, last]
class Solution {
public int[] numberOfLines(int[] widths, String s) {
int lines = 1, last = 0;
for (int i = 0; i < s.length(); ++i) {
int w = widths[s.charAt(i) - 'a'];
if (last + w <= 100) {
last += w;
} else {
++lines;
last = w;
}
}
return new int[] {lines, last};
}
}
class Solution {
public:
vector<int> numberOfLines(vector<int>& widths, string s) {
int lines = 1, last = 0;
for (char c : s) {
int w = widths[c - 'a'];
if (last + w <= 100) {
last += w;
} else {
++lines;
last = w;
}
}
return {lines, last};
}
};
func numberOfLines(widths []int, s string) []int {
lines, last := 1, 0
for _, c := range s {
w := widths[c-'a']
if last+w <= 100 {
last += w
} else {
lines++
last = w
}
}
return []int{lines, last}
}
function numberOfLines(widths: number[], s: string): number[] {
let [lines, last] = [1, 0];
for (const c of s) {
const w = widths[c.charCodeAt(0) - 'a'.charCodeAt(0)];
if (last + w <= 100) {
last += w;
} else {
++lines;
last = w;
}
}
return [lines, last];
}
impl Solution {
pub fn number_of_lines(widths: Vec<i32>, s: String) -> Vec<i32> {
let mut lines = 1;
let mut last = 0;
for c in s.chars() {
let idx = ((c as u8) - b'a') as usize;
let w = widths[idx];
if last + w <= 100 {
last += w;
} else {
lines += 1;
last = w;
}
}
vec![lines, last]
}
}