import turtle

def ve_tam_giac(chieu_dai_canh):
    for i in range(3):
        turtle.forward(chieu_dai_canh)
        turtle.left(120)

def ve_hinh_vuong(chieu_dai_canh):
    for i in range(4):
        turtle.forward(chieu_dai_canh)
        turtle.left(90)

def ve_hinh_da_giac(so_canh, chieu_dai_canh):
    goc = 360 / so_canh
    for i in range(so_canh):
        turtle.forward(chieu_dai_canh)
        turtle.left(goc)

def main():
    chieu_dai_canh = 40

    # a. Vẽ tam giác và hình vuông
    ve_tam_giac(chieu_dai_canh)
    turtle.reset()  # Đặt lại vị trí của bút

    ve_hinh_vuong(chieu_dai_canh)
    turtle.reset()  # Đặt lại vị trí của bút

    # b. Vẽ hình đa giác với số cạnh nhập từ bàn phím
    so_canh = int(input("Nhập số cạnh của hình đa giác: "))
    ve_hinh_da_giac(so_canh, chieu_dai_canh)

    turtle.done()

if __name__ == "__main__":
    main()