2021中国大学生程序设计竞赛（CCPC）- 网络选拔赛 Command Sequence(思维+hash+查找区间和为0的map优化)

Command Sequence

题目

给一串命令,让机器人按它走,求有多少个子命令使机器人回到原位.

思路

• 回到原位则 $su{m}_L=su{m}_R,su{m}_U=su{m}_D$.
• 由上,本题和命令集的前缀和有关,因此,为方便处理前缀和,我们令$R\Rightarrow 1,L\Rightarrow -1,U\Rightarrow 1e6,D\Rightarrow -1e6$.(本题数据规模$2e5$,我们可以将纵向的移动拼接在横向移动前,保证做到对每一个指令集都有独一无二的数字对应)
• 问题转化为了寻找子区间和为0的区间个数,即$su{m}_R-su{m}_{L-1}=0$
  如果暴力枚举,会花费$O(n^2)$的时间,会超时. 因此需要考虑如何优化这个枚举过程.
• 考虑到枚举我们并不需要知道这个区间在具体是什么,只需知道满足$su{m}_R-su{m}_{L-1}=0$的区间有多少个就可以了,因此我们可以遍历所有的sum值,储存它们出现了多少次,对于出现超过两次的求一个组合数就可以了.
  记录$su{m}_i$出现的次数,未节省空间,用map建立映射最方便.

---

如此可以大致写出流程:

• hash处理命令集,处理出前缀和,表示一串指令.
• 用map统计所有$sum$值的出现次数,每两个可以组成一个子区间
• 遍历map计算组合数累计答案

PS:$sum=0$需要特判,因为其本身就表示回到原点.
PS2:处理组合数时会炸int,需注意.

AC代码

时间复杂度: $O(n\ast lo{g}_{2}n)$

#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
#define ll long long
#define linf 0x3f3f3f3f3f3f3f3f
#define ull unsigned long long
#define endl '\n'
//#define int long long
using namespace std;
typedef pair<int,int> PII;
const int N =10 + 2e5 ,mod=1e9 + 7;
char s[N];
ll sum[N];
map<ll,int>mp;
void solve()
{   
	mp.clear(); 
	int n;
	cin>>n>>s+1;
	
	const int B=1e6;
	for(int i=1;i<=n;i++)//  hash统计前缀和
	{
		if(s[i]=='L') sum[i]=sum[i-1]-1;
		else if(s[i]=='R')sum[i]=sum[i-1]+1;
		else if(s[i]=='U')sum[i]=sum[i-1]+B;
		else sum[i]=sum[i-1]-B;
	}
	for(int i=1;i<=n;i++)// 统计sum值出现次数
		mp[sum[i]]++;
	ll ans=0;
	
	for(auto a:mp) 
	{// UDRLUDRL  4+1+3+1+2+1=12
		if(a.second>1) ans+=1ll*a.second*(a.second-1)/2;// C(n,2) 注意炸int
		if(a.first==0)ans+=a.second; // 独立构成答案的		
	}
	
	cout<<ans<<endl;
}
signed main()
{
	ios::sync_with_stdio();cin.tie();cout.tie();

	int T;cin>>T;
	
	while(T--)
		solve();

	return 0;
}