Osajonot

Tehtäväsi on laskea, monessako merkkijonon osajonossa kaikki merkit ovat samoja.

Voit olettaa, että merkkijono muodostuu merkeistä a–z ja siinä on enintään 10^5 merkkiä. Tavoitteena on, että algoritmin aikavaativuus on O(n).

Toteuta tiedostoon substrings.py funktio count, joka palauttaa halutun tuloksen.

def count(s):
    # TODO

if __name__ == "__main__":
    print(count("aaa")) # 6
    print(count("saippuakauppias")) # 17
    print(count("x")) # 1
    print(count("aybabtu")) # 7