2023年第五届河南省CCPC Problem A. 小水獭游河南

题目大意
给定一个字符串s，取它的两个连续子串a,b，如果a由单个字母组成，b是回文串，输出HE，否则输出NaN，如果S只有大小为1，则没办分为两个子串，同样为NaN。

错误思想
将s从头遍历，如果一个字符出现了两次，结束遍历，存第二次出现的字符位置，判断从此时开始b是否为一个回文串
问题是没办法保证s中a子串一定是以第二个相同字符结束。

还有一个坑是存s[i]出现次数的数组a大小不能开太大，因为每次判断需要将a数组归零，太大会TLE

错误代码

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
 
const int N=1e5+10;
int a[30];
signed main()
{
    std::ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int n;
    cin>>n;
    while(n--)
    {
        int q=0;
        memset(a,0,sizeof(a));
        int ans=0,cnt=0;
        string s;cin>>s;
        int len=s.length();
        if(len==1)
        {
            cout<<"NaN"<<endl;
            continue;
        }
        for(int i=0;i<len;i++)
        {
            a[s[i]-'a']++;
            if(a[s[i]-'a']>1)
            {
                q=i;
                cnt=1;
                break;
            }
        }
        
        if(cnt==0)
        {
            cout<<"HE"<<endl;
        }
        else
        {
            for(int i=q,j=len-1;i<j;i++,j--)
            {
                if(s[i]!=s[j])
                {
                    ans=1;
                    break;
                }
            }        
            if(ans==1)
            {
                cout<<"NaN"<<endl;
            }
            else
            {
                cout<<"HE"<<endl;
            }
        }        
    }
    return 0;
}

看过题解后更正思路
因为a子串最多只有26个字符，s大小在1e5，所以可以枚举出所有a子串，判断b子串是否为回文串。

代码实现（正解）

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
 
const int N=1e3;
int a[N];
string s;
int vis(int x,int len);
signed main()
{
    std::ios::sync_with_stdio(0); cout.tie(0); cin.tie(0);
    int n;cin>>n;
    while(n--)
    {
        memset(a,0,sizeof(a));
        cin>>s;
        int len=s.length();
        if(len==1)
        {
            cout<<"NaN"<<endl;
            continue;
        }
        for(int i=0;i<len;i++)
        {
            a[s[i]]++;
            if(a[s[i]]==1&&vis(i+1,len)==1)
            {
                cout<<"HE"<<endl;
                break;
            }
            else if(a[s[i]]>1)
            {
                cout<<"NaN"<<endl;
                break;
            }
        }
    }
    return 0;
}
int vis(int x,int len)
{
    for(int i=x,j=len-1;i<j;i++,j--)
    {
        if(s[i]!=s[j])
        return 0;
    }
    return 1;
}