BZOJ.2738.矩阵乘法(整体二分二维树状数组)

整体二分。把求序列第K小的树状数组改成二维树状数组就行了。
初始答案区间有点大，离散化一下。

因为这题是一开始给点，之后询问，so可以先处理该区间值在l~mid的修改，再处理询问。即二分标准可以直接用点的标号。
结构体的赋值可以改为赋值操作的编号。（~~这样内存没那么连续？~~想多了你）

改了半下午，优化了500ms。。

//4444ms    7.17MB
#include <cstdio>
#include <cctype>
#include <algorithm>
#define lb(x) ((x)&-(x))
#define gc() getchar()
const int N=505,M=60005;

int n,m,Ans[M],q[M],q1[M],q2[M];
struct Point
{
    int x,y,val;
    Point() {}
    Point(int x,int y,int val):x(x),y(y),val(val) {}
    bool operator <(const Point &a)const{
        return val<a.val;
    }
}pt[N*N];
inline int read();
struct Operation//Query
{
    int K,x1,y1,x2,y2;
    inline void Input(){
        x1=read(),y1=read(),x2=read(),y2=read(),K=read();
    }
}op[M];

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
namespace T
{
    int n,t[N][N];

    inline void Modify(int x,int y,int v)
    {
        for(int i=x; i<=n; i+=lb(i))
            for(int j=y; j<=n; j+=lb(j)) t[i][j]+=v;
    }
    inline void Clear(int x,int y)
    {
        for(int i=x; i<=n; i+=lb(i))
            for(int j=y; j<=n; j+=lb(j))
                if(t[i][j]) t[i][j]=0; else break;
    }
    inline int Query(int x,int y)
    {
        int res=0;
        for(int i=x; i; i^=lb(i))
            for(int j=y; j; j^=lb(j)) res+=t[i][j];
        return res;
    }
    inline int Query_Area(Operation q){//prefix sum
        return Query(q.x2,q.y2)-Query(q.x1-1,q.y2)-Query(q.x2,q.y1-1)+Query(q.x1-1,q.y1-1);
    }
}
void Solve(int l,int r,int h,int t)
{
    if(h>t) return;
    if(l==r){
        for(int i=h; i<=t; ++i) Ans[q[i]]/*[op[q[i]].pos]*/=pt[l].val;
        return;
    }
    int mid=l+r>>1, t1=0, t2=0;
    for(int i=l; i<=mid; ++i) T::Modify(pt[i].x,pt[i].y,1);
    for(int now,tmp,i=h; i<=t; ++i)
    {
        now=q[i], tmp=T::Query_Area(op[now]);
        if(tmp>=op[now].K) q1[t1++]=now;
        else op[now].K-=tmp, q2[t2++]=now;
    }
    for(int i=l; i<=mid; ++i) T::Clear(pt[i].x,pt[i].y);
    for(int i=0; i<t1; ++i) q[h+i]=q1[i];
    for(int i=0; i<t2; ++i) q[h+t1+i]=q2[i];
    Solve(l,mid,h,h+t1-1), Solve(mid+1,r,h+t1,t);
}

int main()
{
    T::n=n=read(), m=read();
    int tot=0;
    for(int i=1; i<=n; ++i)
        for(int j=1; j<=n; ++j) pt[++tot]=Point(i,j,read());
    std::sort(pt+1,pt+1+tot);

    for(int i=1; i<=m; ++i) q[i]=i, op[i].Input();
    Solve(1,tot,1,m);
    for(int i=1; i<=m; ++i) printf("%d\n",Ans[i]);

    return 0;
}

优化前：

//4968ms    17.68MB
#include <cstdio>
#include <cctype>
#include <algorithm>
#define lb(x) ((x)&-(x))
#define gc() getchar()
const int N=505,M=60005+N*N;

int n,m,Q,cnt,A[N*N],Ans[60005];
struct Operation
{
    int K,x1,y1,x2,y2,pos;//K=0: Modify (x1,y1):=pos
    Operation() {}
    Operation(int K,int x1,int y1,int x2,int y2,int pos):K(K),x1(x1),y1(y1),x2(x2),y2(y2),pos(pos) {}
}q[M],q1[M],q2[M];

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
namespace T
{
    int n,t[N][N];

    inline void Modify(int x,int y,int v)
    {
        for(int i=x; i<=n; i+=lb(i))
            for(int j=y; j<=n; j+=lb(j)) t[i][j]+=v;
    }
    inline void Clear(int x,int y)
    {
        for(int i=x; i<=n; i+=lb(i))
            for(int j=y; j<=n; j+=lb(j))
                if(t[i][j]) t[i][j]=0; else break;
    }
    inline int Query(int x,int y)
    {
        int res=0;
        for(int i=x; i; i^=lb(i))
            for(int j=y; j; j^=lb(j)) res+=t[i][j];
        return res;
    }
    inline int Query_Area(Operation q){//prefix sum
        return Query(q.x2,q.y2)-Query(q.x1-1,q.y2)-Query(q.x2,q.y1-1)+Query(q.x1-1,q.y1-1);
    }
}
void Solve(int l,int r,int h,int t)
{
    if(h>t) return;
    if(l==r){
        for(int i=h; i<=t; ++i) if(q[i].K) Ans[q[i].pos]=A[l];
        return;
    }
    bool goon=0;
    for(int i=h; i<=t; ++i) if(q[i].K) {goon=1; break;}
    if(!goon) return;

    int mid=l+r>>1, midV=A[mid], t1=0, t2=0;
    for(int i=h; i<=t; ++i)
        if(q[i].K)
        {
            int tmp=T::Query_Area(q[i]);//这样好像少做几次加法！但是多copy两个int。。（你够了→_→）
            if(tmp>=q[i].K) q1[t1++]=q[i];
            else q[i].K-=tmp, q2[t2++]=q[i];
        }
        else
        {
            if(q[i].pos<=midV) T::Modify(q[i].x1,q[i].y1,1), q1[t1++]=q[i];
            else q2[t2++]=q[i];
        }
    for(int i=0; i<t1; ++i) if(!q1[i].K) T::Clear(q1[i].x1,q1[i].y1);
    for(int i=0; i<t1; ++i) q[h+i]=q1[i];
    for(int i=0; i<t2; ++i) q[h+t1+i]=q2[i];
    Solve(l,mid,h,h+t1-1), Solve(mid+1,r,h+t1,t);
}

int main()
{
    T::n=n=read(), m=read(), Q=0;
    for(int i=1; i<=n; ++i)
        for(int j=1; j<=n; ++j) q[++Q]=Operation(0,i,j,0,0,A[Q]=read());

    std::sort(A+1,A+1+Q), cnt=1;
    for(int i=2; i<=Q; ++i) if(A[i]!=A[i-1]) A[++cnt]=A[i];

    for(int x1,y1,x2,y2,i=1; i<=m; ++i)
        x1=read(),y1=read(),x2=read(),y2=read(),q[++Q]=Operation(read(),x1,y1,x2,y2,i);
    Solve(1,cnt,1,Q);
    for(int i=1; i<=m; ++i) printf("%d\n",Ans[i]);

    return 0;
}

BZOJ.2738.矩阵乘法(整体二分 二维树状数组)

猜你喜欢

BZOJ.2738.矩阵乘法(整体二分二维树状数组)