Update ExamSystem.java

src/ExamSystem.java

@@ -21,11 +21,7 @@ public class ExamSystem {
   
   public void generateProblems(Users users) {
     while (true) {
-      
-      System.out.print("准备生成" + users.getType()
-          + "数学题目，请输入生成题目数量（输入-1将退出当前用户，重新登录）：");
       String input = scanner.nextLine().trim();
-      
       if (input.equals("-1")) {
         break;
       }
@@ -41,19 +37,27 @@ public class ExamSystem {
         String target = input.replace("切换为", "");
         if (target.equals("小学") || target.equals("初中") || target.equals("高中")) {
           users = new Users(users.getName(), users.getPassword(), target);
-          System.out.println("系统提示：准备生成" + users.getType() + "数学题目");
+          System.out.print("系统提示：准备生成" + users.getType()
+              + "数学题目，请输入生成题目数量（输入-1将退出当前用户，重新登录）：");
         } else {
           System.out.println("请输入小学、初中和高中三个选项中的一个");
         }
         continue;
       }
       
-      int num = Integer.parseInt(input);
+      try {
-      if (num >= 10 && num <= 30 && generator != null) {
+        int num = Integer.parseInt(input);
-        Generator.generate(users, num, generator);
+        if (num >= 10 && num <= 30 && generator != null) {
-      } else {
+          Generator.generate(users, num, generator);
-        System.out.println("请输入 10-30 之间的数字");
+        } else {
+          System.out.println("请输入 10-30 之间的数字");
+        }
+      } catch (NumberFormatException e) {
+        System.out.println("请输入小学、初中和高中三个选项中的一个");
       }
+      
+      System.out.print("准备生成" + users.getType()
+          + "数学题目，请输入生成题目数量（输入-1将退出当前用户，重新登录）：");
     }
   }
 }