codeforces 814 C An impassioned circulation of affection

Problem

codeforces.com/contest/814/problem/C

Reference

Markdown Syntax

MaHua

Meaning

给定一串长为 n 的只有小写字母的字符串，q 个询问。

对每一个询问，给出字母 c 和最多的使用次数 m，表示最多将原串中 m 个位置的字符替还为 c。

求替换后的串中，最长的全为 c 的子串的长度。

Analysis

因为是连续的，就考虑将邻近的只有 c 的片段连起来。

预初理出每个字母的片段，然后用尺取枚举将哪些片连起来。

注意一些细节的处理。

Code

#include <cstdio>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 1500;

struct seq
{
int st, ed;
seq() {}
seq(int a, int b): st(a), ed(b) {}
int len()
{
return ed - st;
}
};

char s[N+2];
vector<seq> g[26];

int solve(int x, int m, int n)
{
int res = min(m, n), tmp = 0;
for(int l=0, r=0, add, e=g[x].size(); l<e; )
{
if(l == r)
{
if(++r >= e)
{
res = max(res,
g[x][l].len() + min(m, g[x][l].st + n-g[x][l].ed));
break;
}
if(m >= g[x][r].st - g[x][l].ed)
{
m -= g[x][r].st - g[x][l].ed;
tmp = g[x][r].st - g[x][l].ed + g[x][l].len() + g[x][r].len();
++r;
}
else
{
res = max(res,
g[x][l].len() + min(m, g[x][l].st + n-g[x][l].ed));
++l;
}
continue;
}
while(r < e && m >= g[x][r].st - g[x][r-1].ed)
{
m -= g[x][r].st - g[x][r-1].ed;
tmp += g[x][r].st - g[x][r-1].ed + g[x][r].len();
++r;
}
if(m > 0)
add = min(m, g[x][l].st + n
cc98
-g[x][r-1].ed);
else
add = 0;
res = max(res, tmp + add);
if(l + 1 < e)
{
if(l + 1 != r)
{
m += g[x][l+1].st - g[x][l].ed;
tmp -= g[x][l+1].st - g[x][l].ed + g[x][l].len();
}
++l;
}
else
break;
}
return res;
}

int main()
{
int n, q;
scanf("%d %s %d", &n, s, &q);
for(int i=0, j; i<n; i=j)
{
for(j=i+1; j<n && s[j]==s[i]; )
++j;
g[s[i]-'a'].push_back(seq(i, j));
}
for(int m; q--; )
{
char c;
scanf("%d %c", &m, &c);
printf("%d\n", solve(c-'a', m, n));
}
return 0;
}