C - 平衡字符串（Week5作业）

题目

一个长度为 n 的字符串 s，其中仅包含 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符。
如果四种字符在字符串中出现次数均为 n/4，则其为一个平衡字符串。
现可以将 s 中连续的一段子串替换成相同长度的只包含那四个字符的任意字符串，使其变为一个平衡字符串，问替换子串的最小长度?
如果 s 已经平衡则输出0。Input一行字符表示给定的字符串sOutput一个整数表示答案Examples

Input

QWER

Output

0

Input

QQWE

Output

1

Input

QQQW

Output

2

Input

QQQQ

Output

3

Note

1<=n<=10^5
n是4的倍数
字符串中仅包含字符 ‘Q’, ‘W’, ‘E’ 和 ‘R’.

题目思路

思想：尺取法  差分法
实现：给定一段区间（l,r)，判断如何修改这个区间可以使得整个字符串满足条件；
	·计算这个区间外各个字母的数量
	·数量最多的字母：m = max(sum1, sum2, sum3, sum4) 
	·区间内字母总数：total = r-l+1;
	·为了保证整个字符串四个字母个数相同
	fre = total-(m-sumq)-(m-sume)-(m-sumr)-(m-sumw);
		·若fre>=0 && fre%4=0：说明修改这个区间可以使整个字符串满足条件，返回true
			·记录下min(ans,total)
			·l左移，l++
		·否则，说明不可以
			·r右移，扩大区间,r++
优化：可以利用差分法进行优化，在计算各个字母的数量时，可以利用差分转换为对差分数组的单点操作

代码实现

#include<bits/stdc++.h>
using namespace std;
char s[100010];
int sumq,sumw,sume,sumr;
int Q[100010],W[100010],E[100010],R[100010];
int m,total,ans=100000,fre;
int n,l=1,r=1;
bool func()
{
	sumq=Q[n]-Q[r+1]+Q[l];
	sumw=W[n]-W[r+1]+W[l];
	sume=E[n]-E[r+1]+E[l];
	sumr=R[n]-R[r+1]+R[l]; 
	m = max(sumq,max(sumw,max(sume,sumr)));
	total = r-l+1;
	fre = total-(m-sumq)-(m-sume)-(m-sumr)-(m-sumw);
	if(fre>=0 && fre%4==0) return true;
	return false;
}
int main()
{
	while(s[n]!='\n')
	{
		n++;
		s[n]=getchar();
	}
	n--;
	Q[0]=0,W[0]=0,E[0]=0,R[0]=0;
	for(int i=1;i<=n;i++)
	{
		if(s[i]=='Q')
		{
			Q[i]=Q[i-1]+1;
			W[i]=W[i-1];
			E[i]=E[i-1];
			R[i]=R[i-1];
		}
		if(s[i]=='W')
		{
			Q[i]=Q[i-1];
			W[i]=W[i-1]+1;
			E[i]=E[i-1];
			R[i]=R[i-1];	
		}
		if(s[i]=='E')
		{
			Q[i]=Q[i-1];
			W[i]=W[i-1];
			E[i]=E[i-1]+1;
			R[i]=R[i-1];
		}
		if(s[i]=='R') 
		{
			Q[i]=Q[i-1];
			W[i]=W[i-1];
			E[i]=E[i-1];
			R[i]=R[i-1]+1;
		}
	}
	while(r!=n&&l!=n)
	{
		if(func()) 
		{
			ans = min(r-l+1,ans);
			l++;
		}
		else
		{
			r++;
		}	
	}
	cout<<ans<<endl;
	return 0;
}