任何用c语言把下面这个问题放在main函数里面循环

在每次得出结果后，输出提示，问用户是不是还继续，根据输入循环游戏。如：还想再来一次吗？(y/n)；当用户输入‘Y’或者‘y’时，再来一次，否则退出游戏。

#include <stdio.h>
int main(){
    char ans[1];
    while (1){
        printf("还想再来一次吗？(y/n):");
        gets(ans); 
        if ( (ans[0] == 'y' || ans[0] == 'Y') ){
        }else{
            return 0;
        } 
    }
} 

int main()
{
   char ch;
   int game = 1;
   while (game)
   {
      printf("游戏中..\n");
      while (1)
      {
         printf("是否继续？(y/n) ");
         fflush(stdin);
         ch = getchar();
         if (ch == 'n' || ch == 'N')
         {
            game = 0;
            break;
         }
         else if (ch == 'y' || ch == 'Y')
            break;
         else
            printf("输入错误，重新输入\n");
      }
   }
   printf("游戏结束\n");
   return 0;
}