P1438 36pts tle线段树求调
  • 板块学术版
  • 楼主Dream__Sky
  • 当前回复0
  • 已保存回复0
  • 发布时间2023/7/11 08:58
  • 上次更新2023/11/3 10:37:30
查看原帖
P1438 36pts tle线段树求调
554665
Dream__Sky楼主2023/7/11 08:58

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 int read() {
	int x = 0, m = 1;
	char ch = getchar();
	while(!isdigit(ch)) {
		if(ch == '-') m = -1;
		ch = getchar();
	}
	while(isdigit(ch)) {
		x = x * 10 + ch - 48;
		ch = getchar();
	}
	return x * m;
}
inline void write(int x) {
	if(x < 0) {
		putchar('-');
		write(-x);
		return;
	}
	if(x >= 10) write(x / 10);
	putchar(x % 10 + '0');
}

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()
{
	n=read(),m=read();
	for(int i=1;i<=n;i++) a[i]=read();
	for(int i=1;i<=n;i++) b[i]=a[i]-a[i-1];
	
	build(1,1,n);
	for(int i=1;i<=m;i++)
	{
		int opt=read();
		if(opt==1)
		{
			int l=read(),r=read(),k=read(),d=read();
			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 
		{
			int k=read();
			write(ask(1,1,k));
			putchar('\n');
		}
	}
	return 0;
}

谢谢

2023/7/11 08:58
加载中...