763. 划分字母区间

原题

字符串 S 由小写字母组成。我们要把这个字符串划分为尽可能多的片段，同一个字母只会出现在其中的一个片段。返回一个表示每个字符串片段的长度的列表。

示例 1：

输入：S = “ababcbacadefegdehijhklij”

输出：[9,7,8]

解释：

划分结果为 “ababcbaca”, “defegde”, “hijhklij”。

每个字母最多出现在一个片段中。

像 “ababcbacadefegde”, “hijhklij” 的划分是错误的，因为划分的片段数较少。

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/partition-labels

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

提示：

• S的长度在[1, 500]之间。
• S只包含小写字母 'a' 到 'z'。

思路

利用 unordered_map 记录每个字母的最远位置，接着遍历整个字符串，维护一个起始距离 $low$ 和最远的距离 $high$，如果 $i == high $ 说明该区间结束，更新 $low$ 。

代码

• C++

class Solution {
public:
    vector<int> partitionLabels(string S) {
        int size = S.size();
        vector<int> ans;
        unordered_map<char, int> mp;
        for (int i = 0; i < size; ++i) {
            // 记录每个单词最后的位置
            mp[S[i]] = i;
        }
        int low = 0, high = 0;
        for (int i = 0; i < size; ++i) {
            high = max(high, mp[S[i]]);
            if (i == high) {
                // 如果 i == high 说明到了最后一个位置
                ans.emplace_back(high - low + 1);
                // 更新low
                low = i + 1;
            }
        }
        return ans;
    }
};

• Python

class Solution:
    def partitionLabels(self, S: str) -> List[int]:
        mp = {}
        ans = []
        for i in range(0, len(S)):
            mp[S[i]] = i
        low, high = 0, 0
        for i in range(0, len(S)):
            high = max(high, mp[S[i]])
            if i == high:
                ans.append(high - low + 1)
                low = i + 1
        return ans