萌新刚学线段树,后面三个点TLE了,大佬们帮帮我
查看原帖
萌新刚学线段树,后面三个点TLE了,大佬们帮帮我
422992
Chaser2楼主2021/3/21 17:48
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e6+5;
//const ll mod=2147483647;
struct node
{
    ll sum,l,r; //sum为l,r区间的和 
}tree[N];  //树节点编号 
ll a[N]; //输入的数据 
void build(ll i,ll l,ll r)//建树,i是编号,l是左,r是右;
{
    tree[i].l=l;
    tree[i].r=r;
    if(l==r){
        tree[i].sum=a[l];
        return ;
    }
    ll mid=(l+r)>>1;
    build(i*2,l,mid);
    build(i*2+1,mid+1,r);
    tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
    return ;
}
void add(ll i,ll l,ll r,ll k)//区间修改
{
	if(tree[i].l==tree[i].r)
	{
		tree[i].sum+=k;
		return ;
	}
    ll mid=(tree[i].r+tree[i].l)/2;
    if(mid>=l)add(i*2,l,r,k);
    if(mid<r)add(i*2+1,l,r,k);
    tree[i].sum=tree[i*2].sum+tree[i*2+1].sum;
    return; 
}
ll query(ll i,ll l,ll r)//区间查询,查询区间[l,r]的和
{
    if(tree[i].l>=l&&tree[i].r<=r)
        return tree[i].sum;
    if(tree[i].r<l||tree[i].l>r)return 0;
    ll s=0;
    if(tree[i*2].r>=l)//左儿子的l已经定了,直接判断r即可
        s+=query(i*2,l,r);
    if(tree[i*2+1].l<=r)//右儿子的l已经定了,直接判断r即可
        s+=query(i*2+1,l,r);
    return s;
}
int main()
{
	ll n,m;//n为数字的个数,m为操作的次数 ;
	cin>>n>>m; 
	for(int i=1;i<=n;i++)
    cin>>a[i];
    build(1,1,n);
    while(m--)
    {
    	ll p;//p为1表示修改,p为2表示查询 
    	cin>>p;
    	if(p==1)
    	{
    		ll x,y,z;//x,y为修改的区间,y为添加的值 
    		cin>>x>>y>>z;
    		add(1,x,y,z);
		}
		if(p==2)
		{
			ll x,y;//x,y为查询的区间
			cin>>x>>y; 
			cout<<query(1,x,y)<<endl;
		}
	}
	return 0; 
}
2021/3/21 17:48
加载中...