E1-110.完美走位(滑动窗口)
题目描述
在第一人称射击游戏中,玩家通过键盘的 A
、S
、D
、W
四个按键控制游戏人物分别向左、向后、向右、向前进行移动,从而完成走位。
假设玩家每按动一次键盘,游戏任务会向某个方向移动一步,如果玩家在操作一定次数的键盘并且各个方向的步数相同时,此时游戏任务必定会回到原点,则称此次走位为完美走位。
现给定玩家的走位(例如:ASDA
),请通过更换其中一段连续走位的方式使得原走位能够变成一个完美走位。其中待更换的连续走位可以是相同长度的任何走位。
请返回待更换的连续走位的最小可能长度。
如果原走位本身是一个完美走位,则返回 0
。
输入描述
输入为由键盘字母表示的走位 s
,例如:ASDA
输出描述
输出为待更换的连续走位的最小可能长度。
备注
- 走位长度
1
≤s.length
≤100000
s.length
是4
的倍数s
中只含有A
,S
,D
,W
四种字符
用例1
输入
WASDAASD
Copy
输出
1
Copy
说明
将第二个A替换为W,即可得到完美走位
用例2
输入
AAAA
Copy
输出
3
Copy
说明
将其中三个连续的A替换为WSD,即可得到完美走位
思路:
-
统计字符个数:我们用一个
map<char, int>
来统计每个字符在字符串中的出现次数。 -
判断是否完美走位:如果四个字符
A
、S
、D
、W
在字符串中的出现次数相等,直接输出0
。 -
滑动窗口:我们用两个指针
left
和right
来构成一个窗口。right
指针扩展窗口,left
指针收缩窗口。对每个可能的窗口,检查其是否能够通过修改字符使得每个字符的出现次数变得平衡。 -
最小窗口长度:如果当前窗口满足条件,计算窗口的大小,并更新最小窗口大小。
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int M = 1e5 + 10;
int n, root;
void solve() {
string s;
cin>>s;
map<char,int>count;
for(char c:s){
count[c]++;
}
int n=s.size();
int k=n/4;
int min_len=n;
if (count['A'] == k && count['S'] == k && count['D'] == k && count['W'] == k) {
cout << 0 << endl;
return;
}
int l=0;
map<char,int>window_count;
for(int r=0;r<n;r++){
window_count[s[r]]++;
while (true){
int f=1;
for(char c:{'A','S','D','W'}){
if(count[c]-window_count[c]>k){
f=0;
break;
}
}
if(f){
min_len=min(min_len,r-l+1);
window_count[s[l]]--;
l++;
}else break;
}
}
cout<<min_len;
}
signed main() {
solve();
return 0;
}