[ABC389]Snake Queue

思路

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

代码

#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;
}