python 查找指定长度的回文子串

给定一个字符串st和一个正整数n，确定是否有回文类型的子字符串，长度为n。如果是，则打印是，否则打印否。
例如： 

Input：

rqasabcbatv

n=5

Output：

abcba

Yes

st=input()
n=int(input())
flag=0
for i in range(len(st)):
    if i+n>len(st): #索引超过最大值，直接提前退出
        break
    str_tmp=st[i:i+n]
    str_tmp_reverse=str_tmp[::-1] #字符串翻转
    if str_tmp==str_tmp_reverse:
        flag=1
        print(str_tmp)
        break
if flag==1:
    print('YES')
else:
    print('NO')