結果

問題 No.408 五輪ピック
ユーザー Kilisame
提出日時 2016-09-09 11:08:33
言語 Java
(openjdk 23)
結果
AC  
実行時間 620 ms / 5,000 ms
コード長 3,922 bytes
コンパイル時間 2,411 ms
コンパイル使用メモリ 79,624 KB
実行使用メモリ 69,272 KB
最終ジャッジ日時 2024-11-15 23:00:40
合計ジャッジ時間 15,345 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

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 Main {

    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<List<Integer>>> twoStepHistoriesMap = new HashMap<Integer, List<List<Integer>>>();

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