题意:刚开始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