Week5 作业C - 平衡字符串 [Gym - 270737B]

题目大意

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

输入

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

输出

一个整数表示答案

样例

说明

思路1[二分答案]

答案只有一个整数，表示替换字符串的最小长度。显然这个数介于0到n之间，并且越小越好，也就是说这个问题的答案具有明显的"单调性"。而我们可以发现，直接求答案好像并不容易，但是如果给一个整数l，判断替换字符串的长度为l时可不可行，这是容易的(穷举所有长度为l的字符串并判断可不可行即可)，复杂度为$O(n)$。于是，我们使用二分法，二分答案，二分范围是0-n，需要验证$log(n)$次。因此总复杂度是$O(nlog(n))$对于本题的数据规模来说是可行的。

思路2[尺取法]

这个方法更加巧妙，我们试图在时间复杂度$O(n)$的情况下找出所有可能的替换字符串的最小长度。

需要使用两个模拟指针，l，r。原字符串中区间$[l,r)$对应的范围表示替换字符串。

最开始，l，r置为0，表示替换字符串长度为0。如果说当前字符串出去替换部分外已经平衡，那么，我们已经找到了长度为r-l的字符串符合要求，长度大于r-l的字符串能否构成替换字符串对我们意义都不大。所以，我们可以试着将l右移一位；如果说长度为r-l的字符串不符合要求或者l=r，这时，l再往右移是没有意义的（这里可能需要深入思考），于是我们将r右移一位。循环这样的过程，直到l，r均无法右移为止。从可行的字符串中找到长度的最小值，即为答案。

代码(尺取法)

#include<iostream>
#include<string>
using namespace std;
int num[100];
int l, r;
int res;
string s;
int n;
int indexs[] = {'Q', 'W', 'E', 'R'};
bool judge()
{
	int maxN=max(max(num['Q'], num['W']), max(num['E'], num['R'])); 
	int x = r - l - (maxN - num['Q']) - (maxN - num['W']) - (maxN - num['E']) - (maxN - num['R']); 
	return (x >= 0) && (x % 4 == 0);
}
int main ()
{ 
	std::ios::sync_with_stdio(false);
	cin>>s;
	n = s.length();
	res=n; 
	for(int i=0;i<n;i++)	num[s[i]]++;
	if(num['Q'] == num['W'] && num['W'] == num['E'] && num['E'] == num['R'])
	{
		cout<<0;
		return 0;
	}
	while(r<=n)
	{
		if(!judge() || l == r)	num[s[r++]]--;		
		else
		{
			res = min(res, r - l); 
			num[s[l++]]++;
		}	
	}
	cout<<res;
	return 0;
} 

代码(二分答案)

#include<iostream>
#include<string>
using namespace std;
int n;
string s;
bool check(int *num, int d)
{
	int ma = max(max(max(num['Q'], num['W']), num['E']), num['R']);
	d -= 4*ma - num['Q'] - num['W'] - num['E'] - num['R'];
	return d >= 0 && d % 4 == 0;  //这里等于0也可以，但是小于0不可以！！！ 
} 
bool judge(int d)
{
	if(d >= n)	return true;
	int num[100] = {0};
	for(int i=d;i<n;i++)	num[s[i]]++;
	int l = 0, r=d;
	while(r<n)
	{
		if(check(num, d))	return true;
		num[s[r]]--;
		num[s[l]]++;
		l++, r++;
	}
	if(check(num, d))	return true;
	return false;
}
int getRes()
{
	int l=0, r=n+1;
	int mid;
	while(l < r)
	{ 
		mid = l+r>>1;   //右移一位 
		if(judge(mid))	r = mid;
		else l = mid+1;
	}
	return r;
}
int main ()
{
	cin>>s;
	n = s.length();
	cout<<getRes();
	return 0;
}