結果

問題 No.408 五輪ピック
コンテスト
ユーザー Kilisame
提出日時 2016-09-09 11:01:18
言語 Java
(openjdk 23)
結果
AC  
実行時間 928 ms / 5,000 ms
コード長 4,702 bytes
コンパイル時間 2,325 ms
コンパイル使用メモリ 79,860 KB
実行使用メモリ 71,936 KB
最終ジャッジ日時 2024-11-15 22:59:21
合計ジャッジ時間 16,900 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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();
        for (int i = 0; i < sideListArray.length; i++) {
            removeSingleSide(i);
        }
        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]) {
            /* 頂点1からたどれる点から、さらにたどれる頂点のうち、頂点1以外にいくものの経路を全て列挙する */
            List<Integer> oneStepHistory = new ArrayList<Integer>();
            /* 頂点1から1STEP目の頂点はi */
            oneStepHistory.add(i);
            for (int j : sideListArray[i]) {
                if (j == 0) {
                    /* 頂点1に戻るルートだけは除外 */
                    continue;
                }
                List<Integer> twoStepHistory = new ArrayList<Integer>(oneStepHistory);
                twoStepHistory.add(j);
                List<List<Integer>> histories = twoStepHistoriesMap.getOrDefault(j, new ArrayList<List<Integer>>());
                histories.add(twoStepHistory);
                twoStepHistoriesMap.put(j, histories);
            }
        }
        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<Integer> set = new HashSet<Integer>();
                        set.addAll(twoStepHistory);
                        set.addAll(otherTwoStepHistory);
                        if (set.size() == 4) {
                            exit(true);
                        }
                    }
                }
            }
        }
        /* 全探索を終えてもみつからなければ、無し。 */
        exit(false);
    }

    /* 行き止まりの頂点をたどっても無駄なので先に削除しておく */
    private static void removeSingleSide(int index) {
        if (sideListArray[index].size() == 1) {
            int nextIndex = sideListArray[index].get(0);
            sideListArray[index].clear();
            sideListArray[nextIndex].remove(new Integer(index));
            removeSingleSide(nextIndex);
        }
    }

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