結果
| 問題 |
No.848 なかよし旅行
|
| コンテスト | |
| ユーザー |
 kusomushi
|
| 提出日時 | 2019-07-09 04:15:31 |
| 言語 | Java (openjdk 23) |
| 結果 |
AC
|
| 実行時間 | 631 ms / 2,000 ms |
| コード長 | 6,113 bytes |
| コンパイル時間 | 2,884 ms |
| コンパイル使用メモリ | 83,968 KB |
| 実行使用メモリ | 55,144 KB |
| 最終ジャッジ日時 | 2024-11-08 00:59:42 |
| 合計ジャッジ時間 | 11,891 ms |
|
ジャッジサーバーID （参考情報） |
judge2 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 26 |
ソースコード
import java.io.*;
import java.util.*;
import java.util.function.Function;
public class Main {
static int N, M, P, Q, T;
static Edge[] E;
public static void main(String[] args) {
FastScanner sc = new FastScanner(System.in);
N = sc.nextInt();
M = sc.nextInt();
P = sc.nextInt()-1;
Q = sc.nextInt()-1;
T = sc.nextInt();
E = new Edge[M];
for (int i = 0; i < M; i++) {
E[i] = new Edge(sc.nextInt()-1, sc.nextInt()-1, sc.nextInt());
}
System.out.println(solve());
}
static long solve() {
Edge[][] G = adjB(N, E);
long[] distR = calcDist(0, G);
long[] distP = calcDist(P, G);
long[] distQ = calcDist(Q, G);
if( distR[P] + distP[Q] + distQ[0] <= T ) {
return T;
}
long ans = -1;
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
// iで別れてjで待ち合わせる
long ri = distR[i];
long ip = distP[i];
long iq = distQ[i];
long pj = distP[j];
long qj = distQ[j];
long jr = distR[j];
if( ri + ip + pj + jr > T || ri + iq + qj + jr > T ) {
continue; // 間に合わない
} else {
// 別行動してる時間だけ引く
ans = Math.max(ans, T - Math.max(ip+pj, iq+qj));
}
}
}
return ans;
}
static long[] calcDist(int root, Edge[][] G) {
long[] dist = new long[N];
Arrays.fill(dist, Long.MAX_VALUE);
PriorityQueue<State> q = new PriorityQueue<>(Comparator.comparingLong(s -> s.d));
q.add(new State(root, 0));
dist[root] = 0;
while( !q.isEmpty() ) {
State s = q.poll();
if( dist[s.a] != s.d ) continue;
for (Edge e : G[s.a]) {
int b = e.a == s.a ? e.b : e.a;
if( dist[b] > dist[s.a] + e.c ) {
dist[b] = dist[s.a] + e.c;
q.add( new State(b, dist[b]) );
}
}
}
return dist;
}
static class State {
int a;
long d;
public State(int a, long d) {
this.a = a;
this.d = d;
}
}
static Edge[][] adjB(int n, Edge[] E) {
Edge[][] adj = new Edge[n][];
int[] cnt = new int[n];
for (Edge e : E) {
cnt[e.a]++;
cnt[e.b]++;
}
for (int i = 0; i < n; i++) {
adj[i] = new Edge[cnt[i]];
}
for (Edge e : E) {
adj[e.a][--cnt[e.a]] = e;
adj[e.b][--cnt[e.b]] = e;
}
return adj;
}
static class Edge {
int a, b, c;
public Edge(int a, int b, int c) {
this.a = a;
this.b = b;
this.c = c;
}
}
@SuppressWarnings("unused")
static class FastScanner {
private BufferedReader reader;
private StringTokenizer tokenizer;
FastScanner(InputStream in) {
reader = new BufferedReader(new InputStreamReader(in));
tokenizer = null;
}
String next() {
if (tokenizer == null || !tokenizer.hasMoreTokens()) {
try {
tokenizer = new StringTokenizer(reader.readLine());
} catch (IOException e) {
throw new RuntimeException(e);
}
}
return tokenizer.nextToken();
}
String nextLine() {
if (tokenizer == null || !tokenizer.hasMoreTokens()) {
try {
return reader.readLine();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
return tokenizer.nextToken("\n");
}
long nextLong() {
return Long.parseLong(next());
}
int nextInt() {
return Integer.parseInt(next());
}
int[] nextIntArray(int n) {
int[] a = new int[n];
for (int i = 0; i < n; i++)
a[i] = nextInt();
return a;
}
int[] nextIntArray(int n, int delta) {
int[] a = new int[n];
for (int i = 0; i < n; i++)
a[i] = nextInt() + delta;
return a;
}
long[] nextLongArray(int n) {
long[] a = new long[n];
for (int i = 0; i < n; i++)
a[i] = nextLong();
return a;
}
}
static <A> void writeLines(A[] as, Function<A, String> f) {
PrintWriter pw = new PrintWriter(System.out);
for (A a : as) {
pw.println(f.apply(a));
}
pw.flush();
}
static void writeLines(int[] as) {
PrintWriter pw = new PrintWriter(System.out);
for (int a : as) pw.println(a);
pw.flush();
}
static void writeLines(long[] as) {
PrintWriter pw = new PrintWriter(System.out);
for (long a : as) pw.println(a);
pw.flush();
}
static int max(int... as) {
int max = Integer.MIN_VALUE;
for (int a : as) max = Math.max(a, max);
return max;
}
static int min(int... as) {
int min = Integer.MAX_VALUE;
for (int a : as) min = Math.min(a, min);
return min;
}
static void debug(Object... args) {
StringJoiner j = new StringJoiner(" ");
for (Object arg : args) {
if (arg instanceof int[]) j.add(Arrays.toString((int[]) arg));
else if (arg instanceof long[]) j.add(Arrays.toString((long[]) arg));
else if (arg instanceof double[]) j.add(Arrays.toString((double[]) arg));
else if (arg instanceof Object[]) j.add(Arrays.toString((Object[]) arg));
else j.add(arg.toString());
}
System.err.println(j.toString());
}
}