题解:AT_abc389_c [ABC389C] Snake Queue

这道题我用的是前缀和来做的。

如果我们每次在需要输出时进行将蛇的总长度(除了最后一条)加一遍,这个时候,我们会发现代码会 TLE。

我们用前缀和。我们首先使用 \(p\) 数组将这几条蛇的总长度的前缀和记录下来,我们再用双指针记录队列的头和尾。这时
\(p_{f-1}\) 就是所有出了这个队列的蛇的总长度,\(p_{f+k-2}\) 即为题目中叫你求的这前 \(k\) 条蛇的总长度。我们再两个一减,就是答案啦。

AC code:

#include <bits/stdc++.h>
using namespace std;
#define N 300010
#define sz(s) s.size()
#define db double
#define mod 1000000007
#define P 998244353
#define ll long long
#define fi first
#define se second
#define pb push_back
#define mk make_pair
#define per(i,l,r) for(ll i=l;i>=r;i--)
#define rep(i,l,r) for(ll i=l;i<=r;i++)
#define in insert
#define y1 y142857
//pair<ll,ll> PII;
//unordered_map<int,int> f;
vector<int>edges[N];
//set<int>c;
int huiwen(int x){int b[100],l=0,t=x;while(t!=0){b[++l]=t%10;t/=10;}for(int i=1,j=l;i<=j;i++,j--)if(b[i]!=b[j])return 0;return 1;}
int prime(int x){for(int i=2;i*i<=x;i++)if(x%i==0)return 0;return 1;} 
inline int gcd(int a,int b){if(a<b)gcd(b,a);if(!b)return a; return gcd(b,a%b);} 
inline int lcm(int a,int b){return a*b/gcd(a,b);}
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*f;
}
inline void pri(int x){
    if(x==1)putchar(' ');
    if(x==2)putchar('\n');
}
inline void put(int x){
    if(x==1)puts("Yes");
    if(x==2)puts("No");
    if(x==3)puts("YES");
    if(x==4)puts("NO");
    if(x==5)puts("yes");
    if(x==6)puts("no");
}
inline void write(ll x,ll w,ll e){
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10,w,e);
    putchar(x%10+'0');
    if(x==e){
        if(w==1)putchar(' ');
        if(w==2)putchar('\n');
    }
    return;
}
int q=read(),f=1,r,a[N],l=0;ll p[N];
void sovel(){
	rep(i,1,q){
		int type=read();
		if(type==1){cin>>a[++l];r++;if(l==1)continue;else p[l-1]=p[l-2]+a[l-1];}
		else if(type==2)f++;
		else{
			int k;
			cin>>k;
			cout<<p[f+k-2]-p[f-1]<<"\n";
		}
	}
}
int main(){
    int _=1;
    while(_--)sovel();
}
posted @ 2026-01-02 13:36  OI_emperor  阅读(2)  评论(0)    收藏  举报