Zerojudge e287

Zerojudge e287

題敘

https://zerojudge.tw/ShowProblem?problemid=e287
給一張地圖,起點從權重最小的點開始,每次往四周沒走過且權重最小的點移動,求移動權重的總和

想法

用DFS照做即可

Code

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
//By Koios1143
#include<bits/stdc++.h>
#define LL long long
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
#define pii pair<int,int>
using namespace std;
int n,m,ans=0,arr[105][105],dir[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
bool vis[105][105];
pii start={-1,-1};
bool cango(int x,int y){
if(x<0 || y<0 || x>=n || y>=m)
return false;
return true;
}
void dfs(pii s){
int x=s.first;
int y=s.second;
vis[x][y]=true;
ans+=arr[x][y];
int Min=INT_MAX;
pii next={-1,-1};
for(int i=0 ; i<4 ; i++){
int nx=x+dir[i][0];
int ny=y+dir[i][1];
if(!vis[nx][ny] && cango(nx,ny)){
if(arr[nx][ny]<Min){
next=make_pair(nx,ny);
Min=arr[nx][ny];
}
}
}
if(next.first==-1)
return;
else
dfs(next);
}
int main(){
IOS
cin>>n>>m;
int Min=INT_MAX;
for(int i=0 ; i<n ; i++){
for(int j=0 ; j<m ; j++){
cin>>arr[i][j];
if(arr[i][j]<Min){
start=make_pair(i,j);
Min=arr[i][j];
}
}
}
dfs(start);
cout<<ans<<"\n";
return 0;
}

複雜度

輸入時間複雜度為 $O(nm)$
DFS最多每個點都遍歷過,且每個點遍歷時間複雜度為 $O(1)$ ,DFS總時間複雜度為 $O(nm)$
總時間複雜度 $O(nm)$