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