본문 바로가기
Algorithm/백준풀이

[알고리즘 문제풀이] 백준 2098 외판원순회 (JAVA코드)

by 계범 2022. 1. 16.

https://www.acmicpc.net/problem/2098

 

2098번: 외판원 순회

첫째 줄에 도시의 수 N이 주어진다. (2 ≤ N ≤ 16) 다음 N개의 줄에는 비용 행렬이 주어진다. 각 행렬의 성분은 1,000,000 이하의 양의 정수이며, 갈 수 없는 경우는 0이 주어진다. W[i][j]는 도시 i에서 j

www.acmicpc.net

 

실패해서 타 블로그들을 참조해서 풀은 문제

 

/**
 * 
 * 실패 ㅠ
 * 
 * 본인에서 시작해서 본인으로 돌아오는 가장 적은 비용을 쓴 경로
 * 
 * 외판원 순회 알고리즘 사용..
 * 
 * DP + dfs + 비트마스킹
 * 
 * DP[현재도시][방문했던도시정보(bit)] = 이후에 방문하는것들 포함한 최소비용
 * 
 * 어느 도시에서 출발했던지 최소비용 사이클을 찾는거기때문에
 * 1번 도시 시작으로 하겠음
 * 
 * dfs를 돌면서 어떤 도시를 방문할건지 선택
 * 
 * 최종 다돌았을때, 마지막에서 처음으로 돌아오는 값을 더해줌
 * 
 * 
 */
import java.util.*;
import java.io.*;

public class BJ2098_외판원순회 {
    
    public static int n;
    public static int[][] arr, dp;
    public static int INF = 1_000_000*16;
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringTokenizer st;

        n = Integer.parseInt(br.readLine());
        
        arr = new int[n][n];
        dp = new int[n][(1<<n)-1];
        for(int i = 0; i < n; i++){
            Arrays.fill(dp[i],INF);
            st = new StringTokenizer(br.readLine());
            for(int j = 0; j < n; j++){
                arr[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        //첫번째도시 방문
        bw.write(dfs(0,1) +"\n");
        bw.flush();
        bw.close();
    }

    // 최종 비용 반환 dfs
    public static int dfs(int city, int visited){

        // 전체 다 방문했을 때
        if(visited == (1<<n) -1){
            // 마지막도착지점에서 스타트지점 확인
            // 0이면 불가이므로 inf 반환
            if(arr[city][0] == 0){
                return INF;
            }
            return arr[city][0];
        }

        // 저장된 dp면 반환
        if(dp[city][visited] != INF){
            return dp[city][visited];
        }

        // 다음 도시 방문하기
        for(int i = 0; i < n; i++){
            // 아직 방문하지않았고 길이 있으면 방문
            if((visited&(1<<i)) == 0 && arr[city][i] != 0){
                // 이후 방문비용 담기
                dp[city][visited] = Math.min(dp[city][visited],dfs(i,visited|(1<<i)) + arr[city][i]);
            }
        }

        return dp[city][visited];
    }


}

댓글