6398. 【NOIP2018模拟10.30】Generator（树状数组区间修改）

题目描述

Description

Input

Output

输出 q 行，第 i 行表示数据 Di 的答案。

Sample Input

4 3 2

1 1 2 4

2 1 2

1 1 3 5

2 2 3

Sample Output

0

4

14

22

Data Constraint

题解

显然cdq+线段树，计算加上每个操作的贡献

修改取决于之后的询问，询问取决于之前的修改

然而线段树会T，所以用树状数组差分一下

开两个树状数组，询问1~t表示成s1+s2*t，修改t就在t处的s1减(t-1)*s,在s2加s

code

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#define fo(a,b,c) for (a=b; a<=c; a++)
#define fd(a,b,c) for (a=b; a>=c; a--)
#define low(x) (x&-(x))
using namespace std;

struct type{
	int type,l,r,s,t,id;
} a[200001];
long long ans[200001];
long long tr[200001];
long long Tr[200001]; //*t
bool bz[200001];
int d[200001];
int n,Q,i,j,k,l,t,len;

bool cmp(type a,type b)
{
	return a.t<b.t;
}
bool Cmp(type a,type b)
{
	return a.id<b.id;
}

void Change(int t,int s)
{
	long long S=(long long)(t-1)*s;

	while (t<=n)
	{
		tr[t]-=S;
		Tr[t]+=s;

		if (!bz[t])
		bz[t]=1,d[++len]=t;

		t+=low(t);
	}
}
void change(int l,int r,int s)
{
	Change(l,s);
	Change(r+1,-s);
}

long long Find(int t)
{
	long long T=t,ans=0,s=0;

	while (t)
	{
		ans+=tr[t];
		s+=Tr[t];

		t-=low(t);
	}

	return ans+s*T;
}
long long find(int l,int r)
{
	return Find(r)-Find(l-1);
}

void clear()
{
	int i;

	fo(i,1,len)
	tr[d[i]]=0,Tr[d[i]]=0,bz[d[i]]=0;

	len=0;
}

void work(int L,int R)
{
	int i,Mid=(L+R)/2;

	if (L==R) return;

	work(L,Mid);
	work(Mid+1,R);

	sort(a+L,a+R+1,Cmp);

	fo(i,L,R)
	if (a[i].t<=Mid && a[i].type==1)
	change(a[i].l,a[i].r,a[i].s);
	else
	if (a[i].t>Mid && a[i].type==2)
	ans[a[i].t]+=find(a[i].l,a[i].r);

	clear();

	fd(i,R,L)
	if (a[i].t<=Mid && a[i].type==2)
	change(a[i].l,a[i].r,1);
	else
	if (a[i].t>Mid && a[i].type==1)
	ans[a[i].t]+=find(a[i].l,a[i].r)*a[i].s;

	clear();
}

int main()
{
	freopen("generator.in","r",stdin);
	freopen("generator.out","w",stdout);

	scanf("%d%d",&n,&Q);
	fo(i,1,Q-1)
	{
		scanf("%d",&j);
		a[j].t=i;
	}
	fo(i,1,Q)
	{
		scanf("%d",&a[i].type);

		if (a[i].type==1)
		scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].s);
		else
		scanf("%d%d",&a[i].l,&a[i].r);

		a[i].id=i;
		++a[i].t;
	}

	sort(a+1,a+Q+1,cmp);
	work(1,Q);

	fo(i,1,Q)
	{
		ans[i]+=ans[i-1];
		printf("%lld\n",ans[i]);
	}

	fclose(stdin);
	fclose(stdout);

	return 0;
}