求告知C语言难题啊，

设计一个菜单程序，选择数字键1~3分别在屏幕中间输出直角三角形、长方形和正方形，选择数字4结束操作 要用C语言实现

大概是下面这个流程，至于绘制图形，你写三个函数，放在相应的位置就好了。

int select = 0;
while(1){
  scanf("%d",&select);
  if(select == 1){
    printf("三角形");
  }else if(select == 2){
     printf("长方形");
  }else if(select == 3){
     printf("正方形");
  } else{
    break;
  }
}

#include <stdio.h>

void print_right_triangle() {
    for (int i = 0; i < 5; i++) {
        for (int j = 0; j <= i; j++) {
            printf("*");
        }
        printf("\n");
    }
}

void print_rectangle() {
    for (int i = 1; i <= 5; i++) {
        for (int j = 1; j <= 7; j++) {
            printf(" *");
        }
        printf("\n");
    }
}

void print_square() {
    for (int i = 1; i <= 6; i++){
        for (int j = 1; j <= 6; j++){
            printf("* ");
        }
        printf("\n");
    }
}

int main() {
    printf("1:直角三角形 | 2:长方形 | 3:正方形 | 4:退出\n");

    int menu = 0;
    while (1) {
        printf("请选择菜单：\n");
        scanf("%d", &menu);
        if (menu == 1) {
            print_right_triangle();
        }
        else if (menu == 2) {
            print_rectangle();
        }
        else if (menu == 3) {
            print_square();
        }
        else {
            break;
        }
    }

}