程序设计Week5——C-平衡字符串

题目描述

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

Input

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

Output

一个整数表示答案

解题思路

看到这道题目，第一想法是先计算整个字符串不同字符的数目，然后根据不同的数目来判断如何替换，而题目中给出了要求：将s中一段连续的子串进行替换，这让我们想到可以利用尺取法，将这个区间以从小到大再到小、从左到右的方式进行遍历，最后选出这个区间。可以考虑如下策略：

• 从头开始，选择一个区间[L,R]，若当前区间进行替换后可以满足要求，则L++，否则R++；
• 对于给定区间，如何判定满足要求呢？可以先记录整个字符串s中四种字符的个数，再计算不在区间[L,R]内的四种字符的个数，然后再在该区间内任意填充使得当前字符串满足平衡条件，再判断区间内剩余空位是否是四的倍数，若是则满足要求，否则不满足要求；

这样不断更替L和R，直到L遍历完所有位置，对于每一个符合要求的区间，利用下标计算其长度，并加以记录和更新，只记录最短的区间长度，最后输出即可。

实现代码

#include<iostream>
#include<map>
#include<string>
using namespace std;

string s;
int SUM[4],sum[4];
map<char,int> mp;

bool judge(int l,int r)
{
	int max=-1;
	int a[4];
	for(int i=0;i<4;++i)
	{
		a[i]=SUM[i]-sum[i];
		if(max<a[i])
			max=a[i];
	}
	int f=(max-a[0])+(max-a[1])+(max-a[2])+(max-a[3]);
	f=r-l+1-f;
	if(f>=0&&f%4==0)
		return true;
	else
		return false;
}

int main()
{
	mp.insert(pair<char,int>('Q',0));
	mp.insert(pair<char,int>('W',1));
	mp.insert(pair<char,int>('E',2));
	mp.insert(pair<char,int>('R',3));
	cin>>s;
	int i=0;
	for(;i<s.size();++i)
		SUM[mp[s[i]]]++;
	int l=0,r=0,ans;
	if(SUM[0]==SUM[1]&&SUM[0]==SUM[2]&&SUM[0]==SUM[3])
		ans=0;
	else
	{
		sum[mp[s[0]]]++;
		while(l<i)
		{
			if(judge(l,r))
			{
				if(ans>(r-l+1))
					ans=r-l+1;
				if(l==r)
				{
					sum[mp[s[l]]]--;
					l++;
					r++;
					sum[mp[s[r]]]++;
				}
				else
				{
					sum[mp[s[l]]]--;
					l++;
				}
			}
			else if(r<i)
			{
				r++;
				sum[mp[s[r]]]++;
			}
			else if(r>=i)
				break;
		}
	}
	cout<<ans<<endl;
	return 0;
}

总结

本题是典型的尺取法：所求答案为一个连续区间（或其长度），并且区间的左右端点有明确的移动方向，知道方法后，只需要考虑清楚如何判定一个区间满足要求，然后根据满足情况移动端点即可。