LeetCode 763. 划分字母区间(C++、python)

 

 

---

LeetCode 763. 划分字母区间

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

示例 1:

输入: S = "ababcbacadefegdehijhklij"
输出: [9,7,8]
解释:
划分结果为 "ababcbaca", "defegde", "hijhklij"。
每个字母最多出现在一个片段中。
像 "ababcbacadefegde", "hijhklij" 的划分是错误的，因为划分的片段数较少。

代码如下：

C++

class Solution {
public:
    vector<int> partitionLabels(string S) 
    {
        int n=S.length();
        map<char,int> tmp;
        for(int i=0;i<n;i++)
        {
            tmp[S[i]]++;
        }
        vector<int> res;
        set<int> st;
        int start=-1;
        for(int i=0;i<n;i++)
        {
        	st.insert(S[i]);
            tmp[S[i]]--;
            if(0==tmp[S[i]])
            {
            	st.erase(S[i]);
                if(0==st.size())
                {
                    res.push_back(i-start);
                    start=i;
                }
            }
        }
        return res;
    }
};

class Solution:
    def partitionLabels(self, S):
        """
        :type S: str
        :rtype: List[int]
        """
        res=[]
        vec=[]
        for i in range(0,26):
            vec.append(0)
        last = 0
        for i in range(0,len(S)):
            vec[ord(S[i])-ord('a')] += 1
        st = set()
        for i in range(0,len(S)):
            st.add(S[i])
            vec[ord(S[i])-ord('a')] -= 1
            if vec[ord(S[i])-ord('a')]==0:
                if S[i] in st:
                    st.remove(S[i])
                if st==set():
                    res.append(i-last+1)
                    last=i+1
        return res