回文子串

描述

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

思路

从字符串第0位开始，选取n位，记为s，逆序记为r，比较是否相同。

代码

def func(a,b):
    for i in range(len(a)-b+1):
        s = a[i:i+b]
        r = s[::-1]
        if s == r:
            return 'YES'
    return 'NO'

print(func(a,n))