修复选择难度输入字母崩溃

m0Nesyyyyyy · Jul 12, 2020 · 905d3d3 · 905d3d3
1 parent 9801f84
commit 905d3d3
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/input.cpp b/src/input.cpp
@@ -12,10 +12,15 @@ int inputDifficulty()
     std::string cmd;
     while (1)
     {
-        std::cout << "设置难度：1简单 2普通 3困难" << std::endl;
-
-        std::cin >> cmd;
+        while(1)
+        {
+            std::cout << "设置难度：1简单 2普通 3困难" << std::endl;
 
+            std::cin >> cmd;
+            if( cmd == "1" || cmd == "2" || cmd == "3" )
+                break;
+            std::cout << "输入错误！" << std::endl;
+        }
         Difficulty difficulty = static_cast<Difficulty>(std::stoi(cmd));
 
         switch (difficulty)