結果

問題 No.1805 Approaching Many Typhoon
ユーザー kusagame12
提出日時 2024-06-12 16:10:45
言語 Java
(openjdk 23)
結果
AC  
実行時間 228 ms / 2,000 ms
コード長 5,712 bytes
コンパイル時間 4,377 ms
コンパイル使用メモリ 78,044 KB
実行使用メモリ 57,884 KB
最終ジャッジ日時 2024-06-12 16:10:58
合計ジャッジ時間 9,936 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

import java.util.*;
import java.lang.*;
import java.io.*;
class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        int s = sc.nextInt()-1;
        int g = sc.nextInt()-1;
        
        List<Edge> edges = new ArrayList<>();
        for (int i=0; i<m; i++) {
            int f = sc.nextInt()-1;
            int t = sc.nextInt()-1;
            Edge edge = new Edge(f,t);
            edges.add(edge);
        }
        int u = sc.nextInt();
        int[] is = new int[u];
        for (int k=0; k<u; k++) {
            int i = sc.nextInt()-1;
            is[k] = i;
        }
        
        List<Edge> canConnectEdges = new ArrayList<>();
        for (Edge edge : edges) {
            boolean connect = true;
            for (int j=0; j<is.length; j++) {  
                int i = is[j];
                if(edge.from == i || edge.to == i){
                    connect = false;
                    break;
                }
            }
            if(connect){
                canConnectEdges.add(edge);
            }
        }
        
        UnionFindTree uft = new UnionFindTree(n);
        for (Edge edge : canConnectEdges) {
            int from = edge.from;
            int to = edge.to;
            uft.union(from,to);
        }
        System.out.println(uft.same(s,g) ? "Yes" : "No");
    }   
}
//Union find クラス
class UnionFindTree {
    int[] parent; // インデックスとノードを対応させ、そのルートノードのインデックスを格納
    int[] rank; // parentと同様に、木の高さを格納
    int[] size; // 木を構成する、要素の数を格納
    int treeCount; // 木の数
    /**
     * コンストラクタ
     * 
     * @param n : 頂点の数
     */
    public UnionFindTree(int n) {
        this.parent = new int[n];
        this.rank = new int[n];
        this.size = new int[n];
        this.treeCount = n;
        // 根を作る。
        for (int i = 0; i < n; i++) {
            makeSet(i);
        }
    }
    /**
     * 根をつくる。最初は、すべて異なる木に属しているものとする。
     * 
     * @param i : 頂点の数
     */
    private void makeSet(int i) {
        parent[i] = i;
        rank[i] = 0; // 集合の高さ
        size[i] = 1; // 木を構成する要素の数
    }
    /**
     * 要素xが属する木と要素yが属する木を連結する。 木の高さ（ランク）を気にして、低い方に高い方をつなげる。（高い方の木を全体の木とする。）
     *
     * @param x : １つ目の要素(yとは異なる木に属している必要がある。)
     * @param y : ２つ目の要素(xとは異なる木に属している必要がある。)
     */
    public void union(int x, int y) {
        int xRoot = find(x);
        int yRoot = find(y);
        int xSize = getSize(xRoot);
        int ySize = getSize(yRoot);
        // xが属する木の方が大きい場合
        if (rank[xRoot] > rank[yRoot]) {
            parent[yRoot] = xRoot; // yの親をxに更新
            size[xRoot] = xSize + ySize; // 木を構成する要素の数を更新
            treeCount--;
        } else if (rank[xRoot] < rank[yRoot]) {
            parent[xRoot] = yRoot;
            size[yRoot] = ySize + xSize;
            treeCount--;
        } else if (xRoot != yRoot) {
            parent[yRoot] = xRoot;
            rank[xRoot]++; // 同じ高さの木がルートの子として着くから大きさ++;
            size[xRoot] = xSize + ySize;
            treeCount--;
        }
    }
    /**
     * 要素の根を返す。 経路圧縮も同時に行う。（1→3→2となっていて2をfindした際、1→3,2と木の深さを浅くする。）
     *
     * @param i
     * @return 要素iの根 （1→3→2となっていて2をfindした際、1となる。)
     */
    public int find(int i) {
        if (i != parent[i]) {
            parent[i] = find(parent[i]);
        }
        return parent[i];
    }
    /**
     * ２つの要素が同じ木に属するかどうかを返す。
     *
     * @param x : １つ目の要素
     * @param y : ２つ目の要素
     * @return 同じ木に属していればtrue
     */
    public boolean same(int x, int y) {
        return find(x) == find(y);
    }
    /**
     * 木の高さを返す。
     * 
     * @param i : 高さを調べたい木の根の番号
     * @return 木の高さ
     */
    public int getRank(int i) {
        return rank[i];
    }
    /**
     * 木を構成する要素の数を返す。
     * 
     * @param i : 要素数を調べたい木の根の番号
     * @return 木を構成する要素の数
     */
    public int getSize(int i) {
        return size[find(i)];
    }
    /**
     * 木の数を返す。
     * 
     * @return 木の数。
     */
    public int getTreeCount() {
        return treeCount;
    }
}
//グラフのエッジ(辺)を格納するクラス
class Edge
{
    public final int from, to;
 
    /**コンストラクタ
    * @param : from 頂点 
    * @param : to つながっている頂点
    */
   Edge(int from, int to)
    {
        this.from = from;
        this.to = to;
    }
 
}
//無向グラフクラス
class Graph
{
    //隣接リストを表すリストのリスト(ある頂点が、どの頂点とつながっているかを保持するリスト)
    List<List<Edge>> list = null;
 
    /**コンストラクタ
    * @param edges  : 辺
    * @param n  : 頂点の個数
    *
    */
    Graph(List<Edge> edges, int n)
    {
        list = new ArrayList<>();
 
        for (int i = 0; i < n; i++) {
            list.add(new ArrayList<>());
        }
 
        //無向グラフにエッジを追加します
        for (Edge edge: edges)
        {
            int from = edge.from;
            int to = edge.to;
 
            list.get(from).add(edge);
            list.get(to).add(edge);
        }
        
    }
    
    public int size() {
        return list.size();
    }
    public List<Edge> getEdgsList(int vNum) {
        return list.get(vNum);
    }
    
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0