Command Sequence
#题目描述
有一个机器人可以通过接收一系列命令来移动。
命令序列中有四种类型的命令:
U:机器人向上移动一个单元。
D:机器人向下移动一个单元。
L:机器人向左移动一个单位。
R:机器人向右移动一个单元。
现在,给定一个长度 n 的命令序列.您需要找出命令序列有多少个子字符串满足机器人执行子字符串命令,它可以返回到起始位置。
子字符串是字符串中的连续字符序列。例如,“the best of”是“It was the best of times”的子字符串。例如,“Itwastimes”是“It was the best of times”的子序列,但不是子字符串。
输入
此问题包含多个测试用例。
第一行包含一个整数(1 <= t <=20)表示测试用例的数量。
对于每个测试用例,第一行包含一个整数 n (2<= n <=1e5).
第二行包含 UDLR 长度为n 的字符串。
输出
对于每个测试用例,输出一行一个整数,表示答案。
输入样例
1
6
URLLDR
输出样例
2
解析:
可以以每一个字符串的字符的位置为起点,在接下来遍历的过程中如果再次经过该点,则说明回到了起点。
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define ios ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
const int N=1e6+10;
map <int,int> a; //可以使用 map ,key记录位置,value记录经过的次数
string s;
signed main()
{
ios;
int t;
cin>>t;
while (t--)
{
a.clear();//记得清空哦
int n;
cin>>n;
cin>>s;
a[0]=1; //初始化
int sum=0;
int cnt=0;
for (int i=0;i<n;i++)
{
if (s[i]=='U') sum +=2e9;
if (s[i]=='D') sum -=2e9;
if (s[i]=='L') sum +=1;
if (s[i]=='R') sum -=1;
cnt +=a[sum];
a[sum]++;
}
cout<<cnt<<endl;
}
return 0;
}