BOJ 14938 : 서강그라운드

문제링크

풀이요약

플로이드 워셜

풀이상세

1. 입력 값을 통해 각 지역의 아이템 수를 저장한 1차원 배열과, 거리 기반의 2차원 배열을 저장한다.

2. 입력 값 외의 모든 거리는 INF로 설정한다.

3. 플로이드 워셜을 통해, 현재 주어진 경로와 임의의 노드로 돌아가는 경우를 비교했을 때, 임의의 노드보다 거쳐 가는 길이 더 작으면서, 지금까지의 거리 비용이 수색 경로 내에 포함되는 경우 업데이트 한다.

4. 반영된 거리 배열을 탐색하여, 각 시작점 부터, 갈 수 있는 지역의 아이템값을 모두 더하며 노드 간 최대 아이템 값을 반환한다. 갈 수 있는 지역은 거리 기반 2차원 배열에서 자신의 지역과 저장된 거리비용이 수색 경로 보다 작은 경우이다.

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

public class Main {
    static int N, M, R, a[][], t[], INF = 987654321, ans;

    private static void input() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer stk = new StringTokenizer(br.readLine());
        N = Integer.parseInt(stk.nextToken());
        M = Integer.parseInt(stk.nextToken());
        R = Integer.parseInt(stk.nextToken());

        a = new int[N + 1][N + 1];
        t = new int[N + 1];
        stk = new StringTokenizer(br.readLine());
        for (int i = 1; i <= N; i++) {
            t[i] = Integer.parseInt(stk.nextToken());
        }

        for (int i = 1; i <= N; i++) {
            for (int j = 1; j <= N; j++) {
                a[i][j] = INF;
            }
        }

        for (int i = 0; i < R; i++) {
            stk = new StringTokenizer(br.readLine());
            int st = Integer.parseInt(stk.nextToken());
            int ed = Integer.parseInt(stk.nextToken());
            int dist = Integer.parseInt(stk.nextToken());
            if(dist <= M) {
                a[st][ed] = dist;
                a[ed][st] = dist;
            }
        }
    }

    private static void solve() {
        for (int k = 1; k <= N; k++) {
            for (int i = 1; i <= N; i++) {
                for (int j = 1; j <= N; j++) {
                    if (a[i][j] > (a[i][k] + a[k][j]) && (a[i][k] + a[k][j]) <= M) {
                        a[i][j] = a[i][k] + a[k][j];
                    }
                }
            }
        }

        for (int i = 1; i <= N; i++) {
            int ret = 0;
            for (int j = 1; j <= N; j++) {
                if (i==j || a[i][j] <= M) ret += t[j];
            }
            ans = Math.max(ret, ans);
        }
    }

    private static void output() {
        System.out.println(ans);
    }

    public static void main(String[] args) throws Exception {
        input();
        solve();
        output();
    }
}