1. 程式人生 > >POJ 2110 bfs 二分+列舉區間

POJ 2110 bfs 二分+列舉區間

題意:  給你一個N*N的網格,並給出網格中每個點的海拔高度,現在要你找出從左上角到右下角那個點的一條路,且這條路所有點中,最高海拔-最低海拔之差最小。輸出該最小值。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=105;
int map[maxn][maxn];
int vis[maxn][maxn];
int n;
int dr[]={-1,1,0,0};//上下左右
int dc[]={0,0,-1,1};
struct Node
{
    int r,c;
    Node(int r,int c):r(r),c(c){}
};
bool BFS(int low,int up)
{
    if(map[1][1]<low || map[1][1]>up) return false;
    queue<Node> Q;
    Q.push(Node(1,1));
    vis[1][1]=1;
    while(!Q.empty())
    {
        Node node=Q.front();Q.pop();
        int r=node.r,c=node.c;
        for(int dir=0;dir<4;dir++)
        {
            int nr=r+dr[dir] , nc=c+dc[dir];
            if(nr>=1&&nr<=n&&nc>=1&&nc<=n&&!vis[nr][nc])
            {
                vis[nr][nc]=1;
                if(map[nr][nc]>=low && map[nr][nc]<=up)
                {
                    Q.push(Node(nr,nc));
                    if(nr==n&&nc==n) return true;      //錯誤,這句話一定要放在上面那個if裡面且把nc寫成了nr
                }
            }
        }
    }
    return false;
}
bool check(int d)
{
    for(int low=0;low+d<=110;low++)
    {
        memset(vis,0,sizeof(vis));
        if(BFS(low,low+d)) return true;
    }
    return false;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
            scanf("%d",&map[i][j]);
    int L=0,R=110;
    while(R>L)      //用二分依次列舉高度差,並以此用bfs驗證是否可行。
    {
        int mid=(R+L)>>1;
        if(check(mid)) R=mid;
        else L=mid+1;
    }
    printf("%d\n",R);
    return 0;
}