백준 24445 - 알고리즘 수업 - 너비 우선 탐색 2 (C++)

문제 링크입니다. https://www.acmicpc.net/problem/24445

24445번: 알고리즘 수업 - 너비 우선 탐색 2

https://junseok.tistory.com/268

백준 24444 - 알고리즘 수업 - 너비 우선 탐색 1 (C++)

해당 문제에서 내림차순으로 바뀐 문제입니다. 문제 설명은 해당 링크를 참고해주세요.

정렬할 때, 오름차순이 아닌, 내림차순으로 바꿔주면 됩니다.

sort를 한 뒤, reverse를 하는 방법도 있겠지만, 정렬 기준을 잡아주는 방법으로 풀었습니다.

bool cmp(int x, int y){
    if(x > y) return true;
    else return false;
}

해당 코드가 정렬 기준을 잡아주는 코드입니다. return 값이 참이라면 정렬X, 거짓이라면 정렬을 하라는 의미입니다.

 

 

자세한 것은 코드를 참고해주세요.

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
#include <string>
#include <cmath>
#include <cstdio>

using namespace std;

int N, M, K;
vector<int> connect[100001];
int check[100001];

bool cmp(int x, int y){
    if(x > y) return true;
    else return false;
}

void bfs(int start){
    int Visit = 1;
    queue<int> q;
    q.push(start);
    check[start] = Visit++;
    while(!q.empty()){
        int x = q.front();
        q.pop();
        for(int i = 0; i < connect[x].size(); i++){
            int xx = connect[x][i];
            if(check[xx] == 0){
                check[xx] = Visit++;
                q.push(xx);
            }
        }
    }
}

void solve(){
    for(int i = 1; i <= N; i++){
        sort(connect[i].begin(), connect[i].end(), cmp);
    }
    bfs(K);
    for(int i = 1; i <= N; i++){
        cout << check[i] << "\n";
    }
}

int main() {
    cin.tie(0);
    cout.tie(0);
    cin >> N >> M >> K;
    for(int i = 1; i <= M; i++){
        int x, y;
        cin >> x >> y;
        connect[x].push_back(y);
        connect[y].push_back(x);
    }
    solve();
    return 0;
}

 

 

질문 및 조언은 댓글 남겨주세요