CSES - Minimikeko

Minimikeko sisältää luvut 1n1 \dots n. Kuinka monessa eri kohdassa luku kk voi sijaita? Voit olettaa, että nn on korkeintaan 100100 ja että 1kn1 \le k \le n.

Toteuta tiedostoon minheap.py funktio count, joka kertoo kuinka monessa eri kohdassa luku kk voi sijaita.

def count(n, k):
    # TODO

if __name__ == "__main__":
    print(count(1,1)) # 1
    print(count(3,2)) # 2
    print(count(5,4)) # 3
    print(count(5,5)) # 3
    print(count(10,9)) # 6
    print(count(70,34)) # 68

Selitys: Tapauksessa n=3n=3 lukua 22 ei voida sijoittaa keon juureen, sillä muuten luku 11 tulisi sjoittaa tämän alapuolelle, mikä rikkoisi minimikekoehdon. Kaikki muut paikat kuitenkin ovat mahdollisia paikkoja luvulle 22.