JZOJ 4694 火神的魚

題目大意

在池塘裏用一張網(左下角x1,y1,右上角x2,y2)捕魚,網和池塘都是一個矩形,池塘裏有n條魚,這些魚會四處遊動,火神會在魚遊動的間隙問你現在網裏有多少條魚。
魚的遊動可以概括爲兩個動作:
1 l r d : 表示標號在[l,r]這個區間內的魚向x軸正方向遊動了d個單位長度。
2 l r d:表示標號在[l,r]這個區間內的魚向y軸正方向遊動了d個單位長度。

多組數據。
T<=10,n,m<=30000,0<=d<=1e9,所有涉及的座標在[−10^9,10^9]範圍內。
時間限制 1s
空間限制 256M

解題思路

建兩棵線段樹,由於d是正數,所以從網裏面鑽出來的魚不可能再進去,所以我們可以用線段樹上區間[l,r]儲存的信息是:
(以x軸上的線段樹爲例)
1、標號爲[l,r]的魚中有多少條;
2、標號爲[l,r]的魚中座標小於x1的最大座標;
3、標號爲[l,r]的魚中座標在[x1,x2]範圍內的最大座標;

於是魚的遊動就可以用區間加減解決,加減之後就維護上述的第二、三項,及時把座標脫離範圍的魚加入或移出網外就可以了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 100006
#define fr(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const ll ding=2147483647;

struct nod
{
    int x,y,c,s;
} tr1[maxn*3],tr2[maxn*3];
int T,i,n,m,x1,x2,y1,y2,z,l,r,d,s,a[maxn],b[maxn];
void merge(int v)
{
    tr1[v].s=tr1[v+v].s+tr1[v+v+1].s;
    tr1[v].x=max(tr1[v+v].x,tr1[v+v+1].x);
    tr1[v].y=max(tr1[v+v].y,tr1[v+v+1].y);

    tr2[v].s=tr2[v+v].s+tr2[v+v+1].s;
    tr2[v].x=max(tr2[v+v].x,tr2[v+v+1].x);
    tr2[v].y=max(tr2[v+v].y,tr2[v+v+1].y);

    tr1[v].c=tr2[v].c=0;
    return;
}
void maket(int v,int st,int en)
{
    if (st==en)
    {
        tr1[v].c=tr2[v].c=0;
        tr1[v].s=tr2[v].s=0;
        tr1[v].x=tr1[v].y=tr2[v].x=tr2[v].y=-ding;
        if (a[st]<x1) tr1[v].x=a[st];
        else if (a[st]<=x2) tr1[v].y=a[st];
        if (b[st]<y1) tr2[v].x=b[st];
        else if (b[st]<=y2) tr2[v].y=b[st];
        if (a[st]>=x1 && a[st]<=x2 && b[st]>=y1 && b[st]<=y2) 
            tr1[v].s=1;
        return;
    }
    int m=(st+en) >> 1;
    maket(v+v,st,m);
    maket(v+v+1,m+1,en);
    merge(v);
    return;
}
void update1(int v)
{
    if (!tr1[v].c) return;
    if (tr1[v+v].x>-ding) tr1[v+v].x+=tr1[v].c;
    if (tr1[v+v].y>-ding) tr1[v+v].y+=tr1[v].c;
    if (tr1[v+v+1].x>-ding) tr1[v+v+1].x+=tr1[v].c;
    if (tr1[v+v+1].y>-ding) tr1[v+v+1].y+=tr1[v].c;
    tr1[v+v].c+=tr1[v].c;
    tr1[v+v+1].c+=tr1[v].c;
    tr1[v].c=0;
    return;
}
void update2(int v)
{
    if (!tr2[v].c) return;
    if (tr2[v+v].x>-ding) tr2[v+v].x+=tr2[v].c;
    if (tr2[v+v].y>-ding) tr2[v+v].y+=tr2[v].c;
    if (tr2[v+v+1].x>-ding) tr2[v+v+1].x+=tr2[v].c;
    if (tr2[v+v+1].y>-ding) tr2[v+v+1].y+=tr2[v].c;
    tr2[v+v].c+=tr2[v].c;
    tr2[v+v+1].c+=tr2[v].c;
    tr2[v].c=0;
    return;
}
void modify1(int v,int st,int en,int l,int r,int x)
{
    if (st==l && en==r)
    {
        if (tr1[v].x>-ding) tr1[v].x+=x;
        if (tr1[v].y>-ding) tr1[v].y+=x;
        tr1[v].c+=x;
        return;
    }
    update1(v);
    update2(v);
    int m=(st+en) >> 1;
    if (r<=m) modify1(v+v,st,m,l,r,x);
    else if (l>m) modify1(v+v+1,m+1,en,l,r,x);
    else
    {
        modify1(v+v,st,m,l,m,x);
        modify1(v+v+1,m+1,en,m+1,r,x);
    }
    merge(v);
    return;
}
void modify2(int v,int st,int en,int l,int r,int x)
{
    if (st==l && en==r)
    {
        if (tr2[v].x>-ding) tr2[v].x+=x;
        if (tr2[v].y>-ding) tr2[v].y+=x;
        tr2[v].c+=x;
        return;
    }
    update1(v);
    update2(v);
    int m=(st+en) >> 1;
    if (r<=m) modify2(v+v,st,m,l,r,x);
    else if (l>m) modify2(v+v+1,m+1,en,l,r,x);
    else
    {
        modify2(v+v,st,m,l,m,x);
        modify2(v+v+1,m+1,en,m+1,r,x);
    }
    merge(v);
    return;
}
void add(int v,int st,int en,int x)
{
    if (st==en)
    {
        if (x==1)
        {
            if (tr2[v].y>=y1 && tr2[v].y<=y2) tr1[v].s=1;
            else tr1[v].s=0;
            tr1[v].y=tr1[v].x,tr1[v].x=-ding;
        } else
        {
            if (tr1[v].y>=x1 && tr1[v].y<=x2) tr1[v].s=1;
            else tr1[v].s=0;
            tr2[v].y=tr2[v].x,tr2[v].x=-ding;
        }
        return;
    }
    update1(v);
    update2(v);
    int m=(st+en) >> 1;
    if (x==1)
    {
        if (tr1[v+v].x>=x1) add(v+v,st,m,x);
        else add(v+v+1,m+1,en,x);
    } else
    {
        if (tr2[v+v].x>=y1) add(v+v,st,m,x);
        else add(v+v+1,m+1,en,x);
    }
    merge(v);
    return;
}
void del(int v,int st,int en,int x)
{
    if (st==en)
    {
        tr1[v].x=tr1[v].y=tr2[v].x=tr2[v].y=-ding;
        tr1[v].s=0;
        return;
    }
    update1(v);
    update2(v);
    int m=(st+en) >> 1;
    if (x==1)
    {
        if (tr1[v+v].y>x2) del(v+v,st,m,x);
        else del(v+v+1,m+1,en,x);
    } else
    {
        if (tr2[v+v].y>y2) del(v+v,st,m,x);
        else del(v+v+1,m+1,en,x);
    }
    merge(v);
    return;
}
void findd(int v,int st,int en,int l,int r)
{
    if (st==l && en==r)
    {
        s+=tr1[v].s;
        return;
    }
    update1(v);
    update2(v);
    int m=(st+en) >> 1;
    if (r<=m) findd(v+v,st,m,l,r);
    else if (l>m) findd(v+v+1,m+1,en,l,r);
    else
    {
        findd(v+v,st,m,l,m);
        findd(v+v+1,m+1,en,m+1,r);
    }
    merge(v);
    return;
}
int main()
{
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d",&n);
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        fr(i,1,n) scanf("%d%d",&a[i],&b[i]);
        maket(1,1,n);
        scanf("%d",&m);
        fr(i,1,m)
        {
            scanf("%d%d%d",&z,&l,&r);
            if (z!=3)
            {
                scanf("%d",&d);
                if (z==1)
                {
                    modify1(1,1,n,l,r,d);
                    while (tr1[1].x>=x1) add(1,1,n,1);
                    while (tr1[1].y>x2) del(1,1,n,1);
                } else if (z==2)
                {
                    modify2(1,1,n,l,r,d);
                    while (tr2[1].x>=y1) add(1,1,n,2);
                    while (tr2[1].y>y2) del(1,1,n,2);
                }
            } else
            {
                s=0;
                findd(1,1,n,l,r);
                printf("%d\n",s);
            }
        }
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章