結果

問題 No.416 旅行会社
ユーザー Grenache
提出日時 2016-08-28 16:01:47
言語 Java
(openjdk 23)
結果
AC  
実行時間 1,459 ms / 4,000 ms
コード長 4,923 bytes
コンパイル時間 4,339 ms
コンパイル使用メモリ 81,124 KB
実行使用メモリ 94,152 KB
最終ジャッジ日時 2024-12-14 20:12:06
合計ジャッジ時間 20,552 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

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


public class Main_yukicoder416 {

    public static void main(String[] args) {
    	Scanner sc = new Scanner(System.in);
    	Printer pr = new Printer(System.out);

		int n = sc.nextInt();
		int m = sc.nextInt();
		int q = sc.nextInt();

		Set<Long> hs = new HashSet<>();
		for (int i = 0; i < m; i++) {
			int a = sc.nextInt() - 1;
			int b = sc.nextInt() - 1;

			hs.add((long)a << 32 | b);
		}

		Dijkstra di = new Dijkstra(n);
		for (int i = 0; i < q; i++) {
			int c = sc.nextInt() - 1;
			int d = sc.nextInt() - 1;
			di.addEdge(c, d, q - i);
			di.addEdge(d, c, q - i);

			hs.remove((long)c << 32 | d);
		}

		for (long e : hs) {
			int a = (int)(e >> 32);
			int b = (int)(e & 0xffff_ffff);

			di.addEdge(a, b, 0);
			di.addEdge(b, a, 0);
		}

		for (int i = 1; i < n; i++) {
			long d = di.getShortestPath(0, i);

			if (d == 0) {
				pr.println(-1);
			} else if (d == di.INF) {
				pr.println(0);
			} else {
				pr.println(q - d + 1);
			}
		}

    	pr.close();
        sc.close();
    }

	private static class Dijkstra {
		long[] d;
		List<Edge>[] edges;
		PriorityQueue<Vertex> pq;
		int n;
		int s;

		final long INF = Long.MAX_VALUE;

		@SuppressWarnings("unchecked")
		Dijkstra(int n) {
			this.n = n;

			edges = new List[n];
			for (int ii = 0; ii < n; ii++) {
				edges[ii] = new ArrayList<Edge>();
			}

			s = - 1;
		}

		// i, j:0-indexed
		public void addEdge(int i, int j, int c) {
			edges[i].add(new Edge(i, j, c));
		}

		public long getShortestPath(int i, int j) {
			if (s != i) {
				d = new long[n];
				Arrays.fill(d, INF);
				d[i] = 0;
				pq = new PriorityQueue<Vertex>();
				pq.add(new Vertex(i, d[i]));

				while (!pq.isEmpty()) {
					Vertex u = pq.poll();
					if (d[u.me] < u.d) {
						continue;  // skip old vertex
					}

					for (int ii = 0; ii < edges[u.me].size(); ii++) {
						Edge v = edges[u.me].get(ii);
						if (d[u.me] != INF && d[v.v] > Math.max(d[u.me], v.w)) {
							d[v.v] = Math.max(d[u.me], v.w);
							pq.add(new Vertex(v.v, d[v.v]));
						}
					}
				}

				s = i;
			}

			return d[j];
		}

		private static class Edge {
//			int u; // from
			int v; // to
			int w; // cost

			Edge(int u, int v, int w) {
//				this.u = u;
				this.v = v;
				this.w = w;
			}
		}

		private static class Vertex implements Comparable<Vertex> {
			int me; // me
			long d; // cost

			Vertex(int u, long w) {
				this.me = u;
				this.d = w;
			}

			@Override
			public int compareTo(Vertex o) {
//				return Long.compare(this.d, o.d);
				return this.d > o.d ? 1 : this.d < o.d ? -1 : 0;
			}
		}
	}

	@SuppressWarnings("unused")
	private static class Scanner {
		BufferedReader br;

		Scanner (InputStream in) {
			br = new BufferedReader(new InputStreamReader(in));
		}

		private boolean isPrintable(int ch) {
			return ch >= '!' && ch <= '~';
		}

		private boolean isCRLF(int ch) {
			return ch == '\n' || ch == '\r';
		}

		private int nextPrintable() {
			try {
				int ch;
				while ((ch = br.read()) != -1 && !isPrintable(ch)) {
				}
				return ch;
			} catch (IOException e) {
				throw new IllegalStateException();
			}
		}

		String next() {
			try {
				int ch = nextPrintable();
				StringBuilder sb = new StringBuilder();
				while (isPrintable(ch)) {
					sb.appendCodePoint(ch);
					ch = br.read();
				}

				return sb.toString();
			} catch (IOException e) {
				throw new IllegalStateException();
			}
		}

		int nextInt() throws RuntimeException {
			try {
				// parseInt
				boolean negative = false;
				int res = 0;

				int fc = nextPrintable();
				if (fc < '0') {
					if (fc == '-') {
						negative = true;
					} else if (fc != '+') {
						throw new NumberFormatException();
					}
					fc = br.read();
				}

				int ch = fc;
				do {
					if (ch < '0' || ch > '9') {
						throw new NumberFormatException();
					}

					res *= 10;
					res += ch - '0';
				} while ((ch = br.read()) != -1 && isPrintable(ch));

				// parseInt

				return negative ? -res : res;
			} catch (IOException e) {
				throw new IllegalStateException();
			}
		}

		long nextLong() throws RuntimeException {
			return Long.parseLong(next());
		}

		float nextFloat() throws RuntimeException {
			return Float.parseFloat(next());
		}

		double nextDouble() throws RuntimeException {
			return Double.parseDouble(next());
		}

		String nextLine() throws RuntimeException {
			try {
				int ch;
				while ((ch = br.read()) != -1 && !isCRLF(ch)) {
				}
				StringBuilder sb = new StringBuilder();
				while (!isCRLF(ch)) {
					sb.appendCodePoint(ch);
					ch = br.read();
				}

				return sb.toString();
			} catch (IOException e) {
				throw new IllegalStateException();
			}
		}

		void close() {
			try {
				br.close();
			} catch (IOException e) {
//				throw new IllegalStateException();
			}
		}
	}

	private static class Printer extends PrintWriter {
		Printer(PrintStream out) {
			super(out);
		}
	}
}
0