diff --git a/diy1.py b/diy1.py
index 04cf5da..dec04a3 100644
--- a/diy1.py
+++ b/diy1.py
@@ -4,9 +4,12 @@ import random
 list=["石头","剪刀","布"]
 computer=random.choice(list)
 print(computer)
-if computer==player:
-   print("平局")
-elif (player=="石头"and computer=="剪刀") or (player=="剪刀"and computer=="布") or (player=="布"and computer=="石头"):
-    print("恭喜你,你赢了")
-else: 
-    print("恭喜你,你输了")
\ No newline at end of file
+if player in list
+  if computer==player:
+    print("平局")
+  elif (player=="石头"and computer=="剪刀") or (player=="剪刀"and computer=="布") or (player=="布"and computer=="石头"):
+      print("恭喜你,你赢了")
+  else: 
+      print("恭喜你,你输了")
+else:
+    print("输入错误")
\ No newline at end of file