probability thing

main.py

import random
import sys


def main(width: int, mj: int, iters=1000000):
    x = [0] * (width + (mj) - 1)
    for _ in range(iters):
        mins = random.randrange(width)
        jitter = random.randrange(mj)

        final = mins + jitter
        x[final] += 1
    
    for num in x:
        print(num)

    y = x[mj:width]
    frontadd = x[mj-1::-1]
    backadd = x[-1:width-2:-1]

    for i in range(len(frontadd)):
        y[i] += frontadd[i]
    
    for i in range(len(backadd)):
        yi = (width - (mj*2) + i)
        y[yi] += backadd[i]
    
    print("\n\n")
    for num in y:
        print(num)
    

if __name__ == "__main__":
    width = int(sys.argv[1])
    mj = int(sys.argv[2])
    main(width, mj)