結果

問題 No.408 五輪ピック
コンテスト
ユーザー Kilisame
提出日時 2016-09-09 13:14:01
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 3,810 bytes
コンパイル時間 4,525 ms
コンパイル使用メモリ 79,876 KB
実行使用メモリ 66,564 KB
最終ジャッジ日時 2024-11-16 05:21:08
合計ジャッジ時間 17,592 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 14 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

package puzzle.yukicoder.golympic;

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

public class Golympic {

    /* 頂点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);
        }
        List<Integer>[] 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);
        }
        List<Integer>[] twoStepHistoriesArray = new List[n];

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