Commit 9325f2f

authored

Create Weighted-Uniform-Strings.py

1 parent 7242bb6 commit 9325f2fCopy full SHA for 9325f2f

File tree

+47

-0

lines changed

+47

-0

lines changed

Lines changed: 47 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,47 @@`
	`1`	`+#!/bin/python3`
	`2`	`+`
	`3`	`+import os`
	`4`	`+import random`
	`5`	`+import sys`
	`6`	`+`
	`7`	`+# Complete the weightedUniformStrings function below.`
	`8`	`+def weightedUniformStrings(s, queries):`
	`9`	`+ w = set()`
	`10`	`+ prev = -1`
	`11`	`+ l = 0`
	`12`	`+ for c in s:`
	`13`	`+ x = ord(c) - ord('a') + 1`
	`14`	`+ w.add(x)`
	`15`	`+ if prev == c:`
	`16`	`+ l += 1`
	`17`	`+ w.add(l*x)`
	`18`	`+ else:`
	`19`	`+ prev = c`
	`20`	`+ l = 1`
	`21`	`+`
	`22`	`+ ans = []`
	`23`	`+ for q in queries:`
	`24`	`+ if q in w:`
	`25`	`+ ans.append("Yes")`
	`26`	`+ else:`
	`27`	`+ ans.append("No")`
	`28`	`+ return ans`
	`29`	`+if __name__ == '__main__':`
	`30`	`+ fptr = open(os.environ['OUTPUT_PATH'], 'w')`
	`31`	`+`
	`32`	`+ s = input()`
	`33`	`+`
	`34`	`+ queries_count = int(input())`
	`35`	`+`
	`36`	`+ queries = []`
	`37`	`+`
	`38`	`+ for _ in range(queries_count):`
	`39`	`+ queries_item = int(input())`
	`40`	`+ queries.append(queries_item)`
	`41`	`+`
	`42`	`+ result = weightedUniformStrings(s, queries)`
	`43`	`+`
	`44`	`+ fptr.write('\n'.join(result))`
	`45`	`+ fptr.write('\n')`
	`46`	`+`
	`47`	`+ fptr.close()`

Comments

(0)