1. 程式人生 > >BZOJ1047[HAOI2007]理想的正方形——二維ST表

BZOJ1047[HAOI2007]理想的正方形——二維ST表

 

題目描述

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

輸入

  第一行為3個整數,分別表示a,b,n的值第二行至第a+1行每行為b個非負整數,表示矩陣中相應位置上的數。每
行相鄰兩數之間用一空格分隔。
100%的資料2<=a,b<=1000,n<=a,n<=b,n<=1000

輸出

  僅一個整數,為a*b矩陣中所有“n*n正方形區域中的最大整數和最小整數的差值”的最小值。

樣例輸入

5 4 2
1 2 5 6
0 17 16 0
16 17 2 1
2 10 2 1
1 2 2 2

樣例輸出

1     二維ST表,f[i][j][k]表示以i,j為右下端點邊長為2k的正方形內的最大值,最小值同理,列舉每個點作為右下端點求一下最大值和最小值的差來更新答案即可。
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<cstdio>
#include<vector>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int a,b;
int n,m;
int f[1010][1010][12];
int g[1010][1010][12];
int ans=1e9+7;
int main()
{
    scanf("%d%d%d",&a,&b,&n);
    for(int i=1;i<=a;i++)
    {
        for(int j=1;j<=b;j++)
        {
            scanf("%d",&f[i][j][0]);
            g[i][j][0]=f[i][j][0];
        }
    }
    for(int i=1;i<=a;i++)
    {
        for(int j=1;j<=b;j++)
        {
            for(int k=1;k<=10;k++)
            {
                if(i<(1<<k)||j<(1<<k))
                {
                    continue;
                }
                f[i][j][k]=max(max(f[i][j][k-1],f[i-(1<<(k-1))][j-(1<<(k-1))][k-1]),max(f[i-(1<<(k-1))][j][k-1],f[i][j-(1<<(k-1))][k-1]));
                g[i][j][k]=min(min(g[i][j][k-1],g[i-(1<<(k-1))][j-(1<<(k-1))][k-1]),min(g[i-(1<<(k-1))][j][k-1],g[i][j-(1<<(k-1))][k-1]));
            }
        }
    }
    for(int i=0;;i++)
    {
        if((1<<i)<=n)
        {
            m=i;
        }
        else
        {
            break;
        }
    }
    for(int i=n;i<=a;i++)
    {
        for(int j=n;j<=b;j++)
        {
            int mx=max(max(f[i][j][m],f[i-n+(1<<m)][j-n+(1<<m)][m]),max(f[i][j-n+(1<<m)][m],f[i-n+(1<<m)][j][m]));
            int mn=min(min(g[i][j][m],g[i-n+(1<<m)][j-n+(1<<m)][m]),min(g[i][j-n+(1<<m)][m],g[i-n+(1<<m)][j][m]));
            ans=min(mx-mn,ans);
        }
    }
    printf("%d",ans);
}