一开始很蠢地用单调队列模拟,得出了一个O(a*b*n)的方法,然后果断炸掉了。
遂改良了一下,用maxv[i][j]和minv[i][j]分别表示点(i,j)往下垂的一小条(长度为n)的最大最小值,然后再模拟一遍,但是每次正方形移动的时候并不是把所有点都加进去,只加最大最小的点,于是就是O(a*b)。然后就A了^_^。
#include<iostream> #include<cstdio> #include<cstring> #include<deque> using namespace std; const int inf=1e9; inline int read(){ int x=0,f=1;char ch; 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; } deque<int>qmax,qmin; void push(int x){ while(!qmax.empty()&&qmax.back()<x)qmax.pop_back();qmax.push_back(x); while(!qmin.empty()&&qmin.back()>x)qmin.pop_back();qmin.push_back(x); } int pop(int x){ if(x==qmax.front())qmax.pop_front(); if(x==qmin.front())qmin.pop_front(); } int getmax(){ return qmax.front(); } int getmin(){ return qmin.front(); } void clear(){ qmax.clear(); qmin.clear(); } int mat[1005][1005],maxv[1005][1005],minv[1005][1005]; int main(){ int a,b,n;a=read();b=read();n=read(); for(int i=1;i<=a;i++) for(int j=1;j<=b;j++)mat[i][j]=read(); for(int i=1;i<=b;i++){ clear(); for(int j=1;j<=n;j++)push(mat[j][i]); maxv[1][i]=getmax();minv[1][i]=getmin(); for(int j=2;j+n-1<=a;j++){ push(mat[j+n-1][i]);pop(mat[j-1][i]); maxv[j][i]=getmax();minv[j][i]=getmin(); } } int ans=inf; for(int i=1;i+n-1<=a;i++){ clear(); for(int j=1;j<=n;j++)push(maxv[i][j]),push(minv[i][j]); ans=min(ans,getmax()-getmin()); for(int j=2;j+n-1<=b;j++){ push(maxv[i][j+n-1]);push(minv[i][j+n-1]); pop(maxv[i][j-1]);pop(minv[i][j-1]); ans=min(ans,getmax()-getmin()); } } printf("%d",ans); return 0; }