substring-with-k-chars

Author: avidLearnerInProgress

count_substrings_k_distinct_characters.py

@@ -0,0 +1,26 @@
+def substr(mystr, k):
+    if mystr == None or k == 0:
+        return 0
+    ls = list()
+    for i in range(len(mystr) - (k-1)):
+        ls.append(mystr[i:i+k]) # set to remove duplicate k length strings
+    
+    res = set()
+    for ele in ls:
+        if len(set(ele)) == k:
+            res.add(ele)
+    return len(res)
+    
+    #ans = set() # set to eliminate strings having duplicate characters
+    #xt = list(st)
+    #for ele in xt:
+    #    if len(set(ele)) == k:
+    #        ans.add(ele)
+    #return len(ans)
+
+n = int(input())
+for i in range(n):
+    mystr = str(input())
+    k = int(input())
+    print(substr(mystr, k))
+