想用线段树骗分,但不知道哪出问题了

P2801 教主的魔法

泰山飞狐 @ 2019-11-14 10:29:44

#include<bits/stdc++.h>
using namespace std;
int n,q,a[1000001];
struct tree
{
    int l,r;
    int pre,add;
}t[4000020];

void build(int p,int l,int r)
{
    t[p].l=l;
    t[p].r=r;
    if(l==r)
    {
        t[p].pre=a[l];
        return;
    }
    int mid=l+r>>1;
    build(p*2,l,mid);
    build(p*2+1,mid+1,r);
    t[p].pre=t[p*2].pre+t[p*2+1].pre;
}
void spread(int p)
{
    if(t[p].add)
    {
        t[p*2].pre+=t[p].add*(t[p*2].r-t[p*2].l+1);
        t[p*2+1].pre+=t[p].add*(t[p*2+1].r-t[p*2+1].l+1);
        t[p*2].add+=t[p].add;
        t[p*2+1].add+=t[p].add;
        t[p].add=0;
    }
}
long long ask(int p,int x,int y,int k)
{
    if(x<=t[p].l && y>=t[p].r)
    {
        if(t[p].pre>=k)
            return 1;
        else
            return 0;
    }
    spread(p);
    long long ans=0;
    int mid=t[p].l+t[p].r>>1;
    if(x<=mid)  ans+=ask(p*2,x,y,k);
    if(y>mid)   ans+=ask(p*2+1,x,y,k);
}
void change(int p,int x,int y,int z)
{
    if(x<=t[p].l && y>=t[p].r)
    {
        t[p].pre+=(long long)z*(t[p].r-t[p].l+1);
        t[p].add+=z;
        return;
    }
    spread(p);
    int mid=t[p].l+t[p].r>>1;
    if(x<=mid)  change(p*2,x,y,z);
    if(y>mid)   change(p*2+1,x,y,z);
    t[p].pre=t[p*2].pre+t[p*2+1].pre;
}

int main()
{
    scanf("%d %d",&n,&q);
    for(register int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    build(1,1,n);
    for(register int i=1;i<=q;i++)
    {
        char x[10];
        cin>>x;
        if(x[0]=='M')
        {
            int l,r;
            long long w;
            cin>>l>>r>>w;
            change(1,l,r,w);
        }
        if(x[0]=='A')
        {
            int l,r;
            long long c;
            cin>>l>>r>>c;
            cout<<ask(1,l,r,c)<<endl;
        }
    }
    return 0;
}

by decoqwq @ 2019-11-14 10:32:44

这题咋用一个线段树做啊/kk


by 泰山飞狐 @ 2019-11-14 10:34:16

@deco 我知道正解是分块,但我想用线段树骗些分。能看看哪有问题吗?


|