Submission #420586

---

Source Code Expand

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

namespace ATC001
{
    public class B
    {
        private void Run()
        {
            var input = Console.ReadLine().Trim().Split();
            int N = int.Parse(input[0]);
            int Q = int.Parse(input[1]);

            var uf = new UnionFind(N);
            for (int i = 0; i < Q; i++)
            {
                input = Console.ReadLine().Trim().Split();
                int P = int.Parse(input[0]);
                int A = int.Parse(input[1]);
                int B = int.Parse(input[2]);
                if (P == 0) { uf.Union(A, B); }
                else if (P == 1)
                {
                    Console.WriteLine(uf.IsSameSet(A, B) ? "Yes" : "No");
                }
                else { throw new Exception(); }
            }
        }

        private static void Main()
        {
            var old = Console.Out;
            using (var writer = new System.IO.StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false })
            {
                Console.SetOut(writer);
                new B().Run();
                Console.Out.Flush();
                Console.SetOut(old);
            }
        }

        public class UnionFind
        {
            public UnionFind(int elemCount) { this.m_a = Enumerable.Repeat(-1, elemCount).ToArray(); }
            public int Root(int x)
            {
                return this.m_a[x] < 0
                    ? x
                    : this.m_a[x] = this.Root(this.m_a[x]);
            }
            public bool Union(int x, int y)
            {
                x = Root(x);
                y = Root(y);
                if (x == y) { return false; }

                if (this.m_a[x] > this.m_a[y])
                {
                    var temp = this.m_a[x];
                    this.m_a[x] = this.m_a[y];
                    this.m_a[y] = temp;
                }
                this.m_a[x] += this.m_a[y];
                this.m_a[y] = x;
                return true;
            }
            public bool IsSameSet(int x, int y) { return this.Root(x) == this.Root(y); }
            public int Size(int x) { return -this.m_a[this.Root(x)]; }

            private readonly int[] m_a;
        }
    }
}

Submission Info

Judge Result