GYM 100971 K.Palindromization(hash)

来源:互联网 发布:店铺淘宝客sina 编辑:程序博客网 时间:2024/05/21 09:31

Description
给出一个字符串,问删掉一个字符是否可以变成回文串,如果可以则输出该字符位置
Input
一个只由小写字母组成的字符串s,2<=|s|<=2e5
Output
如果删去第x个字符可以使得该字符串变成回文串则输出YES和x,否则输出NO
Sample Input
evertree
Sample Output
YES
2
Solution
hash,正着反着都hash一遍,记录前缀和后缀的hash值,然后枚举删去的点,如果删去该点之后前后缀的hash值相同则暴力判一遍回文,如果回文则找到了一个解,否则接着枚举
Code

#include<cstdio>#include<iostream>#include<cstring>#include<algorithm>#include<cmath>#include<vector>#include<queue>#include<map>#include<set>#include<ctime>using namespace std;typedef long long ll;#define INF 0x3f3f3f3f#define maxn 222222const ll mod=1e9+7;const ll inv26=576923081;ll h[maxn],a[maxn],b[maxn];char s[maxn];ll mod_pow(ll a,ll b,ll c){    ll ans=1ll;    while(b)    {        if(b&1)ans=ans*a%c;        a=a*a%c;        b>>=1;    }    return ans;}int main(){    h[1]=1;    for(int i=2;i<maxn;i++)h[i]=26ll*h[i-1]%mod;    while(~scanf("%s",s+1))    {        int n=strlen(s+1);        a[0]=b[0]=0;        for(int i=1;i<=n;i++)a[i]=(a[i-1]+h[i]*(s[i]-'a')%mod)%mod;        for(int i=1;i<=n;i++)b[i]=(b[i-1]+h[i]*(s[n-i+1]-'a')%mod)%mod;        int ans=-1;        for(int i=1;i<=n;i++)        {            ll tx=(a[i-1]+((a[n]-a[i])%mod+mod)%mod*inv26%mod)%mod;            ll ty=(b[n-i]+((b[n]-b[n-i+1])%mod+mod)%mod*inv26%mod)%mod;            if(tx==ty)            {                int flag=1;                for(int l=1,r=n;l<r;l++,r--)                {                    if(l==i)l++;                    if(r==i)r--;                    if(l<r&&s[l]!=s[r])                    {                        flag=0;                        break;                    }                }                if(flag)                {                    ans=i;                    break;                }            }        }         if(ans==-1)printf("NO\n");        else printf("YES\n%d\n",ans);    }    return 0;}
0 0
原创粉丝点击