結果

問題 No.2290 UnUnion Find
ユーザー kusagame12kusagame12
提出日時 2024-06-11 13:47:31
言語 Java21
(openjdk 21)
結果
AC  
実行時間 1,989 ms / 2,000 ms
コード長 4,253 bytes
コンパイル時間 2,304 ms
コンパイル使用メモリ 79,128 KB
実行使用メモリ 77,092 KB
最終ジャッジ日時 2024-06-11 13:49:12
合計ジャッジ時間 85,818 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 103 ms
40,244 KB
testcase_01 AC 112 ms
40,984 KB
testcase_02 AC 1,467 ms
58,692 KB
testcase_03 AC 1,684 ms
65,912 KB
testcase_04 AC 1,615 ms
65,588 KB
testcase_05 AC 1,667 ms
65,652 KB
testcase_06 AC 1,543 ms
65,800 KB
testcase_07 AC 1,657 ms
65,272 KB
testcase_08 AC 1,738 ms
65,840 KB
testcase_09 AC 1,722 ms
65,592 KB
testcase_10 AC 1,630 ms
65,820 KB
testcase_11 AC 1,707 ms
65,312 KB
testcase_12 AC 1,784 ms
65,920 KB
testcase_13 AC 1,661 ms
65,084 KB
testcase_14 AC 1,713 ms
65,400 KB
testcase_15 AC 1,723 ms
66,096 KB
testcase_16 AC 1,649 ms
65,456 KB
testcase_17 AC 1,623 ms
65,796 KB
testcase_18 AC 1,651 ms
65,436 KB
testcase_19 AC 1,725 ms
66,440 KB
testcase_20 AC 1,858 ms
76,300 KB
testcase_21 AC 1,698 ms
60,060 KB
testcase_22 AC 1,605 ms
62,372 KB
testcase_23 AC 1,742 ms
62,408 KB
testcase_24 AC 1,871 ms
72,868 KB
testcase_25 AC 1,707 ms
61,208 KB
testcase_26 AC 1,911 ms
75,528 KB
testcase_27 AC 1,857 ms
74,028 KB
testcase_28 AC 1,773 ms
64,256 KB
testcase_29 AC 1,904 ms
71,684 KB
testcase_30 AC 1,806 ms
59,696 KB
testcase_31 AC 1,762 ms
67,304 KB
testcase_32 AC 1,695 ms
62,248 KB
testcase_33 AC 1,660 ms
64,612 KB
testcase_34 AC 1,743 ms
77,092 KB
testcase_35 AC 1,758 ms
73,956 KB
testcase_36 AC 1,707 ms
62,360 KB
testcase_37 AC 1,642 ms
63,800 KB
testcase_38 AC 1,665 ms
61,704 KB
testcase_39 AC 1,638 ms
62,952 KB
testcase_40 AC 1,852 ms
74,224 KB
testcase_41 AC 1,686 ms
60,856 KB
testcase_42 AC 1,736 ms
66,568 KB
testcase_43 AC 1,754 ms
66,348 KB
testcase_44 AC 1,646 ms
65,888 KB
testcase_45 AC 1,989 ms
65,504 KB
testcase_46 AC 1,750 ms
65,508 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;
import java.lang.*;
import java.io.*;

// The main method must be in a class named "Main".
class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();
        int q = sc.nextInt();

        UnionFindTree uft = new UnionFindTree(n);

        for (int j=0; j<q; j++) {
            if(sc.nextInt() == 1){
                int u = sc.nextInt()-1;
                int v = sc.nextInt()-1;
                uft.union(u , v);
            }else{
                int v = sc.nextInt()-1; 
                int unUnionPoint = uft.unUnion(v);
                System.out.println(unUnionPoint == -1 ? -1 : unUnionPoint + 1);         
            }
        }
        
    }
}


//Union find クラス
class UnionFindTree {

	int[] parent; // インデックスとノードを対応させ、そのルートノードのインデックスを格納
	int[] rank; // parentと同様に、木の高さを格納
	int[] size; // 木を構成する、要素の数を格納
    TreeSet<Integer> set = new TreeSet<>();
	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; // 木を構成する要素の数
        set.add(i);
	}

	/**
	 * 要素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--;
            set.remove(yRoot);

		} else if (rank[xRoot] < rank[yRoot]) {

			parent[xRoot] = yRoot;
			size[yRoot] = ySize + xSize;
			treeCount--;
            set.remove(xRoot);

		} else if (xRoot != yRoot) {

			parent[yRoot] = xRoot;
			rank[xRoot]++; // 同じ高さの木がルートの子として着くから大きさ++;
			size[xRoot] = xSize + ySize;
			treeCount--;
            set.remove(yRoot);

		}

	}

	/**
	 * 要素の根を返す。 経路圧縮も同時に行う。（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;
	}

    public int unUnion(int i){
        if(set.size() == 1){
            return -1;
        }
        int x = find(i);
        if(x == set.first()){
            return set.last();
        }
        return set.first();
    }

}
0