diff --git "a/Ukj0ng/202510/10 BOJ G3 \354\240\204\352\265\255\354\213\234\353\214\200.md" "b/Ukj0ng/202510/10 BOJ G3 \354\240\204\352\265\255\354\213\234\353\214\200.md" new file mode 100644 index 00000000..06a2014c --- /dev/null +++ "b/Ukj0ng/202510/10 BOJ G3 \354\240\204\352\265\255\354\213\234\353\214\200.md" @@ -0,0 +1,99 @@ +``` +import java.io.*; +import java.util.*; + +public class Main { + private static BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); + private static BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out)); + private static Map result; + private static int[] power, uf, size; + private static int N, M; + public static void main(String[] args) throws IOException { + init(); + + for (int i = 1; i <= N; i++) { + int root = find(i); + + if (!result.containsKey(root) && power[root] > 0) { + result.put(root, power[root]); + } + } + + bw.write(result.size() + "\n"); + + List value = new ArrayList<>(result.values()); + Collections.sort(value); + + for (int element : value) { + bw.write(element + " "); + } + bw.flush(); + bw.close(); + br.close(); + } + + private static void init() throws IOException { + StringTokenizer st = new StringTokenizer(br.readLine()); + N = Integer.parseInt(st.nextToken()); + M = Integer.parseInt(st.nextToken()); + + power = new int[N+1]; + uf = new int[N+1]; + size = new int[N+1]; + + result = new HashMap<>(); + + for (int i = 1; i <= N; i++) { + power[i] = Integer.parseInt(br.readLine()); + uf[i] = i; + size[i] = 1; + } + + for (int i = 0; i < M; i++) { + st = new StringTokenizer(br.readLine()); + int O = Integer.parseInt(st.nextToken()); + int P = Integer.parseInt(st.nextToken()); + int Q = Integer.parseInt(st.nextToken()); + + union(P, Q, O); + } + } + + private static void union(int x, int y, int o) { + int X = find(x); + int Y = find(y); + + if (o == 1) { + if (size[X] < size[Y]) { + uf[X] = Y; + size[Y] += size[X]; + power[Y] += power[X]; + + } else { + uf[Y] = X; + size[X] += size[Y]; + power[X] += power[Y]; + } + } else { + if (power[X] == power[Y]) { + power[X] = 0; + power[Y] = 0; + } else if (power[X] > power[Y]) { + uf[Y] = X; + size[X] += size[Y]; + power[X] -= power[Y]; + } else { + uf[X] = Y; + size[Y] += size[X]; + power[Y] -= power[X]; + } + } + } + + private static int find(int x) { + if (uf[x] == x) return x; + + return uf[x] = find(uf[x]); + } +} +```