week5作业题_C-平衡字符串

C-平衡字符串

---

题目描述

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

Input
一行字符表示给定的字符串s
Output
一个整数表示答案
Examples
Input
QWER
Output
0
Input
QQWE
Output
1
Input
QQQW
Output
2
Input
QQQQ
Output
3

题目思路

这个题目的做法是尺取法。即双指针法，通过两个指针的移动，改变两个指针中的窗口。首先我们统计整个字符串中各个字符的个数，，如果各个字符的个数都相等，就已经平衡。否则确定一个窗口，窗口的长度为total，求出窗口外的各个字符的个数，然后将各个字符的个数向最大的那个对齐，即在窗口内增加字符使得各个字符的个数相同，然后计算free，即增加了字符后窗口中剩下的位置，如果free >= 0并且free%4 == 0，则符合要求将左指针右移，缩小窗口；如果不符合要求将右指针右移，扩大窗口。然后得到最小的total就是结果。

代码实现

#include <bits/stdc++.h>
using namespace std; 
#define _for(i,a,b) for(int i = (a);i < (b); i++)
#define _rep(i,a,b) for(int i = (a);i <= (b); i++)

string s;
int  ans = 2e5;
int sum[5];
int t[5];

int getMax(int a[])			//求四个数的最大值 
{
	int m = 0;
	_rep(i,1,4)
		if(a[i] > m)m = a[i];
	return m;
}

void getCount(int l,int r,int a[])		//求字符串中各字符的个数 
{
	_rep(i,l,r)
	{
		if(s[i] == 'Q')a[1]++;
		else if(s[i] == 'W')a[2]++;
		else if(s[i] == 'E')a[3]++;
		else if(s[i] == 'R')a[4]++;
	}
}

int main()
{
	cin >> s;
	memset(sum,0,sizeof(sum));		//初始化 
	getCount(0,s.size()-1,sum);
	if(sum[1] == sum[2] && sum[2] == sum[3] && sum[3] == sum[4])	//已经平衡 
	{
		cout<<0<<endl;
		return 0;
	}
	int l = 0,r = 0;
	while(r < s.size())
	{
		memset(t,0,sizeof(t));
		getCount(l,r,t);
		_rep(i,1,4)
			t[i] = sum[i] - t[i];		//窗口外各个字符的个数 
			
		int max = getMax(t);
		int total = r-l+1;
		int free = total - (max - t[1]) - (max - t[2]) - (max - t[3]) - (max - t[4]);
		if(free >= 0 && free%4 == 0)		//符合要求 
		{
			if(total < ans) ans = total;
			if(l == r)
			{
				l++;
				r++;
			}
			else l++;
		}
		else r++;
	}
	cout << ans;
}

---

总结

在我们使用全局变量的时候，每次需要对全局变量操作的时候，需要对变量初始化或者更新，否则变量中之前的值会影响到后面的结果。