如果我想编辑一个猜数字的游戏，数字随机，有五次机会，如果输入结果大于或小于这个数字给出相应提示，如果5次结束后仍未猜对，则提示游戏失败并公布答案，以下程序可以吗提示

import random
num = random.randint(1,100)
for i in range(5):
    a = input("请输入一个数字")
    if a == num:
        print("恭喜你答对了")
        break
    elif a < num:
        print("你猜小了")
    else:
        print("你猜大了")
if a != num:
    print("游戏失败，答案是",num)