[백준] 14938. 서강그라운드

문제 링크

[백준] 14938. 서강그라운드

풀이 과정

출발지에서 목적지로 이동하되, 수색범위를 만족시킬 최소 경로로 이동해야 한다는 점에서 플로이드-워셜 알고리즘을 사용했습니다.

for (int i = 1; i <= N; i++) {
    int cnt = items[i];

    for (int j = 1; j <= N; j++) {
        if (adj[i][j] <= M) {
            cnt += items[j];
        }
    }

    answer = Math.max(answer, cnt);
}

플로이드-워셜 알고리즘을 적용시킨 후, adj[i][j]는 출발지 i에서 도착지 j로 이동하는데에 필요한 최소 거리를 담고 있습니다. 이 값이 수색범위 M 이하인 경우, i에서 j로 조건하에 이동할 수 있다는 것을 의미하므로, 아이템 수를 더해줍니다.

코드

import java.util.Arrays;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int M = sc.nextInt();
        int R = sc.nextInt();
        int[] items = new int[N + 1];
        int[][] adj = new int[N + 1][N + 1];

        for (int i = 1; i <= N; i++) items[i] = sc.nextInt();
        for (int i = 1; i <= N; i++) Arrays.fill(adj[i], 987654321);

        for (int r = 0; r < R; r++) {
            int u = sc.nextInt();
            int v = sc.nextInt();
            int cost = sc.nextInt();
            adj[u][v] = adj[v][u] = cost;
        }

        for (int k = 1; k <= N; k++) {
            for (int i = 1; i <= N; i++) {
                if (i == k) continue;

                for (int j = 1; j <= N; j++) {
                    if (j == i || j == k) continue;

                    if (adj[i][j] > adj[i][k] + adj[k][j]) {
                        adj[i][j] = adj[i][k] + adj[k][j];
                    }
                }
            }
        }

        int answer = 0;

        for (int i = 1; i <= N; i++) {
            int cnt = items[i];

            for (int j = 1; j <= N; j++) {
                if (adj[i][j] <= M) {
                    cnt += items[j];
                }
            }

            answer = Math.max(answer, cnt);
        }

        System.out.println(answer);
    }
}