diff --git a/diy1.py b/diy1.py
index 51b23f3..988a589 100644
--- a/diy1.py
+++ b/diy1.py
@@ -4,13 +4,16 @@ print("玩家出拳:"+player)
 b=["石头","剪刀","布"]
 a=random.choice(b)
 print("计算机出"+a)
-if player==a:
-    print("平局")
-elif player=="石头" and a=="剪刀":
-   print("恭喜,你赢了")
-elif player=="剪刀" and a=="布":
-   print("恭喜,你赢了")
-elif player=="布" and a=="石头":
-   print("恭喜,你赢了")
+if player in b:
+   if player==a:
+      print("平局")
+   elif player=="石头" and a=="剪刀":
+      print("恭喜,你赢了")
+   elif player=="剪刀" and a=="布":
+      print("恭喜,你赢了")
+   elif player=="布" and a=="石头":
+      print("恭喜,你赢了")
+   else:
+      print("很遗憾,你输了")
 else:
-   print("很遗憾,你输了")
\ No newline at end of file
+   print("输入错误")
\ No newline at end of file