H-Instructions Substring(牛客暑期第二场)

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

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值