Medium
Select K Disjoint Special Substrings — Python
Full explanation · Time O(n + 26^3) · Space O(26)
# Time: O(n + 26^3)
# Space: O(26)
# hash table, sort, greedy
class Solution(object):
def maxSubstringLength(self, s, k):
"""
:type s: str
:type k: int
:rtype: bool
"""
def erase_overlap_intervals(intervals):
intervals.sort(key=lambda interval: interval[1])
result, right = 0, float("-inf")
for l, r in intervals:
if l <= right:
result += 1
else:
right = r
return result
cnt = [0]*26
lookup1, lookup2 = [-1]*26, [-1]*26
for i, c in enumerate(s):
cnt[ord(c)-ord('a')] += 1
if lookup1[ord(c)-ord('a')] == -1:
lookup1[ord(c)-ord('a')] = i
lookup2[ord(c)-ord('a')] = i
intervals = []
for i in lookup1:
if i == -1:
continue
for j in lookup2:
if j == -1 or i > j:
continue
total = sum(cnt[c] for c in xrange(len(cnt)) if i <= lookup1[c] <= lookup2[c] <= j)
if total == j-i+1 and total < len(s):
intervals.append((i, j))
return len(intervals)-erase_overlap_intervals(intervals) >= k
# Time: O(26 * n + 26 * log(26))
# Space: O(26)
# hash table, sort, greedy
class Solution2(object):
def maxSubstringLength(self, s, k):
"""
:type s: str
:type k: int
:rtype: bool
"""
def erase_overlap_intervals(intervals):
intervals.sort(key=lambda interval: interval[1])
result, right = 0, float("-inf")
for l, r in intervals:
if l <= right:
result += 1
else:
right = r
return result
cnt = [0]*26
lookup1, lookup2 = [-1]*26, [-1]*26
for i, c in enumerate(s):
cnt[ord(c)-ord('a')] += 1
if lookup1[ord(c)-ord('a')] == -1:
lookup1[ord(c)-ord('a')] = i
lookup2[ord(c)-ord('a')] = i
intervals = []
for i in xrange(len(s)):
if i != lookup1[ord(s[i])-ord('a')]:
continue
x = i+1
j = lookup2[ord(s[i])-ord('a')]
while x <= j and lookup1[ord(s[x])-ord('a')] >= i:
j = max(j, lookup2[ord(s[x])-ord('a')])
x += 1
if x == j+1 and (i != 0 or j != len(s)-1):
intervals.append((i, j))
return len(intervals)-erase_overlap_intervals(intervals) >= k