C - 平衡字符串
-
题意:
一个长度为 n 的字符串 s,其中仅包含 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符。
如果四种字符在字符串中出现次数均为 n/4,则其为一个平衡字符串。
现可以将 s 中连续的一段子串替换成相同长度的只包含那四个字符的任意字符串,使其变为一个平衡字符串,问替换子串的最小长度?
如果 s 已经平衡则输出0。 -
输入输出:
Input
一行字符表示给定的字符串s
Output
一个整数表示答案
-
解题思路:
需要计算字母出现的次数,考虑到运用前缀。四个数组Q,W,E,R分别记录四个字母的出现次数,初始为0,出现一次加1,如果未出现当前位置的值与前一位保持一致。输入string类型字符串,遍历字符串对四个数组进行更新,要求最小范围,同时求解答案为连续区间,区间左右端点有明确移动方向,考虑使用尺取法。
大致思路为:两个指针L,R,用于标记范围。对于选定范围,先求出该范围之外的四个字母的个数,要使字符串平衡,计算出要将四个字母的个数替换一致时需要的范围长度,所选范围必须大于该范围才可能达到平衡,同时剩余位置需要是4的整数倍。
ll和rr是两个指针,选定范围以外的字母个数等于总个数减去范围内的字母个数,q,w,e,r记录每次范围中各字母的个数,total为所选范围长度,free是减去要替换的最小长度后的剩余位置长度。随着指针移动,如果free大于0且能被4整除,当前范围满足条件,左指针右移一格,否则右指针右移。ans实时更新为当前最小范围长度,最后输出。
-
代码实现:
#include <iostream>
#include <cstring>
#include <cmath>
using namespace std;
int Q[100010],W[100010],E[100010],R[100010];
int main()
{
string s;
cin>>s;
int length= s.length();
for (int i=0;i<length;i++)
{
if(s[i]=='Q')
{
if(i==0) {
Q[0]=1; W[0