第1部分 基礎算法(提高篇)--第4章 廣搜的優化技巧1454:山峯和山谷

1454:山峯和山谷

時間限制: 1000 ms 內存限制: 65536 KB
提交數: 534 通過數: 213
【題目描述】
給定一個 n×n 的網格狀地圖,每個方格 (i,j)有一個高度 wij​​ 。如果兩個方格有公共頂點,則它們是相鄰的。

定義山峯和山谷如下:

均由地圖上的一個連通塊組成;

所有方格高度都相同;

周圍的方格(即不屬於山峯或山谷但與山峯或山谷相鄰的格子)高度均大於山谷的高度,或小於山峯的高度。

求地圖內山峯和山谷的數量。特別地,如果整個地圖方格的高度均相同,則整個地圖既是一個山谷,也是一個山峯。

【輸入】
第一行一個整數n(2≤n≤1000),表示地圖的大小。

接下來 n 行每行 n 個整數表示地圖。第 i 行有 n 個整數 wi1,wi2,…,win(0≤wij≤1000000000),表示地圖第 i 行格子的高度。

【輸出】
輸出一行兩個整數,分別表示山峯和山谷的數量。

【輸入樣例】
5
8 8 8 7 7
7 7 8 8 7
7 7 7 7 7
7 8 8 7 8
7 8 8 8 8
【輸出樣例】
2 1
【提示】
樣例1解釋:
在這裏插入圖片描述

樣例輸入2:

5
5 7 8 3 1
5 5 7 6 6
6 6 6 2 8
5 7 2 5 8
7 1 0 1 7
樣例輸出2:

3 3
樣例2解釋:在這裏插入圖片描述
山峯:8(兩塊),7
山谷:1,0,5
6 不是山峯也不是山谷:6所圍成的塊周圍有比它大的7,也有比他小的2,其他同理.
思路:
廣搜的剪枝 ,每個點進行廣搜,每搜完一個點,把與它相連接的點標記,不重複的搜每個格子(i,j) 的高度w(i,j)是給定的。
若兩個格子有公共頂點,那麼他們就是相鄰的格子。(所以與(i,j)相鄰的格子有(i−1, j−1),(i−1,j),(i−1,j+1),(i,j−1),(i,j+1),(i+1,j−1),(i+1,j),(i+1,j+1))。

#include<iostream>
#include<cstdlib>
#include<cstdio>
#define now a[qx[tot]][qy[tot]]
using namespace std;
int a[1005][1005];
int n,ans[3],flag,f,nx,ny;
int fx[8]={0,0,1,-1,1,1,-1,-1};
int fy[8]={1,-1,0,0,-1,1,-1,1};
int qx[2000005],qy[2000005],tot,tow;
bool vis[1005][1005];
char ch;
inline void read(int &s){//輸入優化
    s=0;ch=getchar();f=1;
    for (;ch<'0'||ch>'9';ch=getchar())
        if (ch=='-')f = -1;
    for (;ch>='0'&&ch<='9';ch=getchar())
        s=(s<<3)+(s<<1)+ch-'0';
}
inline void bfs(int x,int y){//深搜
    qx[tot=1]=x;qy[tow=1]=y;vis[x][y]=1;
    while (tot<=tow){
         
        for (int i = 0;i < 8;++i){
            nx = qx[tot]+fx[i];
            ny = qy[tot]+fy[i];
            if (nx < 1 || nx > n || ny < 1 || ny > n)continue;
            if (a[nx][ny] == now && !vis[nx][ny]){
                qx[++tow] = nx;
                qy[tow] = ny;
                vis[nx][ny]=1;
            }
            else if (a[nx][ny]<now){
                if (flag != 1 && flag != -1)flag=0;
                else flag=1;
            }
            else if (a[nx][ny]>now){
                if (flag != 2 && flag != -1)flag=0;
                else flag=2;
            }
        }
        ++tot;
    }
}
int main (){
    read(n);
    for (int i = 1;i <= n;++i)
        for (int j = 1;j <= n;++j)
            read(a[i][j]);
    for (int i = 1;i <= n;++i)
        for (int j = 1;j <= n;++j)
            if (!vis[i][j]){
                flag = -1;
                bfs(i,j);
                if (flag != -1)ans[flag]++;
                else {ans[1]++;ans[2]++;}
            }
    printf ("%d %d",ans[1],ans[2]);
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章