結果
問題 |
No.408 五輪ピック
|
ユーザー |
|
提出日時 | 2016-09-09 11:04:00 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 684 ms / 5,000 ms |
コード長 | 4,186 bytes |
コンパイル時間 | 2,293 ms |
コンパイル使用メモリ | 79,284 KB |
実行使用メモリ | 66,980 KB |
最終ジャッジ日時 | 2024-11-15 22:59:54 |
合計ジャッジ時間 | 16,881 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 32 |
ソースコード
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]) { /* 頂点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 exit(boolean result) { if (result) { System.out.println("YES"); } else { System.out.println("NO"); } System.exit(0); } }