oj 回文子串

描述：

给你一个字符串a和一个正整数n,判断a中是否存在长度为n的回文子串。如果存在，则输出YES，否则输出NO。回文串的定义：记串str逆序之后的字符串是str1，若str=str1,则称str是回文串，如"abcba".

Python代码：

# 给你一个字符串a和一个正整数n,# 判断a中是否存在长度为n的回文子串。# 如果存在，则输出YES，否则输出NO。# 回文串的定义：记串str逆序之后的字符串是str1，# 若str=str1,则称str是回文串，如"abcba".def huiwen(a, n):    newA = ''    for i in range(n // 2):        newA += a[i]    mid = (n // 2) + 1    behindA = a[mid:]    behindAR = behindA[:: -1]    if newA == behindAR:        # print(behindAR)        return 'YES'    else:        return 'NO'if __name__ == '__main__':    a = input('a:')    n = int(input('n:'))    if n == 1:        print('YES')    else:        finish = n        flag = 0        for i in range(0, len(a) - n + 1):            strA = a[i: finish]            # print(newA)            finish += 1            result = huiwen(strA, n)            if result == 'YES':                flag = 1                break            else:                flag = 0        if flag == 1:            print('YES')        else:            print('NO')