結果

問題 No.408 五輪ピック
コンテスト
ユーザー Kilisame
提出日時 2016-09-09 12:39:24
言語 Java
(openjdk 23)
結果
AC  
実行時間 596 ms / 5,000 ms
コード長 3,839 bytes
コンパイル時間 3,629 ms
コンパイル使用メモリ 79,144 KB
実行使用メモリ 67,008 KB
最終ジャッジ日時 2024-11-16 05:18:54
合計ジャッジ時間 17,606 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

package puzzle.yukicoder.golympic;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;

public class Golympic {

    private static List<Integer>[] sideListArray;

    /* 頂点1から長さ5の単純閉路がある=頂点1から同一の頂点に対し、2Stepで行くルートと3Stepで行くルートが存在し、その途中にある頂点が重複していない */
    public static void main(String[] args) {
        Scanner cin = new Scanner(System.in);
        String line = cin.nextLine();
        String[] splitedLine = line.split(" ");
        int n = Integer.parseInt(splitedLine[0]);
        if (n < 5) {
            /* 最低5頂点なければ長さ5の単純閉路は作れない */
            exit(false);
        }
        int m = Integer.parseInt(splitedLine[1]);
        if (m < 5) {
            /* 最低5辺なければ長さ5の単純閉路は作れない */
            exit(false);
        }
        sideListArray = new List[n];
        for (int i = 0; i < m; i++) {
            line = cin.nextLine();
            splitedLine = line.split(" ");
            int a = Integer.parseInt(splitedLine[0]) - 1;
            int b = Integer.parseInt(splitedLine[1]) - 1;
            if (sideListArray[a] == null) {
                sideListArray[a] = new ArrayList<Integer>();
            }
            sideListArray[a].add(b);
            if (sideListArray[b] == null) {
                sideListArray[b] = new ArrayList<Integer>();
            }
            sideListArray[b].add(a);
        }
        cin.close();
        if (sideListArray[0].size() < 2) {
            /* 頂点1から異なる2頂点への辺が存在しない場合、頂点1を含む単純閉路は作れない */
            exit(false);
        }
        Map<Integer, List<Integer>> twoStepHistoriesMap = new HashMap<Integer, List<Integer>>();

        for (int i : sideListArray[0]) {
            for (int j : sideListArray[i]) {
                if (j == 0) {
                    /* 頂点1に戻るルートだけは除外 */
                    continue;
                }
                List<Integer> histories = twoStepHistoriesMap.getOrDefault(j, new ArrayList<Integer>());
                histories.add(i);
                twoStepHistoriesMap.put(j, histories);
            }
        }
        Set<Integer> set = new HashSet<Integer>(4);
        for (Map.Entry<Integer, List<Integer>> twoStepHistories : twoStepHistoriesMap.entrySet()) {
            /* 3STEP目で行ける頂点のうち、2TEP目でも行けたものを探す */
            for (int i : sideListArray[twoStepHistories.getKey()]) {
                if (twoStepHistoriesMap.get(i) == null) {
                    continue;
                }
                for (int twoStepHistory : twoStepHistories.getValue()) {
                    for (int otherTwoStepHistory : twoStepHistoriesMap.get(i)) {
                        /* 重複した頂点を通らないかはSetに全て格納、1以外の4頂点がaddできていればOK */
                        set.clear();
                        set.add(twoStepHistories.getKey());
                        set.add(i);
                        set.add(twoStepHistory);
                        set.add(otherTwoStepHistory);
                        if (set.size() == 4) {
                            exit(true);
                        }
                    }
                }
            }
        }
        /* 全探索を終えてもみつからなければ、無し。 */
        exit(false);
    }

    private static void exit(boolean result) {
        if (result) {
            System.out.println("YES");
        } else {
            System.out.println("NO");
        }
        System.exit(0);
    }
}
0