树状数组写炸求条
查看原帖
树状数组写炸求条
926886
kind_Ygg楼主2024/8/19 16:40
#include<bits/stdc++.h>
#define int long long 
using namespace std;
const int N=1e5+5;
int lowbit(int x)
{
	return x&-x;
}
int n,m;
int tree1[N],tree2[N];
int id,l,r,k,d,a;
void update(int x,int k)
{
	for(int i=x;i<=n;i+=lowbit(i))
	{
		tree1[i]+=k;
		tree2[i]+=k*i;
	}
}
int sum(int x)
{
	int ans=0;
	for(int i=x;i;i-=lowbit(i))
	{
		ans+=tree1[i]*(x+1)-tree2[i];
	}
	return ans;
}
signed main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>a;
		update(i,a);
		update(i+1,-2*a);
		update(i+2,a);
		for(int i=1;i<=n;i++)
			cout<<sum(i)<<' ';
		cout<<'\n';
	}
	while(m--)
	{
		cin>>id;
		if(id==1)
		{
			cin>>l>>r>>k>>d;
			update(l,k);
			update(l+1,d-k);
			update(r+1,-(k+(r-l+1)*d));
			update(r+2,(r-l)*d+k);
			for(int i=1;i<=n;i++)
				cout<<sum(i)<<' ';
			cout<<'\n';
		}
		else
		{
			cin>>l;
			cout<<sum(l)<<'\n';
			for(int i=1;i<=n;i++)
				cout<<sum(i)<<' ';
			cout<<'\n';
		}
	}
	return 0;
}

输出5

2024/8/19 16:40
加载中...