题意:
有一个n*n的矩形,每一列上都会有一个被标记的方块,所有被标记的方块的行数都不相同。一个矩形被称为美丽矩形当且仅当它的四个角正好有两个方块被标记。有q个询问,每次给你一个矩形,问你有多少被标记的矩形与这个矩形相交。
题解:
一个美丽矩形与这个矩形不相交当且仅当这两个被标记的方块在所选矩形的外侧同一侧,我们用主席数来维护,按x轴建树,存每一列被标记格子的行数。接下来只需要用一下容斥就好了:
如果我们要求黑色方框的相交完美矩形数,那么就是所有情况-红色区域的情况+绿色区域的情况。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+5;
ll sum[N];
int num[N*40],rt[N*40],ls[N*40],rs[N*40],tot;
void update(int l,int r,int root,int last,int val)
{
ls[root]=ls[last];
rs[root]=rs[last];
num[root]=num[last]+1;
if(l==r)
return ;
int mid=l+r>>1;
if(mid>=val)
update(l,mid,ls[root]=++tot,ls[last],val);
else
update(mid+1,r,rs[root]=++tot,rs[last],val);
num[root]=num[ls[root]]+num[rs[root]];
}
int query(int l,int r,int root,int last,int ql,int qr)
{
if(l>=ql&&r<=qr)
return num[root]-num[last];
int mid=l+r>>1;
int ans=0;
if(mid>=ql)
ans=query(l,mid,ls[root],ls[last],ql,qr);
if(mid<qr)
ans+=query(mid+1,r,rs[root],rs[last],ql,qr);
return ans;
}
int main()
{
ll n;
int q,x;
scanf("%lld%d",&n,&q);
for(ll i=1;i<=n;i++)
{
sum[i]=sum[i-1]+i;
scanf("%d",&x);
update(1,n,rt[i]=++tot,rt[i-1],x);
}
while(q--)
{
int l,r,d,u;
scanf("%d%d%d%d",&l,&d,&r,&u);
ll ans=sum[n-1];
if(u<n)
ans-=sum[query(1,n,rt[n],rt[0],u+1,n)-1];
if(d>1)
ans-=sum[query(1,n,rt[n],rt[0],1,d-1)-1];
if(l>1)
ans-=sum[query(1,n,rt[l-1],rt[0],1,n)-1];
if(r<n)
ans-=sum[query(1,n,rt[n],rt[r],1,n)-1];
if(l>1&&u<n)
ans+=sum[query(1,n,rt[l-1],rt[0],u+1,n)-1];
if(r<n&&u<n)
ans+=sum[query(1,n,rt[n],rt[r],u+1,n)-1];
if(l>1&&d>1)
ans+=sum[query(1,n,rt[l-1],rt[0],1,d-1)-1];
if(r<n&&d>1)
ans+=sum[query(1,n,rt[n],rt[r],1,d-1)-1];
printf("%lld\n",ans);
}
return 0;
}