인프런 커뮤니티 질문&답변

정태승님의 프로필 이미지
정태승

작성한 질문수

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

3-B

3-B 시간 복잡도..!

해결된 질문

작성

·

249

1

#include <bits/stdc++.h>

using namespace std;

int N, M, res;
int vis[54][54];
char bd[54][54];
string s;
vector<pair<int,int>> land;
vector<int> select_land;

int dy[4] = {0, -1, 0, 1};
int dx[4] = {1, 0, -1, 0};

void bfs(){
    pair<int,int> st = land[select_land[0]]; // land
    pair<int, int> ed = land[select_land[1]]; // land
    
    queue<pair<int,int>> Q;
    
    Q.push({st.first, st.second}); // Q삽입
    vis[st.first][st.second] = 1; // 방문표시
    
    
    while(!Q.empty()){
        pair<int, int> cur = Q.front(); Q.pop();
        
        for(int dir = 0 ; dir < 4 ; dir++){
            int ny = cur.first + dy[dir];
            int nx = cur.second + dx[dir];
            
            if(ny==ed.first && nx==ed.second) {
                vis[ny][nx] = vis[cur.first][cur.second] + 1;
                res = max(res, vis[ny][nx]);
                return; // 도착하면 종료
            }
            
            if(ny >= N || nx >= M || ny < 0 || nx < 0) continue;
            if(bd[ny][nx]=='W' || vis[ny][nx]!=0) continue; // 방문했거나 물이면 pass
            
            Q.push({ny, nx});
            vis[ny][nx] = vis[cur.first][cur.second] + 1;
            
        }
    }
    
    // 도착하지 못했다면, 그냥 for문을 빠져나옴. (res 업데이트 필요 X)
    
}

void solve(int n){
    if(select_land.size()==2){ // 2개를 고름.
        bfs();
        fill(&vis[0][0], &vis[0][0] + 54*54, 0); // vis배열 초기화
        return;
    }
    
    for(int i = n+1 ; i < land.size() ; i++){  // 땅 전체 개수 (0번부터 land.size()-1번까지)
        select_land.push_back(i); // 01, 02, 03 .....
        solve(i);
        select_land.pop_back();
    }
    
    return;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    
    cin >> N >> M;

    for(int i = 0 ; i < N ; i++){
        cin >> s;
        for(int j = 0 ; j < M ; j++){
            bd[i][j] = s[j];
            if(s[j]=='L') land.push_back({i,j});
        }
    } // 입력 받기
    
    solve(-1);
    
    cout << res-1;
}

저는 랜드를 모두 구해서 랜드 중에 2개를 뽑고, 최단거리를 구하고 최단거리중 최대값을 뽑는 로직으로 구해봤습니다.

그런데 시간복잡도에서 계속 걸리네요 ㅠㅠ 혹시 크기가 50*50이라 재귀적으로 하려했는데 이 코드는 시간복잡도가 어떻게 되나요 ㅠㅠ

답변 1

0

큰돌님의 프로필 이미지
큰돌
지식공유자

안녕하세요 태승님 ㅎㅎ

    for(int i = n+1 ; i < land.size() ; i++){  // 땅 전체 개수 (0번부터 land.size()-1번까지)
        select_land.push_back(i); // 01, 02, 03 .....
        solve(i);
        select_land.pop_back();
    }

예를 들어

0 >> (1, 2, 3, 4...)

1 >> (2, 3, 4 ... )

이런로직이죠?

(N * M)까지니

(N * M) !

이 되게 됩니다.

BFS 자체는 N * M이기 때문에 해당 시간복잡도를 곱해

합을 계산하면

(N * M) *(N * M) !

이 됩니다.

 

그래서 저렇게 짜면 안됩니다 ㅠㅠ

 

감사합니다.

정태승님의 프로필 이미지
정태승
질문자

감사합니다 ㅠㅠ엄청나네요 .........

정태승님의 프로필 이미지
정태승

작성한 질문수

질문하기