题意
给出一个r*c的方格,其中有n个点被随机标记。求有多少个子矩阵满足其中至少有一个点被标记。
R,C<=40000,N<=100000,资源点的位置两两不同,且位置为随机生成。
分析
一道神题。
我们可以用总的子矩阵数量减去不包含任何资源点的子矩阵数量。那么我们现在就要求不包含任何资源点的子矩阵数量。
我一开始也想到了扫描线,但没想到用什么来维护答案。看了题解发现可以用treap。
我们可以一行一行地扫,统计以这行为底边的矩形数量就好,显然答案只跟每一列最多可以往上拓展多少格有关,也就是这一列的高度,设为h[i].我们可以对整个序列建立一棵treap,把每个点的h当做其key值,由于资源点位置是随机的,所以treap的期望高度为logn。这样treap上每一个节点所代表的矩形对答案的贡献就是
size[x]∗(size[x]+1)∗(h[fa]−h[x])
,因为这个矩形可以左右移动且高度可以任意取。那么每扫到一行,我们可以把treap所有点的h+1,同时把这一行上的标记位置的h设为0,再统计一下以这行为底边的矩形数量就好了。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long LL;
const int N=50005;
int n,m,T,root;
vector<int> pts[N];
struct tree{int l,r,tag,size,val;LL ans;}t[N];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void updata(int d)
{
t[d].size=t[t[d].l].size+t[t[d].r].size+1;
t[d].ans=0;
if (t[d].l) t[d].ans+=(LL)t[t[d].l].size*(t[t[d].l].size+1)/2*(t[t[d].l].val-t[d].val)+t[t[d].l].ans;
if (t[d].r) t[d].ans+=(LL)t[t[d].r].size*(t[t[d].r].size+1)/2*(t[t[d].r].val-t[d].val)+t[t[d].r].ans;
}
void pushdown(int d)
{
if (!t[d].tag) return;
int w=t[d].tag;t[d].tag=0;
if (t[d].l) t[t[d].l].tag+=w,t[t[d].l].val+=w;
if (t[d].r) t[t[d].r].tag+=w,t[t[d].r].val+=w;
}
int build(int l,int r)
{
int mid=(l+r)/2;
if (l<mid) t[mid].l=build(l,mid-1);
if (r>mid) t[mid].r=build(mid+1,r);
updata(mid);
return mid;
}
void rttr(int &x)
{
int y=t[x].l;
t[x].l=t[y].r;
t[y].r=x;
updata(x);updata(y);
x=y;
}
void rttl(int &x)
{
int y=t[x].r;
t[x].r=t[y].l;
t[y].l=x;
updata(x);updata(y);
x=y;
}
void modify(int &x,int y)
{
pushdown(x);
if (x==y)
{
t[y].val=0;updata(x);return;
}
if (y<x)
{
modify(t[x].l,y);
if (t[t[x].l].val<t[x].val) rttr(x);
}
else
{
modify(t[x].r,y);
if (t[t[x].r].val<t[x].val) rttl(x);
}
updata(x);
}
int main()
{
n=read();m=read();T=read();
while (T--)
{
int x=read(),y=read();
pts[x].push_back(y);
}
root=build(1,m);
LL ans=(LL)n*(n+1)/2*m*(m+1)/2;
for (int i=1;i<=n;i++)
{
t[root].tag++;t[root].val++;
for (vector<int>::iterator it=pts[i].begin();it!=pts[i].end();it++)
modify(root,*it);
ans-=t[root].ans+(LL)t[root].size*(t[root].size+1)/2*t[root].val;
}
printf("%lld",ans);
return 0;
}