Boredom CodeForces - 854E

http://codeforces.com/problemset/problem/854/E

和图形学的剪切算法有些相通之处

对每一次查询 将整个n*n的平面划分为9个区域 其中5是题目查询的区间

1 2 3

4 5 6

7 8 9

对于1区中的点 在5 6 8 9中出现的点对于1区中点结合 对答案构成贡献

对于2区中的点 在4 5 6 7 8 9中出现的点对于2区中点结合 对答案构成贡献

...

对于5区中的点 在1 2 3 4 6 7 8 9中出现的点对于5区中点结合 对答案构成贡献 并且还要算上5中的点两两配对

#include <bits/stdc++.h>
using namespace std;
#define ll long long

struct node
{
    int l;
    int r;
    int val;
};

node tree[8000010];
int pos[200010],root[200010];
int n,q,num;

int build(int l,int r)
{
    int cur,m;
    cur=num++;
    tree[cur].l=0,tree[cur].r=0,tree[cur].val=0;
    if(l==r) return cur;
    m=(l+r)/2;
    tree[cur].l=build(l,m);
    tree[cur].r=build(m+1,r);
    return cur;
}

int update(int rot,int tar,int val,int l,int r)
{
    int cur,m;
    cur=num++;
    tree[cur]=tree[rot];
    tree[cur].val+=val;
    if(l==r) return cur;
    m=(l+r)/2;
    if(tar<=m) tree[cur].l=update(tree[rot].l,tar,val,l,m);
    else tree[cur].r=update(tree[rot].r,tar,val,m+1,r);
    return cur;
}

int query(int lrot,int rrot,int pl,int pr,int l,int r)
{
    int res,m;
    if(pl<=l&&r<=pr) return tree[rrot].val-tree[lrot].val;
    res=0,m=(l+r)/2;
    if(pl<=m) res+=query(tree[lrot].l,tree[rrot].l,pl,pr,l,m);
    if(pr>m) res+=query(tree[lrot].r,tree[rrot].r,pl,pr,m+1,r);
    return res;
}

int main()
{
    ll v[10];
    ll ans;
    int i,l,d,r,u;
    scanf("%d%d",&n,&q);
    for(i=1;i<=n;i++) scanf("%d",&pos[i]);
    num=0;
    root[0]=build(1,n);
    for(i=1;i<=n;i++)
    {
        root[i]=update(root[i-1],pos[i],1,1,n);
    }
    while(q--)
    {
        scanf("%d%d%d%d",&l,&d,&r,&u);
        memset(v,0,sizeof(v));
        if(l-1>=1&&d-1>=1) v[1]=query(root[0],root[l-1],1,d-1,1,n);
        if(d-1>=1) v[2]=query(root[l-1],root[r],1,d-1,1,n);
        if(r+1<=n&&d-1>=1) v[3]=query(root[r],root[n],1,d-1,1,n);

        if(l-1>=1) v[4]=query(root[0],root[l-1],d,u,1,n);
        v[5]=query(root[l-1],root[r],d,u,1,n);
        if(r+1<=n) v[6]=query(root[r],root[n],d,u,1,n);

        if(l-1>=1&&u+1<=n) v[7]=query(root[0],root[l-1],u+1,n,1,n);
        if(u+1<=n) v[8]=query(root[l-1],root[r],u+1,n,1,n);
        if(r+1<=n&&u+1<=n) v[9]=query(root[r],root[n],u+1,n,1,n);

        //for(i=1;i<=9;i++) printf("%lld ",v[i]);
        //printf("\n");

        ans=v[5]*(v[5]-1);
        ans+=v[1]*(v[5]+v[6]+v[8]+v[9])+v[2]*(v[4]+v[5]+v[6]+v[7]+v[8]+v[9])+v[3]*(v[4]+v[5]+v[7]+v[8]);
        ans+=v[4]*(v[2]+v[3]+v[5]+v[6]+v[8]+v[9])+v[5]*(v[1]+v[2]+v[3]+v[4]+v[6]+v[7]+v[8]+v[9])+v[6]*(v[1]+v[2]+v[4]+v[5]+v[7]+v[8]);
        ans+=v[7]*(v[2]+v[3]+v[5]+v[6])+v[8]*(v[1]+v[2]+v[3]+v[4]+v[5]+v[6])+v[9]*(v[1]+v[2]+v[4]+v[5]);

        printf("%lld\n",ans/2);
    }
    return 0;
}