WEEK5 作业 C - 平衡字符串

C - 平衡字符串

题目描述

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

Input

一行字符表示给定的字符串s

Output

一个整数表示答案

Sample Input 1

QWER

Sample Output 1

0

Sample Input 2

QQWE
###  Sample Output 2
```handlebars
1

Sample Input 3

QQQW

Sample Output 3

2

Sample Input 3

QQQQ

Sample Output 3

3

题解

• 采用尺取法。
• 设置左右两个指针，如果只改变两指针之间的区间能满足平衡的条件的话，则让左边的指针右移一个（减小范围），如果不能则让右边的指针右移一个（扩大范围）。直至右边指针到达最右端。然后只缩小左边指针至区间为0。记录下过程中区间大小的最小值。
• 如何判断两指针之间的区间能否满足平衡呢：维护四个数字，这四个数字存放的是指针之外的字母的个数，maxx是这四个数中的最大值，把其他三个数字都填平至maxx假设需要改变区间内的m个位置，total是指针之间的数字个数，如果total-m>=0且total是4的倍数。则说明只改变指针区间内的数字可以变成平衡。

Note

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

代码

#include <iostream>
#include <stdio.h>
#include <string>
#include <algorithm>
using namespace std;
string c;
int n=0,sum_q=0,sum_w=0,sum_e=0,sum_r=0;
int l=0,r=-1,ans;
bool success(){
	int maxx=max(max(sum_q,sum_w),max(sum_e,sum_r));
	int total=r-l+1;
	total=total-4*maxx+sum_q+sum_w+sum_e+sum_r;
	if(total>=0&&total%4==0){
		if(r-l+1<ans){ans=r-l+1;}
		return true;
	}
	return false;
}
int main(int argc, char** argv) {
	cin>>c;
	ans=n=c.size();
	for(int i=0;i<n;i++){
		switch(c[i]){
			case 'Q':sum_q++;break;
			case 'W':sum_w++;break;
			case 'E':sum_e++;break;
			case 'R':sum_r++;
		}
	}
	bool b=false;
	while(r<n-1){
		if(!b){//右边界右移 
			r++;
			switch(c[r]){
				case 'Q':sum_q--;break;
				case 'W':sum_w--;break;
				case 'E':sum_e--;break;
				case 'R':sum_r--;
			}
		}else{//左边界右移
			switch(c[l]){
				case 'Q':sum_q++;break;
				case 'W':sum_w++;break;
				case 'E':sum_e++;break;
				case 'R':sum_r++;
			}
			l++; 
		}
		b=success();
	}
	//右边界为最右端 
	while(l!=n){
		if(b){//左边界右移
			switch(c[l]){
				case 'Q':sum_q++;break;
				case 'W':sum_w++;break;
				case 'E':sum_e++;break;
				case 'R':sum_r++;
			}
			l++; 
		}else{break;}
		b=success();
	}
	printf("%d",ans);
	return 0;
}