백준 31871번 영일랜드 Java

: ) YOUNG·2024년 6월 2일
1

알고리즘

목록 보기
372/411
post-thumbnail

백준 31871번
https://www.acmicpc.net/problem/31871

문제



생각하기


  • 백트래킹 그래프 문제입니다.

  • 간선 중복을 제거하는 것이 핵심인 것 같습니다.



동작

        adjList = new ArrayList<>();
        edgeMap = new HashMap<>();
        for (int i = 0; i <= N + 1; i++) {
            adjList.add(new ArrayList<>());
            edgeMap.put(i, new HashMap<>());
        }
        isVisited = new boolean[N + 1];

edgeMap의 HashMap을 통해서 중복되는 간선 중 최대값을 가진 간선만 활용해서 인접리스트를 구현합니다.




결과


코드



import java.io.*;
import java.util.*;

public class Main {

    // input
    private static BufferedReader br;

    // variables
    private static int N, M;
    private static long ans;
    private static List<List<Edge>> adjList;
    private static boolean[] isVisited;
    private static Map<Integer, Map<Integer, Integer>> edgeMap;

    private static class Edge {
        int num;
        int dist;

        private Edge(int num, int dist) {
            this.num = num;
            this.dist = dist;
        }
    } // End of Edge class

    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        input();

        bw.write(solve());
        bw.close();
    } // End of main()

    private static String solve() {
        StringBuilder sb = new StringBuilder();

        DFS(0, 0, 0);
        sb.append(ans);
        return sb.toString();
    } // End of solve()

    private static void DFS(int node, int dist, int count) {
        if (node == 0 && count == N + 1) {
            ans = Math.max(ans, dist);
            return;
        }

        for (Edge next : adjList.get(node)) {
            if (isVisited[next.num]) continue;

            isVisited[next.num] = true;
            DFS(next.num, dist + next.dist, count + 1);
            isVisited[next.num] = false;
        }
    } // End of DFS()

    private static void input() throws IOException {
        N = Integer.parseInt(br.readLine());
        M = Integer.parseInt(br.readLine());
        ans = -1;

        adjList = new ArrayList<>();
        edgeMap = new HashMap<>();
        for (int i = 0; i <= N + 1; i++) {
            adjList.add(new ArrayList<>());
            edgeMap.put(i, new HashMap<>());
        }
        isVisited = new boolean[N + 1];

        for (int i = 0; i < M; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int u = Integer.parseInt(st.nextToken());
            int v = Integer.parseInt(st.nextToken());
            int d = Integer.parseInt(st.nextToken());

            if (u == v) continue; // 자신한테로 오는 간선은 제거

            Map<Integer, Integer> curMap = edgeMap.get(u);
            if (curMap.containsKey(v)) {
                curMap.put(v, Math.max(curMap.get(v), d));
            } else {
                curMap.put(v, d);
            }
        }

        for (int i = 0; i <= N; i++) {
            Map<Integer, Integer> con = edgeMap.get(i);

            for (int key : con.keySet()) {
                int d = con.get(key);
                adjList.get(i).add(new Edge(key, d));
            }
        }
    } // End of input()
} // End of Main class

0개의 댓글