题解: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();
}

浙公网安备 33010602011771号