From a62cecdf2d710e7321d98ec3931f9eb6c167b6fd Mon Sep 17 00:00:00 2001 From: Ukj0ng <90972240+Ukj0ng@users.noreply.github.com> Date: Fri, 10 Oct 2025 09:23:00 +0900 Subject: [PATCH] =?UTF-8?q?[20251010]=20BOJ=20/=20G3=20/=20=EC=A0=84?= =?UTF-8?q?=EA=B5=AD=EC=8B=9C=EB=8C=80=20/=20=ED=95=9C=EC=A2=85=EC=9A=B1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ...04\352\265\255\354\213\234\353\214\200.md" | 99 +++++++++++++++++++ 1 file changed, 99 insertions(+) create mode 100644 "Ukj0ng/202510/10 BOJ G3 \354\240\204\352\265\255\354\213\234\353\214\200.md" 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]); + } +} +```