fonction poids case Connect 4

def poids_cases():
    """Calcule le poids des cases en fonction de la dimension de la grille et du nombre de pions à aligner pour gagner"""
    """[3,4,5,7,5,4,3,4,6,8,10,8,6,4,5,8,11,13,11,8,5,5,8,11,13,11,8,5,4,6,8,10,8,6,4,3,4,5,7,5,4,3] pour une grille 7x6 avec 4 pions à aligner"""
    poids = [0] * WIDTH * HEIGHT
    # Sur les horizontales
    for j in range(HEIGHT):
        for i in range(WIDTH - ALIGNEMENT + 1):
            for k in range(ALIGNEMENT):
                poids[WIDTH * j + i + k] += 1
    # Sur les verticales
    for j in range(HEIGHT - ALIGNEMENT + 1):
        for i in range(WIDTH):
            for k in range(ALIGNEMENT):
                poids[WIDTH * j + i + k * WIDTH] += 1
    # Sur les diagonales montantes
    for j in range(HEIGHT - ALIGNEMENT + 1):
        for i in range(WIDTH - ALIGNEMENT + 1):
            for k in range(ALIGNEMENT):
                poids[WIDTH * j + i + k * WIDTH + k] += 1
    # Sur les diagonales descendantes
    for j in range(ALIGNEMENT - 1, HEIGHT):
        for i in range(WIDTH - ALIGNEMENT + 1):
            for k in range(ALIGNEMENT):
                poids[WIDTH * j + i - k * WIDTH + k] += 1
    return poids