洛谷P2216 [HAOI2007]理想的正方形
阿新 • • 發佈:2018-12-16
今天新學了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; }