codeforces 713 D（二維st表）

題目鏈接

題解

題意：給你一個01矩陣，詢問一個矩形區域內最大的全1正方形。

考慮到硬做很麻煩，所以先二分出一個值就可以了。

st表時間複雜度：n^2*log^2 

 

代碼：

#include<bits/stdc++.h>
using namespace std;
int Q,n,m,ans,XX,YY,XXX,YYY,LLL[1010],a[1010][1010],dp[1010][1010],f[1010][1010][10][10];
int query(int XX,int YY,int XXX,int YYY){
	int n1=LLL[XXX-XX+1],m1=LLL[YYY-YY+1];
	return max(max(f[XX][YY][n1][m1],f[XX][YYY-(1<<m1)+1][n1][m1]),
	 max(f[XXX-(1<<n1)+1][YY][n1][m1],f[XXX-(1<<n1)+1][YYY-(1<<m1)+1][n1][m1]));
}
int pd(int x){
	return query(XX+x-1,YY+x-1,XXX,YYY)>=x;
}
int main(){
	int i,j,ii,jj,l,r,mid;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++)
	 for(j=1;j<=m;j++)scanf("%d",&a[i][j]);
	for(i=1;i<=n;i++)
	 for(j=1;j<=m;j++){
	   if(a[i][j]==1)dp[i][j]=min(dp[i-1][j-1],min(dp[i][j-1],dp[i-1][j]))+1;   
	   f[i][j][0][0]=dp[i][j];
 }
   for(i=2;i<=1000;i++)LLL[i]=LLL[i>>1]+1;
   for(ii=1;ii<=LLL[n];ii++)
    for(i=1;i+(1<<ii)-1<=n;i++)
     for(j=1;j<=m;j++)
      f[i][j][ii][0]=max(f[i][j][ii-1][0],f[i+(1<<(ii-1))][j][ii-1][0]);
    for(ii=0;ii<=LLL[n];ii++)
	 for(jj=1;jj<=LLL[m];jj++) 
	  for(i=1;i+(1<<ii)-1<=n;i++)
	   for(j=1;j+(1<<jj)-1<=m;j++)
	    f[i][j][ii][jj]=max(f[i][j][ii][jj-1],f[i][j+(1<<(jj-1))][ii][jj-1]);
	scanf("%d",&Q);
	while(Q--){
		ans=0;
		scanf("%d%d%d%d",&XX,&YY,&XXX,&YYY);
		l=0;r=min(XXX-XX+1,YYY-YY+1);
		while(l<=r){
			mid=(l+r)/2;
			if(pd(mid)){
				ans=max(ans,mid);
				l=mid+1;
			}
			 else r=mid-1;
		}
		printf("%d\n",ans);
	}	  
}