36pts TLE求助
查看原帖
36pts TLE求助
554665
Dream__Sky楼主2023/7/10 21:30

rt.

#include <bits/stdc++.h>
#define int long long 
using namespace std;
int a[100001],b[100001],n,m;
struct info
{
	int l,r,tag,sum;
}t[400001];
inline void build(int p,int l,int r)
{
	t[p].l=l,t[p].r=r;
	if(l==r) 
	{
		t[p].sum=b[l];
		return ;
	}
	int mid=(l+r)>>1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
inline void spread(int p)
{
	if(t[p].tag)
	{
		t[p*2].sum+=t[p].tag*(t[p*2].r-t[p*2].l+1);
		t[p*2+1].sum+=t[p].tag*(t[p*2+1].r-t[p*2+1].l+1);
		
		t[p*2].tag+=t[p].tag;
		t[p*2+1].tag+=t[p].tag;
		
		t[p].tag=0;
	} 
}
inline void query(int p,int l,int r,int k)
{
	if(t[p].l>=l&&t[p].r<=r) 
	{
		t[p].tag+=k;
		t[p].sum+=(t[p].r-t[p].l+1)*k;
		return;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	if(l<=mid) query(p*2,l,r,k);
	if(mid<r) query(p*2+1,l,r,k); 
	t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
inline void add(int p,int l,int k)
{
	if(t[p].l==t[p].r)
	{
		t[p].tag+=k;
		t[p].sum+=k;
		return ;
	}
	spread(p);
	int mid=(t[p].l+t[p].r)>>1;
	if(l<=mid) add(p*2,l,k);
	else add(p*2+1,l,k); 
	t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
inline int ask(int p,int l,int r)
{
	if(t[p].l==t[p].r)
		return t[p].sum;
	spread(p);
	int mid=(t[p].l+t[p].r)>>1,ans=0;
	if(mid>=l) ans+=ask(p*2,l,r);
	if(mid<r) ans+=ask(p*2+1,l,r);
	return ans;
}
signed main()
{
	ios::sync_with_stdio(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=1;i<=n;i++) b[i]=a[i]-a[i-1];
	
	build(1,1,n);
	for(int i=1,opt,l,r,k,d;i<=m;i++)
	{
		cin>>opt;
		if(opt==1)
		{
			cin>>l>>r>>k>>d;
			add(1,l,k);
			if(l+1<=r) query(1,l+1,r,d);
			if(r<n) add(1,r+1,-(k+(r-l)*d));
		}
		else 
		{
			cin>>k;
			cout<<ask(1,1,k)<<"\n";
		}
	}
	return 0;
}
2023/7/10 21:30
加载中...