DFS로는 맞춰서 BFS로 풀어보았습니다!
227
작성한 질문수 10
BFS로 풀었을 때 답은 맞는데 제출하면 메모리 초과로 틀렸다고 나오는데,
혹시 이 코드에서 메모리를 줄여서 맞힐 수 있는 방법이 있는지 여쭤보고 싶습니다!
#include<bits/stdc++.h>
using namespace std;
int N;
const int dy[4] = {-1,0,1,0};
const int dx[4] = {0,1,0,-1};
const int max_n = 100;
int arr[max_n][max_n];
int visited[max_n][max_n];
int x,y;
int Max = 0;
int main(){
cin >> N;
fill(&arr[0][0], &arr[0][0] + max_n*max_n, 0);
for(int i = 0; i < N; i++){
for(int j = 0; j < N; j++){
cin >> arr[i][j];
}
}
for(int h = 1; h <= 100; h++){
int cnt = 0;
fill(&visited[0][0], &visited[0][0] + max_n*max_n, 0);
for(int i = 0; i < N; i++){
for(int j = 0; j < N; j++){
if(arr[i][j] >= h && visited[i][j] == 0){
queue<pair<int,int>> q;
q.push({i,j});
cnt++;
while(q.size()){
tie(y,x) = q.front();
q.pop();
visited[y][x] = 1;
for(int n = 0; n < 4; n++){
int ny = y + dy[n];
int nx = x + dx[n];
if(ny < 0 || ny >= N || ny < 0 || ny >= N){
continue;
}
if(arr[ny][nx] < h){
continue;
}
if(visited[ny][nx]){
continue;
}
q.push({ny,nx});
}
}
}
}
}
Max = max(Max, cnt);
}
cout << Max;
return 0;
}
답변 1
0
안녕하세요 민석님 ㅎㅎ
bfs로는 이렇게 하시면 됩니다.
queue에 넣을 때는 반드시 visited를 걸고 넣는다라는 것을 명심해주세요.
#include<bits/stdc++.h>
using namespace std;
const int N_MAX = 100;
int N;
const int dy[4] = {-1, 0, 1, 0};
const int dx[4] = {0, 1, 0, -1};
int arr[N_MAX][N_MAX];
int visited[N_MAX][N_MAX];
int Max = 0;
void bfs(int i, int j, int h) {
queue<pair<int, int>> q;
q.push({i, j});
visited[i][j] = 1;
while(!q.empty()) {
auto [y, x] = q.front();
q.pop();
for(int n = 0; n < 4; n++) {
int ny = y + dy[n], nx = x + dx[n];
if(ny < 0 || ny >= N || nx < 0 || nx >= N) continue;
if(arr[ny][nx] < h || visited[ny][nx]) continue;
q.push({ny, nx});
visited[ny][nx] = 1;
}
}
}
int main() {
cin >> N;
for(int i = 0; i < N; i++) {
for(int j = 0; j < N; j++) {
cin >> arr[i][j];
}
}
for(int h = 1; h <= 100; h++) {
fill(&visited[0][0], &visited[0][0] + N_MAX*N_MAX, 0);
int cnt = 0;
for(int i = 0; i < N; i++) {
for(int j = 0; j < N; j++) {
if(arr[i][j] >= h && visited[i][j] == 0) {
bfs(i, j, h);
cnt++;
}
}
}
Max = max(Max, cnt);
}
cout << Max;
return 0;
}
그리고 기존 코드에서
if(ny < 0 || ny >= N || ny < 0 || ny >= N){이부분이 잘못되서 UB가 난 것 같습니다. 경계체크는 항상 주의해주세요. ㅎㅎ
또 질문 있으시면 언제든지 질문 부탁드립니다.
좋은 수강평과 별점 5점은 제게 큰 힘이 됩니다. :)
감사합니다.
강사 큰돌 올림.
문제를 고민하는 시간 관련
0
13
2
코딩살구클럽
0
26
2
코딩살구클럽 문의
0
30
2
코딩살구클럽 승인
0
31
2
DP 경우의 수 설명이 이해가 되지 않습니다.
0
32
2
3-F 채점 관련 질문
0
29
1
BFS, DFS 활용이 되는 상황에서의 방향성
0
32
2
코딩살구클럽 승인
0
43
2
코딩살구클럽승인
0
39
3
코딩살구클럽 승인
0
51
2
3-D 관련 질문
0
35
2
코살구 회원가입 문의
0
45
2
코살구 로그인 문제
0
65
2
3-A 문제 풀이 관련 질문
0
56
3
2-O 질문 있습니다
0
38
2
2-T 문제에 관한 질문
0
40
2
코딩 살구 클럽 접속 및 사용방법 문의
0
63
2
안녕하세요~. 현재 코살코딩클럽 사이트가 접속이 안됩니다~
0
64
2
코딩살구클럽 로그인문제
0
78
3
코딩 살구 클럽 로그인 문제
0
85
2
2-J 채점관련 질문
0
67
3
코딩 살구 클럽 Python 지원 가능 여부
0
77
1
살구클럽 아이디 없음 문제
0
76
1
1-O 코딩살구클럽 채점관련 질문
0
61
2





