Leetcode 763 划分字母区间
问题重述
字符串 S 由小写字母组成。我们要把这个字符串划分为尽可能多的片段,同一字母最多出现在一个片段中。返回一个表示每个字符串片段的长度的列表。
示例:
输入:S = “ababcbacadefegdehijhklij” 输出:[9,7,8]
解释:划分结果为 “ababcbaca”, “defegde”, “hijhklij”。每个字母最多出现在一个片段中。像 “ababcbacadefegde”, “hijhklij” 的划分是错误的,因为划分的片段数较少。
思路
只需要一个大小为26的数组last,通过一次循环记下每一个字符出现的末尾位置。即last[S(i) – ‘a] = i;即数组第一个表示字符a,第二个表示b等等。
for (int i=0; i < n; i++) {
last[S[i] - 'a'] = i;
}
所以第二个循环记下,start和end两个变量。
比如S的第一个字符是a,a出现的start是 1 , end是10.第二个字符是b,b出现的 start 是 2,end是11.那么这个时候,end就需要更新为11.一次类推一直到I 等于 end,
for (int i=0; i < n; i++) {
end = max(end, last[S[i] - 'a']);
if (i == end) {
answer.push_back(end - start + 1);
start = end + 1;
}
}
说明满足结果,这个串包含的字符在其他串中不出现,因为start和end都在这里面。
例子:abcabcdede,初始start=end=0;第一个字符a,end更新为4.第二个字符b,end更新为5,第三个字符c,end更新为6.一直到i=6的时候都没有更新。此时,i == end,所以前六个字符是一个串。
最终代码
class Solution {
public:
vector<int> partitionLabels(string S) {
int n = S.size();
int last[26];
for (int i=0; i < n; i++) {
last[S[i] - 'a'] = i;
}
vector<int> answer;
int start = 0, end = 0;
for (int i=0; i < n; i++) {
end = max(end, last[S[i] - 'a']);
if (i == end) {
answer.push_back(end - start + 1);
start = end + 1;
}
}
return answer;
}
};