Problem - 159D. Palindrome pairs(回文串对)
题意:
给一个字符串,求有多少个回文串对
解析:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define inf 2099999999
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep1(i,a,b) for(int i=a;i>=b;i--)
const int N=2e3+100;
ll l[N];
ll r[N];
/**
l[i]表示从以i为结尾的字符串里包含的回文串数
r[i]表示从i到最后一个字符这么长一段距离里包含的回文串数
那么直接求出l[i]和以i开始的字符串数,
然后用递推求出真正的r[i]
之后枚举i
sum+=以i为结尾的字符串含有的回文串数*r[i+1]
*/
int main()
{
//freopen("D://rush.txt","r", stdin);
ios::sync_with_stdio(false),cin.tie(0);
string str;
cin>>str;
for(int i=0;i<str.size();i++)
{
int l1=i,r1=i;
while(str[l1]==str[r1]&&l1>=0&&r1<str.size())
{
l[r1]++;
r[l1]++;
//cout<<' '<<r1<<' '<<l[r1]<<' '<<l1<<' ' <<r[l1]<<endl;
l1--;r1++;
}
l1=i,r1=i+1;
while(str[l1]==str[r1]&&l1>=0&&r1<str.size())
{
l[r1]++;
r[l1]++;
//cout<<r1<<' '<<l[r1]<<' '<<l1<<' ' <<r[l1]<<endl;
l1--;r1++;
}
}
for(int i=str.size()-2;i>=0;i--)
{
r[i]+=r[i+1];
}
//cout<<r[0]<<' '<<l[2]<<endl;
ll sum=0;
for(int i=0;i<str.size();i++)
{
sum+=l[i]*r[i+1];
}
cout<<sum<<endl;
return 0;
}