Find unique substring of String s with given length

Question

I am trying to solve this question. For example, the string equals "caaab" and the length K=2, the return substrings should be "aa","ab","ca" in an alphabetic order. I prefer the solution in python

Answer 1

Try this:

input_str = 'caaab'
k = 2

unique_substrings = set(input_str[i: i + k] for i in range(len(input_str) - k + 1))
sorted_substrings = sorted(unique_substrings)

print(sorted_substrings)