链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
思路:使用前缀和map来记录每次的坐标出现的次数,每一次以当前点作为区间结束位去寻找区间头出现的次数乘以后面跟随的尾巴即以当前位置为结尾的答案次数;
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define int long long
int ans=0;
map<int, map<int, int> > m;
signed main()
{
int n,x,y,w1=0,d1=0,ans=0;
char q;
cin>>n>>x>>y;
getchar();
m[0][0]=1;
for(long long i=1;i<=n;i++)
{
cin>>q;
if(q=='A')
{
m[--d1][w1]++;
}
if(q=='W')
{
m[d1][++w1]++;
}
if(q=='S')
{
m[d1][--w1]++;
}
if(q=='D')
{
m[++d1][w1]++;
}
ans+=(m[d1-x][w1-y]*(n-i+1));
m[d1-x][w1-y]=0;
}
if(x==0&&y==0)
ans=(1+n)*n/2;
cout<<ans;
}