[ABC389]Snake Queue

灶龙

于 2025-01-19 09:37:01 发布

阅读量870

点赞数 9

分类专栏：题解文章标签： c++ 数据结构性能优化

本文链接：https://blog.youkuaiyun.com/Python13131313/article/details/145238052

版权

题解专栏收录该内容

19 篇文章

订阅专栏

思路

我们可以考虑维护两个数组 $a$ 和 $b$ ，存放每个蛇头坐标和长度。对于第一类查询，我们可以将队尾的指针 $k 2$ 加 $1$ ,存放他的蛇头坐标，并且把记录蛇的总长度的数 $s 2$ 加上 $l$ 。对于第二类查询，我们可以拿一个指针 $k 1$ 和一个数 $s 1$ 来分别记录现在的 $1$ 号蛇头坐标和被弹出的蛇的总长度，并把指针加 $1$ , $s 1$ 加上 $b [k 1]$ 。对于第三类查询，我们直接输出第 $k$ 条蛇的蛇头坐标减去弹出的蛇的总长度即可，也就是输出 $a [k + k 1] - s 1$ 。

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int q,k1,k2,s1,s2,a[500005],b[500005];
main(){
	cin>>q;
	while(q--){
		int x;
		cin>>x;
		if(x==1){
			int l;
			cin>>l;
			a[++k2]=s2;
			b[k2]=l;
			s2+=l;
		}
		if(x==2){
			a[++k1]=0;
			s1+=b[k1];
		}
		if(x==3){
			int k;
			cin>>k;
			k+=k1;
			cout<<a[k]-s1<<"\n";
		}
	}
	return 0;
}