平衡字符串
一个长度为 n 的字符串 s,其中仅包含 ‘Q’, ‘W’, ‘E’, ‘R’ 四种字符。
如果四种字符在字符串中出现次数均为 n/4,则其为一个平衡字符串。
现可以将 s 中连续的一段子串替换成相同长度的只包含那四个字符的任意字符串,使其变为一个平衡字符串,问替换子串的最小长度?
如果 s 已经平衡则输出0.
输入一行字符表示给定的字符串s,输出一个整数表示答案
sample input:
QWER
sample output:
0
sample input:
QQWE
sample output:
1
sample input:
QQQW
sample output:
2
sample input:
QQQQ
sample output:
3
数据类型:
1<=n<=10^5
n是4的倍数
字符串中仅包含字符 ‘Q’, ‘W’, ‘E’ 和 ‘R’
思路:
- 利用滑动窗口的方法来对窗口的大小进行控制,最终找到能满足条件的最小窗口
- 先利用一个<char , int>的map对整个字符串中的字符的出现次数进行记录,计算一下每个字符都比应该得到的字符数多了多少并记录在map中
- 因为题目说明了更改的一定是他的子串,也就是肯定是s的一部分,所以可以设置一个滑动窗口,他有左边界left,右边界right.每次判断滑窗中的字符串正不正常(没有大于0的就正常),如果正确说明这一段可以替换,替换后为平衡字符串。
- 如果满足,更新长度并尝试缩小(left++),如果不满足,尝试加大窗口(right++)
- 找出所有可以替换的子串,寻找其中最短的(减小滑窗)对答案进行更新。
#include<iostream>
#include<cstdio>
#include<map>
#include<cstring>
#include<cmath>
using namespace std;
char member[4]={'Q','W','E','R'};
map<char,int> record;
int balance(string s)
{
int len = s.length();
int ans = len;
for(int i = 0; i < len; i++)
{
if(s[i] == 'Q')
record[member[0]]++;
if(s[i] == 'W')
record[member[1]]++;
if(s[i] == 'E')
record[member[2]]++;
if(s[i] == 'R')
record[member[3]]++;
}
int n=len/4;
int judge = 0;
for(int i = 0; i < 4; i++)
{
if(record[member[i]]>n)
{
record[member[i]]=record[member[i]]-n;
judge += record[member[i]];
}
else
{
record[member[i]]=0;
}
}
if(judge == 0)
return 0;
int left = 0;
int right = 1;
int rec[4] = {0,0,0,0};
for(int i = left; i <= right; i++)
{
if(s[i] == 'Q')
rec[0]++;
if(s[i] == 'W')
rec[1]++;
if(s[i] == 'E')
rec[2]++;
if(s[i] == 'R')
rec[3]++;
}
while(right < len)
{
if(ans == judge)
break;
if(rec[0] >= record[member[0]] && rec[1]>=record[member[1]] && rec[2]>=record[member[2]] && rec[3]>= record[member[3]])
{
if((right-left)<ans)
ans=right-left+1;
if(s[left] == 'Q')
rec[0]--;
if(s[left] == 'W')
rec[1]--;
if(s[left] == 'E')
rec[2]--;
if(s[left] == 'R')
rec[3]--;
left++;
}
else
{
right++;
if(s[right] == 'Q' && right != len)
rec[0]++;
if(s[right] == 'W' && right != len)
rec[1]++;
if(s[right] == 'E' && right != len)
rec[2]++;
if(s[right] == 'R' && right != len)
rec[3]++;
}
}
return ans;
}
int main()
{
string str;
cin>>str;
int ans=0;
ans=balance(str);
printf("%d\n",ans);
return 0;
}