标签:int 蔚来 多校 long mid ++ 补题 mr define
G KFC Crazy Thursday 马拉车算法
题意:
给定一个字符串,问有多少个以K或者F或者C结尾的回文子串。
思路:
马拉车算法,求出len。
利用区间加法获得总和即可。
也就是(直接看代码更容易理解)对于新串在i处“+1”,在i+len[i]+1处“-1”。因为这个区间内的字符都有某个以他为结尾的回文串。
代码:
#include <bits/stdc++.h>
#define endl '\n'
#define int long long
#define pii pair<int, int>
#define pll pair<int, int>
#define ull unsigned long long
using namespace std;
const int N = 2e7 + 10;
char a[N], b[N];
int p[N];
int n;
void init()
{
int k = 0;
b[k++] = '$', b[k++] = '#';
for (int i = 0; i < n; i++)
b[k++] = a[i], b[k++] = '#';
b[k++] = '^';
n = k;
}
void manacher()
{
int mr = 0, mid;
for (int i = 0; i < n; i++)
{
if (i < mr)
p[i] = min(p[mid * 2 - i], mr - i);
else
p[i] = 1;
while (b[i - p[i]] == b[i + p[i]])
p[i]++;
if (i + p[i] > mr)
{
mr = i + p[i];
mid = i;
}
}
}
int d[N];
int cnt[128];
void solve()
{
cin>>n;
cin >> a;
init();
manacher();
for (int i = 0; i < n; i++)
{
int L = i, R = i + p[i] - 1;
d[L]++;
d[R + 1]--;
}
int pre = 0;
for (int i = 0; i <n; i++)
{
pre += d[i];
cnt[b[i]] += pre;
}
cout << cnt['k'] << " " << cnt['f'] << " " << cnt['c'] << endl;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t = 1;
// cin >> t;
while (t--)
solve();
return 0;
}
标签:int,蔚来,多校,long,mid,++,补题,mr,define 来源: https://www.cnblogs.com/kingwz/p/16547313.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。