P14889: 스타트와 링크

wnajsldkf·2023년 1월 18일
0

Algorithm

목록 보기
28/58
post-thumbnail

설명

팀 조합을 생성하고, 차를 구한 다음 값의 크기를 비교한다.
이를 코드로 어떻게 구현할까?

1. 팀 조합 생성하기

    private static void combi(int i, int depth) {
        if (depth == N / 2) {
            diff();
            return;
        }

        for (int j = i; j < N; j++) {
            visit[j] = true;
            combi(j + 1, depth + 1);
            visit[j] = false;
        }
    }
  • visit[] 배열은 해당 팀원 방문 여부를 나타낸다.
  • 반복문을 통해 팀원 한명한명의 방문 여부를 확인한다. 만약 해당 팀원을 방문하지 않았다면, 해당 지점부터 인접한 이웃을 살펴본다.
    • depth는 깊이를 몇명까지 확인했는지를 의미한다.
  • 만약 depth가 팀원 수의 절반이라면 조합이 생성되었다는 의미이니, 생성된 조합으로 value를 구한다.

2. 생성된 조합으로 value 구하기

    private static void diff() {
        int start_team = 0;
        int link_team = 0;

        for (int i = 0; i < N; i++) {
            for (int j = i; j < N; j++) {
                if (visit[i] == true && visit[j] == true) {
                    start_team += map[i][j];
                    start_team += map[j][i];
                } else if (visit[i] == false && visit[j] == false) {
                    link_team += map[i][j];
                    link_team += map[j][i];
                }
            }
        }

        value = Math.abs(start_team - link_team);
        min = Math.min(value, min);
    }
  • start_teamlink_team은 각각의 크기를 말한다.
  • visit[] 배열에서 true, false를 기준으로 조합이 생성되었다. 이를 확인하여 start_teamlink_team의 크기를 각각 구한다.
  • 두 크기의 차를 구하여, value를 찾고 작은 값을 min으로 지정한다.

코드

package Algorithm.P14889;

import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class P14889 {

    static int N;
    static int[][] map;
    static boolean[] visit;
    static int min = Integer.MAX_VALUE;
    static int value;
    static StringBuilder sb;

    public static void main(String[] args) throws IOException {
        System.setIn(new FileInputStream("src/Algorithm/P14889/input.txt"));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        sb = new StringBuilder();

        N = Integer.parseInt(st.nextToken());
        map = new int[N][N];
        visit = new boolean[N];

        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < N; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        combi(0, 0);
        sb.append(min);

        System.out.println(sb);
    }

    private static void combi(int i, int depth) {
        if (depth == N / 2) {
            diff();
            return;
        }

        for (int j = i; j < N; j++) {
            visit[j] = true;
            combi(j + 1, depth + 1);
            visit[j] = false;
        }
    }

    private static void diff() {
        int start_team = 0;
        int link_team = 0;

        for (int i = 0; i < N; i++) {
            for (int j = i; j < N; j++) {
                if (visit[i] == true && visit[j] == true) {
                    start_team += map[i][j];
                    start_team += map[j][i];
                } else if (visit[i] == false && visit[j] == false) {
                    link_team += map[i][j];
                    link_team += map[j][i];
                }
            }
        }

        value = Math.abs(start_team - link_team);
        min = Math.min(value, min);
    }
}
profile
https://mywnajsldkf.tistory.com -> 이사 중

0개의 댓글