bzoj4716假摔

bzoj4716假摔

题意：

给出一个矩阵，求这个矩阵中权值和第k小的长在xmin到n之间，宽在ymin到m之间的子矩阵。n，m≤1000，k≤250000。

题解：

首先求出长为xmin，宽为ymin的子矩阵放入优先队列，每次取出时如果该矩阵之前没有出现过（用set判重），则将其扩展并放入优先队列，输出第k个不重复的（这里指位置不重复的，权值可以相等）。

代码：

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <queue>
 #include <set>
 #define inc(i,j,k) for(int i=j;i<=k;i++)
 #define maxn 1010
 using namespace std;

 inline int read(){
     char ch=getchar(); int f=,x=;
     while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();}
     while(ch>=''&&ch<='')x=x*+ch-'',ch=getchar();
     return f*x;
 }
 int sum[maxn][maxn],n,m,xmin,ymin,k,tot;
 struct nd{
     int x,y,l,c,v;
     bool operator < (const nd &a)const{
         if(v!=a.v)return v>a.v; if(x!=a.x)return x<a.x; if(y!=a.y)return y<a.y;
         if(l!=a.l)return l<a.l; return c<a.c;
     };
 };
 priority_queue<nd>q; set<nd>st;
 int main(){
     n=read(); m=read(); xmin=read(); ymin=read(); k=read();
     inc(i,,n)inc(j,,m){int x=read(); sum[i][j]=sum[i-][j]-sum[i-][j-]+sum[i][j-]+x;}
     inc(i,,n-xmin+)inc(j,,m-ymin+){
         q.push((nd){i,j,xmin,ymin,sum[i+xmin-][j+ymin-]-sum[i+xmin-][j-]-sum[i-][j+ymin-]+sum[i-][j-]});
     }
     while(!q.empty()){
         nd x=q.top(); q.pop(); if(st.find(x)!=st.end())continue;
         tot++; if(tot==k){printf("%d",x.v+); break;} st.insert(x);
         if(x.x+x.l<=n)
             q.push((nd){x.x,x.y,x.l+,x.c,sum[x.x+x.l][x.y+x.c-]-sum[x.x+x.l][x.y-]-sum[x.x-][x.y+x.c-]+sum[x.x-][x.y-]});
         if(x.y+x.c<=m)
             q.push((nd){x.x,x.y,x.l,x.c+,sum[x.x+x.l-][x.y+x.c]-sum[x.x+x.l-][x.y-]-sum[x.x-][x.y+x.c]+sum[x.x-][x.y-]});
         if(x.x+x.l<=n&&x.y+x.c<=m)
             q.push((nd){x.x,x.y,x.l+,x.c+,sum[x.x+x.l][x.y+x.c]-sum[x.x+x.l][x.y-]-sum[x.x-][x.y+x.c]+sum[x.x-][x.y-]});
     }
     return ;
 }

20161110