平衡字符串 Week5作业C题

题目：

一个长度为 n 的字符串 s，其中仅包含 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符。

如果四种字符在字符串中出现次数均为 n/4，则其为一个平衡字符串。

现可以将 s 中连续的一段子串替换成相同长度的只包含那四个字符的任意字符串，使其变为一个平衡字符串，问替换子串的最小长度?

如果 s 已经平衡则输出0。

输入：
一行字符表示给定的字符串s

输出：
一个整数表示答案

样例：
Input:

QWER

Output:

0

Input:

QQWE

Output:

1

Input:

QQQW

Output:

2

思路：

本题适用于尺取法。
用 s1, s2, s3, s4 分别记录不包含区间 [L, R] 这一段时，字符 ‘A’, ‘B’, ‘C’, ‘D’ 的个数 。
先通过替换使4类字符数量一致，再判断剩余空闲位置是否为4的倍数。
首先另l=r=0（字符串从位置0开始），s1s2s3s4更新，为ma=max(s1,s2,s3,s4)，
total=r-l+1,
free=total-(maxx-s1)-(maxx-s2)-(maxx-s3)-(maxx-s4).（free为将四个字母全部平衡后还剩下的空位）。
如果该空位为0或是4的倍数，则该区间是合法的，此时l++，更新s1s2s3s4。如果不满足上述条件，则认为区间不合法，r++，同时更新s1s2s3s4。

代码：

#include <iostream>
#include <cstring>
#include <stdio.h>
using namespace std;

int l=0,r=0,n,ans=1000050;
string s;

int main()
{
	int s1=0,s2=0,s3=0,s4=0;
	cin>>s;
	n=s.length();
	for(int i=0;i<n;i++)
	{
		if(s[i]=='Q')s1++;
		if(s[i]=='W')s2++;
		if(s[i]=='E')s3++;
		if(s[i]=='R')s4++;
	}
	if(s1==s2 && s2==s3 && s3==s4)
	{
		cout << 0;
		return 0;
	}
	if(s[0]=='Q') s1--;
	if(s[0]=='W') s2--;
	if(s[0]=='E') s3--;
	if(s[0]=='R') s4--;
	int free,total;
	while(1)
	{
		int ma=max(max(s1,s2),max(s3,s4));
		total=r-l+1;
		free=total-(ma-s1)-(ma-s2)-(ma-s3)-(ma-s4);
		if((free>0&&free%4==0)||free==0)
		{
			ans=min(ans,total);
			if(s[l]=='Q') s1++;
			if(s[l]=='W') s2++;
			if(s[l]=='E') s3++;
			if(s[l]=='R') s4++;
			l++;
		}
		else
		{
			if(r+1<n) r++;
			else break;
			if(s[r]=='Q') s1--;
			if(s[r]=='W') s2--;
			if(s[r]=='E') s3--;
			if(s[r]=='R') s4--;
		}
	}
	printf("%d\n",ans);
	return 0;
}