2021.CCPC-1009Command Sequence
来源:HDU
1009Command Sequence
题不题的不重要,比赛时候服务器真的太卡了,真的很搞心态
题意
ULDR分别代表向上左下右四个方向走一格
给定长度为n的字符串,然后找子串,需要满足的条件是,我遍历这个子串,可以正好回到起始位置
思路
看到其他人用的都是前缀和维护,我这里用的是图
将运动看成一个轨迹,起始在(0,0)的位置上,然后遍历字符串,如果我在遍历过程中重复到达一个点,那么我一定可以从这个点出发再回到这个点,这就是对答案的一个贡献,当我第二次再到达这个点时,贡献就变成了二
这里画个图更容易理解
如当前图,红色线代表我走的路径,在我第一次重复到达一个点的时候,对答案的贡献是蓝色的线,只有一条,当我第二次再次到达的时候,对答案的贡献是绿色的两条线
同理,第三次到达的时候贡献为三,我也推过,是没有问题的,所以答案就很明显了
由于n最大有五次方,不能开二维数组,所以这里用map存点的状态和这是我第几次到达
但是我比赛的时候Wa了,因为我没开long long,尬住了
还有一个点就是,不要用unodered_map,不知道为什么,反正就是不可以用来存PII,我用了会编译报错
听说要重赛了,还是挺开心的,毕竟这次失误有点大
#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
typedef long long ll;
const int N = 110;
map<PII, bool> p;//存x,y这个点是否到达过
map<PII, int> cnt;//如果重复到达,对答案的贡献
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, -1, 0, 1};
char a[100010];
int main()
{
int t;
scanf("%d", &t);
while (t --) {
ll res = 0;
p.clear();
cnt.clear();
int n;
int x = 0, y = 0;
p[make_pair(x, y)] = true;
cnt[make_pair(x, y)] = 1;
scanf("%d", &n);
scanf("%s", a + 1);
for (int i = 1;i <= n;i ++) {
int id;
if(a[i] == 'U') id = 0;
else if(a[i] == 'L') id = 1;
else if(a[i] == 'D') id = 2;
else if(a[i] == 'R') id = 3;
x = x + dx[id], y = y + dy[id];
if(!p.count(make_pair(x, y))) {//如果没有到达过,那么标记一下
p[make_pair(x, y)] = true;
cnt[make_pair(x, y)] = 1;
}else {//如果到达过,加上贡献
res += cnt[make_pair(x, y)];
cnt[make_pair(x, y)] ++;
}
}
cout << res << endl;
}
return 0;
}