Sierpinski三角Pygame递归函数

import pygame

def draw_sierpinski_triangle(screen, color, depth, x1, y1, x2, y2, x3, y3):
    if depth == 0:
        pygame.draw.polygon(screen, color, [(x1, y1), (x2, y2), (x3, y3)], 0)
    else:
        # 计算三个顶点的中点
        mid1_x = (x1 + x2) // 2
        mid1_y = (y1 + y2) // 2
        mid2_x = (x2 + x3) // 2
        mid2_y = (y2 + y3) // 2
        mid3_x = (x3 + x1) // 2
        mid3_y = (y3 + y1) // 2

        # 递归绘制三个小三角形
        draw_sierpinski_triangle(screen, color, depth-1, x1, y1, mid1_x, mid1_y, mid3_x, mid3_y)
        draw_sierpinski_triangle(screen, color, depth-1, mid1_x, mid1_y, x2, y2, mid2_x, mid2_y)
        draw_sierpinski_triangle(screen, color, depth-1, mid3_x, mid3_y, mid2_x, mid2_y, x3, y3)

def main():
    pygame.init()
    width, height = 800, 600
    screen = pygame.display.set_mode((width, height))
    pygame.display.set_caption("Sierpinski Triangle")
    clock = pygame.time.Clock()
    running = True

    while running:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False

        screen.fill((255, 255, 255))
        draw_sierpinski_triangle(screen, (0, 0, 0), 5, 400, 100, 100, 500, 700, 500)
        pygame.display.flip()
        clock.tick(60)

    pygame.quit()

if __name__ == "__main__":
    main()