bzoj2658[ZJOI2012]小蓝的好友(mrx)

题目链接
校内模拟赛出了这题，然后考试时写出了一个做法，评测机貌似有点慢被卡成90了，原题上可以A掉。首先可以知道\(O(n^2)\)的做法，就是只算全0的矩形个数，枚举下边界是哪一行，定义\(h\)值表示这个点向上碰到的第一个1的距离，然后可以从左到右枚举右边界，用单调栈维护以每个列作为左边界的矩形最大高度，那么以某一列作为右边界的对答案的贡献就是单调栈中的元素和。
然后数据随机，可以进一步优化这个做法，用线段树存每个点的高度值，当下边界向下移动时，相当于每个点的高度+1，且新出现的1位置的要设置成0。设新出现1的位置为x，由于每个跨过x的矩形的贡献都是0，要把它减掉，我们按照\(O(n^2)\)的想法，把x左边的单调栈通过线段树给做出来，然后显然后面的点r如果影响到了x以左的单调栈(就是存在一个\(l\le x\)且\(l\)~\(r-1\)的矩形最大高度与\(l\)~\(r\)的最大矩形高度不同)，才会使贡献变化(若没有影响贡献不变，直接用原来的贡献乘区间长度即可)，然后就维护一下x以左的单调栈。为了找到这样的r，可以从x开始每次往右找第一个高度小于x的点，x跳过去，这样的每个点都是会影响贡献值的点。
复杂度\(O(qklogn)\)，k是"从一个点开始每次向右跳到第一个比它小的点的期望跳的次数"，由于数据随机k的平均大小是小于\(logn\)级别的。
刚刚去看了看别人的做法，怎么都是用treap啊，只能说想不到想不到。实际运行速度我的做法顶多比treap慢一倍的样子，可能是因为线段树比treap快了蛮多吧。
\(O(n^2)\)的:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
#define P puts("lala")
#define cp cerr<<"lala"<<endl
#define ln putchar('\n')
#define pb push_back
#define fi first
#define se second 
#define shmem(x) cerr<<sizeof(x)/(1024*1024.0)<<"MB"
using namespace std;
inline int read()
{
    char ch=getchar();int g=1,re=0;
    while(ch<'0'||ch>'9') {if(ch=='-')g=-1;ch=getchar();}
    while(ch<='9'&&ch>='0') re=(re<<1)+(re<<3)+(ch^48),ch=getchar();
    return re*g;
}
typedef long long ll;
typedef pair<int,int> pii;

const int N=2050;
int dot[N][N],up[N][N];
int n,m,T;
pii stk[N]; int top=0;

void wj()
{
    freopen("alice.in","r",stdin);
    freopen("alice.out","w",stdout);
}
int main()
{
    wj();
    n=read(); m=read(); T=read();
    for(int i=1;i<=T;++i) dot[read()][read()]=1;
    for(int i=1;i<=n;++i) for(int j=1;j<=m;++j)
        up[i][j]=(dot[i][j]?0:up[i-1][j]+1);
    ll ans=0;
    for(int i=1;i<=n;++i)
    {
        ll sum=0;
        for(int j=1;j<=m;++j)
        {
            pii v=pii(up[i][j],1);
            while(top&&stk[top].fi>=v.fi) 
            {
                sum-=1ll*stk[top].fi*stk[top].se;
                v.se+=stk[top].se;
                top--;
            }
            stk[++top]=v;
            sum+=1ll*v.fi*v.se;
            ans+=sum;
        }
    }
    printf("%lld\n",(1ll*n*(n+1)/2)*(1ll*m*(m+1)/2)-ans);
    return 0;
}

\(O(qklogn)\)的:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
#include<ctime>
#define P puts("lala")
#define cp cerr<<"lala"<<endl
#define ln putchar('\n')
#define pb push_back
#define fi first
#define se second 
#define shmem(x) cerr<<sizeof(x)/(1024*1024.0)<<"MB"
using namespace std;
inline int read()
{
    char ch=getchar();int g=1,re=0;
    while(ch<'0'||ch>'9') {if(ch=='-')g=-1;ch=getchar();}
    while(ch<='9'&&ch>='0') re=(re<<1)+(re<<3)+(ch^48),ch=getchar();
    return re*g;
}
typedef long long ll;
typedef pair<int,int> pii;

const int N=70050;
int n,m,T;
int minv[N<<2],addv[N<<2],add=0;
inline void modify(int o,int l,int r,int x,int k)
{
    if(l==r) {minv[o]=k; return ;}
    int mid=l+r>>1;
    if(x<=mid) modify(o<<1,l,mid,x,k);
    else modify(o<<1|1,mid+1,r,x,k);
    minv[o]=min(minv[o<<1],minv[o<<1|1]);
}
inline int query(int o,int l,int r,int x)
{
    if(l==r) return minv[o];
    int mid=l+r>>1;
    if(x<=mid) return query(o<<1,l,mid,x);
    else return query(o<<1|1,mid+1,r,x);
}
int ret=0,found=0,val;
inline void gorig(int o,int l,int r,int x)
{
    if(l==r) {if(minv[o]<x) ret=l,found=1,val=minv[o];return ;}
    int mid=l+r>>1;
    if(minv[o<<1|1]<x) gorig(o<<1|1,mid+1,r,x);
    else gorig(o<<1,l,mid,x);
}
inline void findpre(int o,int l,int r,int x,int y,int k)
{
    if(found) return ;
    if(x<=l&&r<=y) 
    {
        if(minv[o]<k) gorig(o,l,r,k);
        return ;
    }
    int mid=l+r>>1;
    if(y>mid) findpre(o<<1|1,mid+1,r,x,y,k);
    if(x<=mid) findpre(o<<1,l,mid,x,y,k);
}
inline int getpre(int x,int y,int k)
{
    if(x>y) return 0;
    ret=0; found=0; val=0; findpre(1,1,m,x,y,k);
    return ret;
}
inline void golef(int o,int l,int r,int x)
{
    if(l==r) {if(minv[o]<x) ret=l,found=1,val=minv[o];return ;}
    int mid=l+r>>1;
    if(minv[o<<1]<x) golef(o<<1,l,mid,x);
    else golef(o<<1|1,mid+1,r,x);
}
inline void findnex(int o,int l,int r,int x,int y,int k)
{
    if(found) return ;
    if(x<=l&&r<=y) 
    {
        if(minv[o]<k) golef(o,l,r,k);
        return ;
    }
    int mid=l+r>>1;
    if(x<=mid) findnex(o<<1,l,mid,x,y,k);
    if(y>mid) findnex(o<<1|1,mid+1,r,x,y,k);
}
inline int getnex(int x,int y,int k)
{
    if(x>y) return m+1;
    ret=m+1; found=0; val=0; findnex(1,1,m,x,y,k);
    return ret;
}

pii stk[N],stk2[N]; int top=0,top2=0;
vector<int>ask[N];

void wj()
{
    freopen("alice.in","r",stdin);
    freopen("alice.out","w",stdout);
}
int main()
{
    wj();
    clock_t sta=clock();
    n=read(); m=read(); T=read();
    for(int i=1;i<=T;++i) 
    {
        int x=read(),y=read();
        ask[x].pb(y);
    }
    ll ans=0,sum=0;
    for(int row=1;row<=n;++row)
    {
        add++;
        sum+=1ll*(m+1)*m/2;
        for(int i=0,siz=ask[row].size();i<siz;++i)
        {
            top=0; top2=0;
            int in=ask[row][i];
            ll tot=0;
            int hi=query(1,1,m,in);
            while(in)
            {
                int las=getpre(1,in-1,hi);
                stk2[++top2]=pii(hi,in-las);
                sum-=1ll*(hi+add)*(in-las);
                tot+=1ll*(hi+add)*(in-las);
                in=las; hi=val;
            }

            top=top2;
            for(int j=1;j<=top;++j) stk[top-j+1]=stk2[j];
            in=ask[row][i];

            hi=query(1,1,m,in);
            while(in<=m)
            {
                int las=getnex(in+1,m,hi);
                sum-=1ll*(las-in)*tot;
                if(in==ask[row][i]) sum+=tot;

                if(las>m) break;
                pii v=pii((hi=val),0);
                while(top&&stk[top].fi>=v.fi) 
                {
                    tot-=1ll*(stk[top].fi+add)*stk[top].se;
                    v.se+=stk[top].se;
                    top--;
                }
                stk[++top]=v;
                tot+=1ll*(v.fi+add)*v.se;
                in=las;
            }
            modify(1,1,m,ask[row][i],-add);
        }
        ans+=sum;
    }
    printf("%lld\n",(1ll*n*(n+1)/2)*(1ll*m*(m+1)/2)-ans);
    //cerr<<(1ll*n*(n+1)/2)*(1ll*m*(m+1)/2)-ans<<endl;
    clock_t fin=clock();
    //cerr<<(double)(fin-sta)/CLOCKS_PER_SEC<<endl;
    return 0;
}

转载于:https://www.cnblogs.com/thkkk/p/8491922.html