1. 程式人生 > >bzoj 2738: 矩陣乘法【整體二分+樹狀數組】

bzoj 2738: 矩陣乘法【整體二分+樹狀數組】

ons amp 主席樹 二分 %d r+ 乘法 根據 ostream

腦子一抽開始寫主席樹,敲了一會發現不對……
整體二分,用二維樹狀數組維護值為當前區間的格子個數,然後根據k的大小和當前詢問的子矩陣裏的值和k的大小關系來決定這個詢問放在哪一部分向下遞歸

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=300005;
int n,q,t[505][505],rl[N],ans[N],tot;
struct ques
{
    int x,y,xx,yy,k,id;
}f[N],g[N];
struct qwe
{
    int i,j,v;
}a[N];
bool cmp(const qwe &a,const qwe &b)
{
    return a.v<b.v;
}
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void update(int x,int y,int v)
{
    for(int i=x;i<=n;i+=(i&(-i)))
        for(int j=y;j<=n;j+=(j&(-j)))
            t[i][j]+=v;
}
int ques(int x,int y)
{
    int r=0;
    for(int i=x;i>=1;i-=(i&(-i)))
        for(int j=y;j>=1;j-=(j&(-j)))
            r+=t[i][j];
    return r;
}
void wk(int ll,int rr,int l,int r)
{
    if(a[ll].v==a[rr].v)
    {
        for(int i=l;i<=r;i++)
            ans[f[i].id]=a[ll].v;
        return;
    }
    int mid=(ll+rr)>>1,c1=l,c2=r;
    for(int i=ll;i<=mid;i++)
        update(a[i].i,a[i].j,1);
    for(int i=l;i<=r;i++)
    {
        int nw=ques(f[i].xx,f[i].yy)-ques(f[i].x-1,f[i].yy)-ques(f[i].xx,f[i].y-1)+ques(f[i].x-1,f[i].y-1);
        if(f[i].k<=nw)
            g[c1++]=f[i];
        else
        {
            f[i].k-=nw;
            g[c2--]=f[i];
        }
    }
    for(int i=l;i<=r;i++)
        f[i]=g[i];
    for(int i=ll;i<=mid;i++)
        update(a[i].i,a[i].j,-1);
    wk(ll,mid,l,c2);
    wk(mid+1,rr,c1,r);
}
int main()
{
    n=read(),q=read();
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            a[++tot]=(qwe){i,j,read()};
    sort(a+1,a+1+tot,cmp);
    for(int i=1;i<=q;i++)
        f[i].x=read(),f[i].y=read(),f[i].xx=read(),f[i].yy=read(),f[i].k=read(),f[i].id=i;
    wk(1,tot,1,q);
    for(int i=1;i<=q;i++)
        printf("%d\n",ans[i]);
    return 0;
}

bzoj 2738: 矩陣乘法【整體二分+樹狀數組】