bzoj 2658: [Zjoi2012]小蓝的好友(mrx) treap+扫描线

题意

给出一个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;
}