Submission #11620624

---

Source Code Expand

using AtCoderTypicalContest001.Questions;
using AtCoderTypicalContest001.Extensions;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;

namespace AtCoderTypicalContest001.Questions
{
    public class QuestionB : AtCoderQuestionBase
    {
        int[] _parents = null;

        public override IEnumerable<object> Solve(TextReader inputStream)
        {
            var nq = inputStream.ReadIntArray();
            var n = nq[0];
            var q = nq[1];
            _parents = InitializeParents(n);

            for (int i = 0; i < q; i++)
            {
                var pab = inputStream.ReadIntArray();
                var p = pab[0];
                var a = pab[1];
                var b = pab[2];

                if (p == 0)
                {
                    Unite(a, b);
                }
                else
                {
                    yield return IsInSameGroup(a, b) ? "Yes" : "No";
                }
            }
        }

        int[] InitializeParents(int n)
        {
            var r = new int[n];
            for (int i = 0; i < r.Length; i++)
            {
                r[i] = i;
            }
            return r;
        }

        void Unite(int x, int y)
        {
            var parentX = _parents[x];
            var parentY = _parents[y];

            if (parentX == parentY)
            {
                return;
            }
            _parents[x] = parentY;
        }

        int FindRoot(int x)
        {
            if (_parents[x] == x)
            {
                return x;
            }
            else
            {
                _parents[x] = FindRoot(_parents[x]);
                return _parents[x];
            }
        }

        bool IsInSameGroup(int x, int y) => FindRoot(x) == FindRoot(y);
    }
}


namespace AtCoderTypicalContest001
{
    class Program
    {
        static void Main(string[] args)
        {
            IAtCoderQuestion question = new QuestionB();    // 問題に合わせて書き換え
            var answers = question.Solve(Console.In);
            foreach (var answer in answers)
            {
                Console.WriteLine(answer);
            }
        }
    }
}

#region Base Classes

namespace AtCoderTypicalContest001.Questions
{

    public interface IAtCoderQuestion
    {
        IEnumerable<object> Solve(string input);
        IEnumerable<object> Solve(TextReader inputStream);
    }

    public abstract class AtCoderQuestionBase : IAtCoderQuestion
    {
        public IEnumerable<object> Solve(string input)
        {
            var stream = new MemoryStream(Encoding.Unicode.GetBytes(input));
            var reader = new StreamReader(stream, Encoding.Unicode);

            return Solve(reader);
        }

        public abstract IEnumerable<object> Solve(TextReader inputStream);
    }

}

#endregion

#region Extensions

namespace AtCoderTypicalContest001.Extensions
{
    internal static class TextReaderExtensions
    {
        internal static int ReadInt(this TextReader reader) => int.Parse(ReadString(reader));
        internal static long ReadLong(this TextReader reader) => long.Parse(ReadString(reader));
        internal static double ReadDouble(this TextReader reader) => double.Parse(ReadString(reader));
        internal static string ReadString(this TextReader reader) => reader.ReadLine();

        internal static int[] ReadIntArray(this TextReader reader, char separator = ' ') => ReadStringArray(reader, separator).Select(int.Parse).ToArray();
        internal static long[] ReadLongArray(this TextReader reader, char separator = ' ') => ReadStringArray(reader, separator).Select(long.Parse).ToArray();
        internal static double[] ReadDoubleArray(this TextReader reader, char separator = ' ') => ReadStringArray(reader, separator).Select(double.Parse).ToArray();
        internal static string[] ReadStringArray(this TextReader reader, char separator = ' ') => reader.ReadLine().Split(separator);
    }
}

#endregion

Submission Info

Judge Result