結果

問題 No.416 旅行会社
ユーザー KilisameKilisame
提出日時 2016-09-12 14:37:24
言語 Java21
(openjdk 21)
結果
TLE  
実行時間 -
コード長 7,158 bytes
コンパイル時間 4,145 ms
コンパイル使用メモリ 78,124 KB
実行使用メモリ 106,760 KB
最終ジャッジ日時 2024-11-17 03:57:37
合計ジャッジ時間 71,683 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 AC 52 ms
41,808 KB
testcase_02 AC 52 ms
89,396 KB
testcase_03 AC 50 ms
41,820 KB
testcase_04 AC 52 ms
89,524 KB
testcase_05 AC 53 ms
41,868 KB
testcase_06 AC 55 ms
86,136 KB
testcase_07 AC 97 ms
43,028 KB
testcase_08 AC 261 ms
97,256 KB
testcase_09 AC 634 ms
53,308 KB
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

package puzzle.yukicoder.travelagency;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.NoSuchElementException;

public class TravelAgency {

    /* 橋が壊れていく過程を考えるのではなく、壊れきった状態をスタート状態とし、そこから巻き戻して1つずつ橋が修復されていったときに、到達できる島がどのように増えていくかを考える。 */
    /* 橋の情報は島A、島Bともに10の5乗数までなので、AとBの島のIDを0-99999としたとき、A*10の5乗+BというLong値で管理する */
    public static void main(String[] args) {
        FastScanner cin = new FastScanner(System.in);
        int n = cin.nextInt();
        int m = cin.nextInt();
        int q = cin.nextInt();
        List<Long> unreachedBridgeList = new ArrayList<Long>(m);
        for (int i = 0; i < m; i++) {
            long bridge = (cin.nextLong() - 1L) * 100000L + (cin.nextLong() - 1L);
            unreachedBridgeList.add(bridge);
        }
        List<Long> brokenBridgeList = new ArrayList<Long>(q);
        for (int i = 0; i < q; i++) {
            long brokenBridge = (cin.nextLong() - 1L) * 100000L + (cin.nextLong() - 1L);
            brokenBridgeList.add(brokenBridge);
            /* 壊れる橋はいったん橋リストから削除する */
            unreachedBridgeList.remove(brokenBridge);
        }
        /* 島のステータス管理。島のIDに対し、何回目に壊れた橋を復旧したらその島に初めて到達できるようになるかを求める */
        int[] reached = new int[n];
        /* 島1を到達したIDに指定し、eventIndexを-1として実行 */
        reachedIsland(0, -1, unreachedBridgeList, reached);
        /* 最後に壊れた橋から順に復旧をはじめる */
        for (int i = brokenBridgeList.size(); i > 0; i--) {
            long recoveredBridge = brokenBridgeList.get(i - 1);
            /* その橋が復旧したことで新たに行ける島が増える=片方の島はすでに到達できており、もう片方の島が未到達だった */
            int islandA = (int) (recoveredBridge / 100000L);
            int islandB = (int) (recoveredBridge % 100000L);
            if (reached[islandA] != 0 && reached[islandB] == 0) {
                reachedIsland(islandB, i, unreachedBridgeList, reached);
            } else if (reached[islandB] != 0 && reached[islandA] == 0) {
                reachedIsland(islandA, i, unreachedBridgeList, reached);
            } else if (reached[islandA] == 0 && reached[islandB] == 0) {
                /* まだ未踏破同士の島をつなぐ橋だった場合、未踏破橋リストに加える */
                unreachedBridgeList.add(recoveredBridge);
            }
            /* それ以外のケース=すでに到達済み同士をつなぐ橋なので、無視 */
        }
        exit(reached);
    }

    private static void reachedIsland(int islandId, int eventIndex, List<Long> unreachedBridgeList, int[] reached) {
        /* 到達した島に、何回目に壊れた橋を復旧して辿り着いたか記録 */
        reached[islandId] = eventIndex;
        /* 橋リストから、その島とつながっていてものをピックアップし、つながった先の島IDを確保しつつ橋リストから削除 */
        List<Integer> reachedIslandList = new ArrayList<Integer>();
        Iterator<Long> bridgeIterator = unreachedBridgeList.iterator();
        while (bridgeIterator.hasNext()) {
            long bridge = bridgeIterator.next();
            int islandA = (int) (bridge / 100000L);
            int islandB = (int) (bridge % 100000L);
            if (islandA == islandId && reached[islandB] == 0 && !reachedIslandList.contains(islandB)) {
                reachedIslandList.add(islandB);
                bridgeIterator.remove();
            } else if (islandB == islandId && reached[islandA] == 0 && !reachedIslandList.contains(islandA)) {
                reachedIslandList.add(islandA);
                bridgeIterator.remove();
            }
        }
        for (int reachedIslandId : reachedIslandList) {
            reachedIsland(reachedIslandId, eventIndex, unreachedBridgeList, reached);
        }
    }

    private static void exit(int[] reached) {
        for (int i = 1; i < reached.length; i++) {
            System.out.println(reached[i]);
        }
    }

    static class FastScanner {
        private final InputStream in;
        private final byte[] buffer = new byte[1024];
        private int ptr = 0;
        private int buflen = 0;

        public FastScanner(InputStream in) {
            this.in = in;
        }

        private boolean hasNextByte() {
            if (ptr < buflen) {
                return true;
            } else {
                ptr = 0;
                try {
                    buflen = in.read(buffer);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                if (buflen <= 0) {
                    return false;
                }
            }
            return true;
        }

        private int readByte() {
            if (hasNextByte())
                return buffer[ptr++];
            else
                return -1;
        }

        private static boolean isPrintableChar(int c) {
            return 33 <= c && c <= 126;
        }

        public boolean hasNext() {
            while (hasNextByte() && !isPrintableChar(buffer[ptr]))
                ptr++;
            return hasNextByte();
        }

        public String next() {
            if (!hasNext())
                throw new NoSuchElementException();
            StringBuilder sb = new StringBuilder();
            int b = readByte();
            while (isPrintableChar(b)) {
                sb.appendCodePoint(b);
                b = readByte();
            }
            return sb.toString();
        }

        public long nextLong() {
            if (!hasNext())
                throw new NoSuchElementException();
            long n = 0;
            boolean minus = false;
            int b = readByte();
            if (b == '-') {
                minus = true;
                b = readByte();
            }
            if (b < '0' || '9' < b) {
                throw new NumberFormatException();
            }
            while (true) {
                if ('0' <= b && b <= '9') {
                    n *= 10;
                    n += b - '0';
                } else if (b == -1 || !isPrintableChar(b)) {
                    return minus ? -n : n;
                } else {
                    throw new NumberFormatException();
                }
                b = readByte();
            }
        }

        public int nextInt() {
            long nl = nextLong();
            if (nl < Integer.MIN_VALUE || nl > Integer.MAX_VALUE)
                throw new NumberFormatException();
            return (int) nl;
        }

        public double nextDouble() {
            return Double.parseDouble(next());
        }
    }
}
0