UVA12130 Summits(BFS + 貪心)

UVA12130 Summits(BFS + 貪心)

題目連接

題目大意：
給你一個h ∗ w 的矩陣，矩陣的每個元素都有一個值，表明這個位置的高度。

題目要求你找出這個圖中有多少個位置是峯值點。從每個點（高度H）出發遍歷這個圖有一個要求。就是走過的點的高度不能小於等於H - d；成爲峯值點的要求就是從這個點出發走到的位置不能有高度大於H的。

解題思路：
因爲圖很是大。用dfs確定不行。將這些點依照高度從大到小的排序。而後每個點做爲起點來遍歷，假設找到比這個點大的點就說明不是峯值點。

並且遍歷的過程當中就會將途中走過的點標記上它能到的最大高度。假設下次要找的這個點已經被標記過了。就說明這個點可以到達更大的高度。確定不是峯值點，就不需要遍歷。

代碼：

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 505;

int G[maxn][maxn];
int vis[maxn][maxn];//標記對於單獨的一次遍歷中是否有反覆遍歷的點，標記可否夠到達更大的高度
int n, m, d;

const int dir[4][2] = {{0, -1}, {0, 1}, {-1, 0}, {1, 0}};

struct Node {

    int x, y, v;
}node[maxn * maxn], q[maxn * maxn];

int cmp (const Node &a, const Node &b) {
    return a.v > b.v;
}

int BFS (int k) {

    int front , rear;
    int cnt = 1;
    front = 0;
    rear = 1;
    q[front] = node[k];
    vis[node[k].x][node[k].y] = node[k].v;
    bool flag = 1;
    while (front < rear) {

        for (int i = 0; i < 4; i++) {

            int newx = q[front].x + dir[i][0];
            int newy = q[front].y + dir[i][1];
            if (newx < 0 || newx >= n || newy < 0 || newy >= m)    
                continue;

            if (G[newx][newy] > node[k].v) {
                flag = 0;//往下繼續遍歷
                continue;
            }
            if (vis[newx][newy] == node[k].v || node[k].v - G[newx][newy] >= d)
                continue;
            vis[newx][newy] = node[k].v;
            if (G[newx][newy] == node[k].v)
                cnt++;
            else 
                G[newx][newy] = node[k].v;
            q[rear].x = newx;
            q[rear++].y = newy;    
        }
        front++;
    }
    if (flag)
        return cnt;
    return 0;
}

int main () {

    int T;
    scanf ("%d", &T);
    while (T--) {

        scanf ("%d%d%d", &n, &m, &d);
        int cnt = 0;
        for (int i = 0; i < n; i++)
            for (int j = 0; j < m; j++) {
                scanf ("%d", &G[i][j]);
                node[cnt].x = i;
                node[cnt].y = j;
                node[cnt++].v = G[i][j];
            }

        sort (node, node + cnt, cmp);
        memset (vis, -1, sizeof (vis));    
        int ans = 0;
        for (int i = 0; i < cnt; i++) {
            if (vis[node[i].x][node[i].y] == -1) {
                ans += BFS(i);
            }
        }

        printf ("%d\n", ans);
    }
    return 0;
}