61-tgyuuAn

[tgyuuAn]

🔗 문제 링크

우주신과의 교감

✔️ 소요된 시간

30분

✨ 수도 코드

디스코드 추천문제 채널 에서 나왔던

별자리 그리기 문제랑 매우 매우 같습니다!

어.. 이 문제를 푸려면 크루스칼 알고리즘에 대해서 알아야 하는데요..

최소 스패닝 트리라고 부르기도 해요..

근데 위 알고리즘을 알려고 한다면 유니온 파인드 알고리즘에 대해서 알아야 해요...

흠... 조만간 알고리즘 노트 하나 만들겠습니다...........!

📚 새롭게 알게된 내용

[alstjr7437]

오호 이번에 학교 강의 중 알고리즘 (재)수강을 하게 되어서 유니온 파인드 알고리즘을 배우고 C언어로 구현을 했었는데
아래와 같이 교황군의 도움을 받아 작성했었습니다!!!

C언어 유니온 파인드

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#define MAX_NODE 1000
#define MAX_EDGES 10000

typedef struct {
    int start;
    int end;
    int weight;
} Edge;

int adjacency_matrix[MAX_NODE][MAX_NODE];
int result_matrix[MAX_NODE][MAX_NODE];

Edge edges[MAX_EDGES]; // 간선 배열
Edge result_edge[MAX_EDGES];
int parent[MAX_NODE];
int size[MAX_NODE];

int node_count = 0;
int edge_count = 0; // 현재 간선의 개수
int result_count = 0; // 현재 간선의 개수

// 파일 input 및 간선 만들기
int file_input(char *argv[]){
    FILE* fp = fopen(argv[1], "r");
    int max_node = 0;
    if (fp == NULL) {
        printf("File Not Found\n");
        exit(1);
    }

    char buffer[MAX_NODE];
    // 파일 읽어오기
    while (fgets(buffer, MAX_NODE, fp)) {
        char* p = strtok(buffer, " ");
        int col = 0;
        while (p) {
            adjacency_matrix[node_count][col++] = atoi(p);
            p = strtok(NULL, " ");
        }
        node_count++;
    }

    return 1;
}

// 정렬 함수
int static compare (const void* first, const void* second)
{
    if (((Edge *)first)->weight > ((Edge *)second)->weight)
        return 1;
    else if (((Edge *)first)->weight < ((Edge *)second)->weight)
        return -1;
    else
        return 0;
}

void edge_setting(){
    // 파일 읽어온 부분 Edge로 만들어주기
    for(int i = 0; i < node_count; ++i) {
        for(int j = i + 1; j < node_count; ++j) {
            if(adjacency_matrix[i][j] == -1){
                continue;
            }

            edges[edge_count].start = i;
            edges[edge_count].end = j;
            edges[edge_count].weight = adjacency_matrix[i][j];
            edge_count++;
        }
    }

    // weight별로 정렬해주기
    qsort(edges, edge_count, sizeof(Edge), compare);

    // // 정렬한 간선 출력하기
    // for (int i = 0; i < edge_count; ++i) {
    //     printf("간선 %d: start=%d, end=%d, weight=%d\n", i+1, edges[i].start, edges[i].end, edges[i].weight);
    // }

    for(int i = 0; i < edge_count; i++){
        parent[i] = i;
    }
}

// 부모 찾기 
int find(int node){
    while(node != parent[node]){
        parent[node] = parent[parent[node]];
        node = parent[node];
    }
    return parent[node];
}

// cycle 인지 확인하기(부모가 같은지)
int is_cycle(int node1, int node2){
    if (find(node1) == find(node2)){
        return 1;
    } else {
        return 0;
    }
}

// union , find
void unite(int node1, int node2){
    int parent1 = find(node1);
    int parent2 = find(node2);

    // 한쪽에 몰리지 않기 위해 dept 사용
    if(size[parent1] > size[parent2]){
        parent[parent2] = parent1;
        size[parent1] += size[parent2];
    } else {
        parent[parent1] = parent2;
        size[parent2] += size[parent1];
    }
}

// krusckal 실행(모두 실행하면서 사이클인지 확인하기)
void krusckal(){
    for(int i = 0; i < edge_count; i++){
        int start = edges[i].start;
        int end = edges[i].end;
        int weight = edges[i].weight;

        if(!is_cycle(start,end)){
            unite(start, end);
            result_edge[result_count++] = edges[i];
        }
    }
}

// 결과 출력 함수
void result_print(){
    // 메트릭스 만들기
    memset(result_matrix, -1, sizeof(result_matrix));
    for (int i = 0; i < node_count; i++){
        result_matrix[i][i] = 0;
    }

    for (int i = 0; i < result_count; i++){
        int start = result_edge[i].start;
        int end = result_edge[i].end;

        result_matrix[start][end] = result_edge[i].weight;
        result_matrix[end][start] = result_edge[i].weight;
    }

    for(int i = 0; i < node_count; i++){
        printf("\t%c", 'A' + i);
    }
    printf("\n");

    for(int i = 0; i < node_count; i++){
        printf("%c ", 'A' + i); // 노드 출력
        for(int j = 0; j < node_count; j++){
            printf("\t%d", result_matrix[i][j]);
        }
        printf("\n");
    }
}

int main(int argc, char *argv[]) {
    file_input(argv);

    edge_setting();

    krusckal();

    // // krusckal로 만들어진 간선 출력하기
    // for (int i = 0; i < result_count; ++i) {
    //     printf("간선 %d: start=%d, end=%d, weight=%d\n", i+1, result_edge[i].start, result_edge[i].end, result_edge[i].weight);
    // }
    
    result_print();

    return 0;
}

아직 파이썬으로 유니온파인드 작성을 해본 적은 없는데 따로 바쁘지 않게 시간이 되면(아마 다음주쯤..)에 노트나 정리를 한번 해야겠군요 잊어버리지 않게!!

[alstjr7437]

이 부분은 좌표들의 제곱 합을 구하고 루트 씌워서 피타고라스 정리 이용해서 두점 사이의 거리 구하는 방법인가요???

[tgyuuAn]

맞씁니다!!!!!!!!!!!!!!!!!!!!!!!