题意:刚开始red站在初始位置(0,0),她有一串指令:向上,向下,向左,向右(上将y++,下将y--,左x--,右x++),可以选择连续的指令子字符串并执行他们,最终能通过目标坐标(x,y),有多少种选择选项。
知识点:前缀和,mp
分析:从起点开始第一次经过目标点,后面的也都要计算进结果。计算从起点开始之后每一次操作完的坐标。某段前缀和的值为目标值,那么它就算一次,并且要加上前缀和的最后一个值后面的所有数,只要经过目标值,再到后面的值也算。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int main(){
int n,x,y;cin>>n>>x>>y;
string s;cin>>s;
vector<array<int,2>>a(n+1);
a[0]={0,0};
for(int i=0;i<n;i++){
a[i+1]=a[i];
if(s[i]=='W'){
a[i+1][1]++;
}
else if(s[i]=='A'){
a[i+1][0]--;
}
else if(s[i]=='S'){
a[i+1][1]--;
}
else{
a[i+1][0]++;
}
}
ll ans=0;
map<array<int,2>,int>mp;
for(int i=n;i>=0;i--){//枚举目标值的位置
mp[a[i]]=i;
if(mp.count({a[i][0]+x,a[i][1]+y})){
int j=mp[{a[i][0]+x,a[i][1]+y}];
j=max(j,i+1);//要是前缀和的最后一位
ans+=n-j+1;//加上目标值后面的数哦
}
}
cout<<ans<<endl;
return 0;
}
//a 0,0
//s -1,0
//a -1,-1
//w -2 -1
//d -2 0
//d -1 0
// 0,0