using System; using System.Collections.Generic; using System.IO; using System.Linq; using System.Text; using System.Runtime.CompilerServices; using System.Runtime.InteropServices; using YukicoderContest271.Extensions; using YukicoderContest271.Questions; namespace YukicoderContest271.Questions { public class QuestionC : AtCoderQuestionBase { const int MaxColor = 7; public override IEnumerable Solve(TextReader inputStream) { var (cities, roads, queries) = inputStream.ReadValue(); var n = cities * MaxColor; var colors = new bool[cities][]; var uf = new UnionFind(n); for (int i = 0; i < colors.Length; i++) { colors[i] = inputStream.ReadLine().Select(c => c == '1').ToArray(); for (int c = 0; c < MaxColor; c++) { if (colors[i][c] && colors[i][(c + 1) % MaxColor]) { uf.Unite(GetIndex(i, c), GetIndex(i, (c + 1) % MaxColor)); } } } var edgeList = Enumerable.Repeat(0, cities).Select(_ => new List()).ToArray(); for (int i = 0; i < roads; i++) { var (u, v) = inputStream.ReadValue(); u--; v--; edgeList[u].Add(v); edgeList[v].Add(u); for (int c = 0; c < MaxColor; c++) { if (colors[u][c] && colors[v][c]) { uf.Unite(GetIndex(u, c), GetIndex(v, c)); } } } for (int q = 0; q < queries; q++) { var (type, city, color) = inputStream.ReadValue(); city--; color--; if (type == 1) { colors[city][color] = true; for (int c = 0; c < MaxColor; c++) { if (colors[city][c] && colors[city][(c + 1) % MaxColor]) { uf.Unite(GetIndex(city, c), GetIndex(city, (c + 1) % MaxColor)); } } foreach (var next in edgeList[city]) { if (colors[next][color]) { uf.Unite(GetIndex(city, color), GetIndex(next, color)); } } } else { yield return uf.GetGroupSizeOf(GetIndex(city, 0)); } } } int GetIndex(int city, int color) => city * MaxColor + color; public class UnionFind { private int[] _parentsOrSizes; public int Count => _parentsOrSizes.Length; public int Groups { get; private set; } public UnionFind(int count) { _parentsOrSizes = new int[count]; _parentsOrSizes.AsSpan().Fill(-1); Groups = _parentsOrSizes.Length; } [MethodImpl(MethodImplOptions.AggressiveInlining)] public bool Unite(int a, int b) { var x = FindRoot(a); var y = FindRoot(b); if (x == y) { return false; } else { if (-_parentsOrSizes[x] < _parentsOrSizes[y]) { (x, y) = (y, x); } _parentsOrSizes[x] += _parentsOrSizes[y]; _parentsOrSizes[y] = x; Groups--; return true; } } public bool IsInSameGroup(int a, int b) => FindRoot(a) == FindRoot(b); [MethodImpl(MethodImplOptions.AggressiveInlining)] public int FindRoot(int index) { if (unchecked((uint)index >= (uint)_parentsOrSizes.Length)) { ThrowArgumentException(); } return FindRecursive(index); int FindRecursive(int index) { if (_parentsOrSizes[index] < 0) { return index; } else { return _parentsOrSizes[index] = FindRecursive(_parentsOrSizes[index]); } } } public int GetGroupSizeOf(int index) => -_parentsOrSizes[FindRoot(index)]; public int[][] GetAllGroups() { var resultIndices = new int[_parentsOrSizes.Length]; var index = 0; var results = new int[Groups][]; for (int i = 0; i < _parentsOrSizes.Length; i++) { if (_parentsOrSizes[i] < 0) { results[index] = new int[-_parentsOrSizes[i]]; resultIndices[i] = index++; } } var counts = new int[results.Length]; for (int i = 0; i < _parentsOrSizes.Length; i++) { var group = resultIndices[FindRoot(i)]; results[group][counts[group]++] = i; } return results; } private void ThrowArgumentException() => throw new ArgumentException(); } } } namespace YukicoderContest271 { class Program { static void Main(string[] args) { IAtCoderQuestion question = new QuestionC(); var answers = question.Solve(Console.In); var writer = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false }; Console.SetOut(writer); foreach (var answer in answers) { Console.WriteLine(answer); } Console.Out.Flush(); } } } #region Base Class namespace YukicoderContest271.Questions { public interface IAtCoderQuestion { IEnumerable Solve(string input); IEnumerable Solve(TextReader inputStream); } public abstract class AtCoderQuestionBase : IAtCoderQuestion { public IEnumerable Solve(string input) { var stream = new MemoryStream(Encoding.Unicode.GetBytes(input)); var reader = new StreamReader(stream, Encoding.Unicode); return Solve(reader); } public abstract IEnumerable Solve(TextReader inputStream); } } #endregion #region Extensions namespace YukicoderContest271.Extensions { public static class StringExtensions { public static string Join(this IEnumerable source) => string.Concat(source); public static string Join(this IEnumerable source, char separator) => string.Join(separator, source); public static string Join(this IEnumerable source, string separator) => string.Join(separator, source); } public static class TextReaderExtensions { public static int ReadInt(this TextReader reader) => int.Parse(ReadString(reader)); public static long ReadLong(this TextReader reader) => long.Parse(ReadString(reader)); public static double ReadDouble(this TextReader reader) => double.Parse(ReadString(reader)); public static string ReadString(this TextReader reader) => reader.ReadLine(); public static int[] ReadIntArray(this TextReader reader, char separator = ' ') => ReadStringArray(reader, separator).Select(int.Parse).ToArray(); public static long[] ReadLongArray(this TextReader reader, char separator = ' ') => ReadStringArray(reader, separator).Select(long.Parse).ToArray(); public static double[] ReadDoubleArray(this TextReader reader, char separator = ' ') => ReadStringArray(reader, separator).Select(double.Parse).ToArray(); public static string[] ReadStringArray(this TextReader reader, char separator = ' ') => reader.ReadLine().Split(separator); // Supports primitive type only. public static T1 ReadValue(this TextReader reader) => (T1)Convert.ChangeType(reader.ReadLine(), typeof(T1)); public static (T1, T2) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); return (v1, v2); } public static (T1, T2, T3) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); var v3 = (T3)Convert.ChangeType(inputs[2], typeof(T3)); return (v1, v2, v3); } public static (T1, T2, T3, T4) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); var v3 = (T3)Convert.ChangeType(inputs[2], typeof(T3)); var v4 = (T4)Convert.ChangeType(inputs[3], typeof(T4)); return (v1, v2, v3, v4); } public static (T1, T2, T3, T4, T5) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); var v3 = (T3)Convert.ChangeType(inputs[2], typeof(T3)); var v4 = (T4)Convert.ChangeType(inputs[3], typeof(T4)); var v5 = (T5)Convert.ChangeType(inputs[4], typeof(T5)); return (v1, v2, v3, v4, v5); } public static (T1, T2, T3, T4, T5, T6) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); var v3 = (T3)Convert.ChangeType(inputs[2], typeof(T3)); var v4 = (T4)Convert.ChangeType(inputs[3], typeof(T4)); var v5 = (T5)Convert.ChangeType(inputs[4], typeof(T5)); var v6 = (T6)Convert.ChangeType(inputs[5], typeof(T6)); return (v1, v2, v3, v4, v5, v6); } public static (T1, T2, T3, T4, T5, T6, T7) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); var v3 = (T3)Convert.ChangeType(inputs[2], typeof(T3)); var v4 = (T4)Convert.ChangeType(inputs[3], typeof(T4)); var v5 = (T5)Convert.ChangeType(inputs[4], typeof(T5)); var v6 = (T6)Convert.ChangeType(inputs[5], typeof(T6)); var v7 = (T7)Convert.ChangeType(inputs[6], typeof(T7)); return (v1, v2, v3, v4, v5, v6, v7); } public static (T1, T2, T3, T4, T5, T6, T7, T8) ReadValue(this TextReader reader, char separator = ' ') { var inputs = ReadStringArray(reader, separator); var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1)); var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2)); var v3 = (T3)Convert.ChangeType(inputs[2], typeof(T3)); var v4 = (T4)Convert.ChangeType(inputs[3], typeof(T4)); var v5 = (T5)Convert.ChangeType(inputs[4], typeof(T5)); var v6 = (T6)Convert.ChangeType(inputs[5], typeof(T6)); var v7 = (T7)Convert.ChangeType(inputs[6], typeof(T7)); var v8 = (T8)Convert.ChangeType(inputs[7], typeof(T8)); return (v1, v2, v3, v4, v5, v6, v7, v8); } } } #endregion