Cod sursa(job #3242279)

Utilizator obsidianMidnight Majesty obsidian Data 10 septembrie 2024 19:52:54
Problema Paduri de multimi disjuncte Scor 40
Compilator java Status done
Runda Arhiva educationala Marime 2.97 kb
//package disjoint;

import java.io.*;
import java.util.StringTokenizer;

public class Main {
    static final String INPUT_FILE = "disjoint.in";
    static final String OUTPUT_FILE = "disjoint.out";

    public static class TokenizedReader {
        private final BufferedReader reader;
        private StringTokenizer tokenizer;

        TokenizedReader(String filePath) throws FileNotFoundException {
            reader = new BufferedReader(new FileReader(filePath));
        }

        private String nextToken() {
            while (tokenizer == null || !tokenizer.hasMoreTokens()) {
                try {
                    tokenizer = new StringTokenizer(reader.readLine());
                } catch (IOException e) {
                    throw new RuntimeException(e);
                }
            }
            return tokenizer.nextToken();
        }

        private int nextInt() {
            return Integer.parseInt(nextToken());
        }

        public void close() throws IOException {
            reader.close();
        }
    }

    public static void main(String[] args) throws IOException {
        TokenizedReader reader = new TokenizedReader(INPUT_FILE);
        PrintWriter writer = new PrintWriter(OUTPUT_FILE);
        solve(reader, writer);
        reader.close();
        writer.flush();
        writer.close();
    }

    public static void solve(TokenizedReader reader,
                             PrintWriter writer) {
        int n = reader.nextInt();
        int m = reader.nextInt();
        DisjointSets disjointSets = new DisjointSets(n);
        while (m-- > 0) {
            int op = reader.nextInt();
            int a = reader.nextInt();
            int b = reader.nextInt();
            if (op == 1) {
                disjointSets.union(a, b);
            } else if (op == 2) {
                boolean isDisjoint = disjointSets.query(a, b);
                writer.println(isDisjoint ? "NU" : "DA");
            }
        }
    }

    public static class DisjointSets {
        public int n;
        public int[] parent;

        public DisjointSets(int n) {
            this.n = n;
            this.parent = new int[n + 1];
            for (int i = 1; i <= n; ++i) {
                parent[i] = i;
            }
        }

        public boolean query(int a, int b) {
            int p0 = getParent(a);
            int p1 = getParent(b);
            return p0 != p1;
        }

        private int getParent(int a) {
            if (a == parent[a]) {
                // we reached the root
                return a;
            }
            parent[a] = getParent(parent[a]);
            return parent[a];
        }

        private void union(int a, int b) {
            if (b == parent[b]) {
                // we reached the root
                parent[b] = parent[a];
                return;
            }
            union(a, parent[b]);
            parent[b] = parent[a];
        }
    }
}