灾难预警题解报告

灾难预警题目链接
题目描述
众所周知，浙农林是一条河。
由于浙江农林大学的特殊地形，当你在下雨后漫步在农林大路上的时候 难免会出现一脚踩进一个水坑的情况的情况。而农农非常不喜欢踩到水坑的感觉，请你帮忙设计一个程序来帮助农农判断他能否在不踩入水坑的情况下回到寝室。

已知，浙江农林大学可以表示为一个 $N\ast N$ 的矩阵。
对于每个位置有一个海拔数据 $h[i][j]$，当水位高度大于 $h[i][j]$ 的时候，这个位置就会形成一个水坑。
农农现在的坐标是 (1,1), 他的宿舍位于 (n,n).农农只可以沿着上下左右四个方向走。
假如农农现在位于 （2， 2）那么在不考虑水位的情况下，他可以去的地方有 (1,2)，(2,1)，(3,2)，（2,3)
输入格式

N （表示矩阵大小）
接下来 N 行为一个 N * N 的矩阵 h
Q （表示询问数量）
接下来 Q 行每行一个数字，表示当前水位 X 

1 <= N <= 1000
1 <= h[i][j] <= 100000
1 <= X <= 100000
1 <= Q <= 100000

输出格式

共 Q 行，表示在对应水位下，农农能否在不踩入水坑的情况下
回到寝室, 如果农农可以回到寝室，请你输出“Wuhu”， 反之请输出
“Hmmm” 

输入样例

4 
5 2 3 2
4 5 3 4
2 1 4 5
3 3 3 3
2
1
5

输出样例

Wuhu
Hmmm

最开始用并查集初始化把所有点都看作是一个单独的集合，遍历扩展按照高度从大到小遍历，如果扩展的点位置合法，并且每次扩展如果是从高度小的扩展到高度大的，则合并这两个点，最后判断起点和终点是否连通，如果连通，把当前让其连通点的高度记录下来，就是那个连通的最大雨量是多少，超过这个雨量无法到达终点。


扩展图解自己自行手动画一下就明白了。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e3 + 7, maxm = 1e6 + 7;
int H,h[maxn][maxn]; //存储高度
int n, q, cnt;
struct node{
    int x, y, k;  // x:横坐标 y:纵坐标 k:高度
    bool operator < (const node &t) const{
        return k < t.k;  //按照高度从小到大排序
    }
};
node a[maxm];
int fa[maxm];
int dir[4][2] = {0, 1, -1, 0, 0, -1, 1, 0};
void init(int x)  //并查集初始化
{
    for(int i = 1; i <= x; i++) fa[i] = i;
}
int Find(int x)  //路径压缩 如果fa[x]的父节点不是x则递归访问fa[x]的父节点
{
    if(fa[x] != x) fa[x] = Find(fa[x]);
    return fa[x];
}
void Merge(int x, int y)  //合并 y的代表元 作为 x的父节点
{
    int fax = Find(x), fay = Find(y);
    if(fax != fay) fa[fax] = fay;  //x 和 y 不在一个集合，则合并
}
bool inbound(int x, int l, int r)  //扩展方向是否合法 [1,n]
{
    if(x < l || x > r) return false;
    return true;
}
int main()
{
    scanf("%d",&n);
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++){
            scanf("%d",&h[i][j]);
            a[++cnt] = {i, j, h[i][j]}; //存储每个点
        }

    sort(a + 1, a + 1 + cnt);
    init(cnt);  
    int ans = 0; //记录连通块能否走通
    int st = 1, ed = n*n;  //起点 终点
    for(int i = cnt; i; i--) //从大到小遍历
    {
        int x = a[i].x, y = a[i].y, k = a[i].k;   //当前点的(x,y) 和 h
        int num = (x - 1)*n + y;    //当前点是几号点
        for(int k = 0; k < 4; k++)  //四个扩展方向
        {  //tx是从x走过来  ty是从y走过来
            int tx = x + dir[k][0], ty = y + dir[k][1];
            int tnum = (tx - 1)*n + ty;  //扩展的点是几号点
            if(!inbound(tx, 1, n) || !inbound(ty, 1,  n)) continue;  //位置非法
            if(h[tx][ty] >= h[x][y]) Merge(num, tnum);  //一个连通块
        }
        if(Find(st) == Find(ed)) //判断起点和终点是否连通了
        {
            ans = k; 
            break;
        }
    }
    scanf("%d",&q);  //q次询问
    while(q--)
    {
        scanf(" %d",&H);
        if(H<= ans) puts("Wuhu");
        else puts("Hmmm");
    }
    return 0;
}