Pretty plain code. def weightedUniformStrings(s, queries): # Write your code here

c2v = {
    'a': 1,
    'b': 2,
    'c': 3,
    'd': 4,
    'e': 5,
    'f': 6,
    'g': 7,
    'h': 8,
    'i': 9,
    'j': 10,
    'k': 11,
    'l': 12,
    'm': 13,
    'n': 14,
    'o': 15,
    'p': 16,
    'q': 17,
    'r': 18,
    's': 19,
    't': 20,
    'u': 21,
    'v': 22,
    'w': 23,
    'x': 24,
    'y': 25,
    'z': 26
}

collect = set()

carry = []
for si in s:
    if len(carry) == 0 or si != carry[-1]:
        carry = [si]
        collect.add(c2v[si])
    else:
        carry.append(si)
        collect.add(c2v[si]*len(carry))

ans = []
for q in queries:
    if q in collect:
        ans.append('Yes')
    else:
        ans.append('No')

return ans