1. 程式人生 > >洛谷P2216 [HAOI2007]理想的正方形

洛谷P2216 [HAOI2007]理想的正方形

今天新學了RMQ資料結構,開開心心地刷了一波題。(嗯~~~~我真棒!)

從容易到簡單 

以上全是套上RMQ模板的題,如果想練練手的可以試試,難度其實不大(第五題可是省選!!!!!!!)(其實都是假的。。。)省選也是簡單套用RMQ模板,不過變態的是它超細節的模擬,這心覺得變態,過了好幾次才過了。(省選哪有那麼容易過。)

題目描述

有一個a*b的整陣列成的矩陣,現請你從中找出一個n*n的正方形區域,使得該區域所有數中的最大值和最小值的差最小。

其實這道題就是把RMQ從一維擴充套件成二維,稍微推理一下,一維(線段)只有兩個頂點,二維(平面)就有四個頂點(至於三維。。。自己想想吧),於是就有

f[k][i][j] = Max(f[k-1][i][j],f[k-1][ti][j],f[k-1][i][tj],f[k-1][ti][tj]);

z[k][i][j] = Min(z[k-1][i][j],z[k-1][ti][j],z[k-1][i][tj],z[k-1][ti][tj]);

其實這樣已經差不多了,思路很簡單,但是程式碼很考驗細節,為了節省程式碼打了三個define,自定義了Min和Max(emm~~~~~~~~)

最後附上鄙人的程式碼

#include <iostream>
#include <cstdio>
#define re register int 
#define ti i+(1<<(k-1))
#define tj j+(1<<(k-1))
#define tk (1<<k)
using namespace std;
const int N = 1011;
int f[15][N][N],z[15][N][N],log[N];
int a,b,n;
inline int read(){
    int x=0,f=1;
    char ch = getchar();
    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;
}
inline int Min(int a,int b,int c,int d){return min(a,min(b,min(c,d)));}
inline int Max(int a,int b,int c,int d){return max(a,max(b,max(c,d)));}
inline int qMax(int x,int y,int x1,int y1){
    re k = log[n];
    re t1 = f[k][x][y], t2=f[k][x1-tk+1][y];	//錯誤:t2、t3重複了
    re t3 = f[k][x][y1-tk+1], t4=f[k][x1-tk+1][y1-tk+1];
    return Max(t1,t2,t3,t4);
}
inline int qMin(int x,int y,int x1,int y1){
    re k = log[n];
    re t1 = z[k][x][y], t2=z[k][x1-tk+1][y];	//錯誤:t2、t3重複了 
    re t3 = z[k][x][y1-tk+1], t4=z[k][x1-tk+1][y1-tk+1];
    return Min(t1,t2,t3,t4);
}
int main(){
    a=read(); b=read(); n=read();
    log[0] = -1;//***** 
    for(re i=1; i<=a; i++){
        for(re j=1; j<=b; j++){
            f[0][i][j] = z[0][i][j] = read();
        }
    }
    for(re i=1; i<=1001; i++){
        log[i] = log[i>>1]+1;
    }
    for(re k=1; k<=log[min(a,b)]; k++){
        for(re i=1; i+tk-1<=a; i++){
            for(re j=1;j+tk-1<=b; j++){
                f[k][i][j] = Max(f[k-1][i][j],f[k-1][ti][j],f[k-1][i][tj],f[k-1][ti][tj]);
                z[k][i][j] = Min(z[k-1][i][j],z[k-1][ti][j],z[k-1][i][tj],z[k-1][ti][tj]);
            }//z寫成f,該去看看眼科了 
        }
    }
    int ans = 0x7fffffff;
    for(re i=1; i<=a-n+1; i++){
        for(re j=1; j<=b-n+1; j++){
            ans = min(ans,qMax(i,j,i+n-1,j+n-1)-qMin(i,j,i+n-1,j+n-1));
	    }
    }
    printf("%d",ans);
    return 0;
}