#581825 (C#(csc)) No.1293 2種類の道路

提出ソース

結果

問題	No.1293 2種類の道路
ユーザー	fairy_lettuce
提出日時	2020-11-20 21:56:31
言語	C#(csc) (csc 3.9.0)
結果	AC
実行時間	304 ms / 2,000 ms
コード長	14,897 bytes
コンパイル時間	3,234 ms
コンパイル使用メモリ	118,260 KB
実行使用メモリ	52,020 KB
最終ジャッジ日時	2024-07-23 12:56:14
合計ジャッジ時間	6,373 ms
ジャッジサーバーID （参考情報）	judge5 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	32 ms 25,328 KB
testcase_01	AC	33 ms 25,680 KB
testcase_02	AC	33 ms 27,368 KB
testcase_03	AC	32 ms 27,372 KB
testcase_04	AC	33 ms 27,464 KB
testcase_05	AC	32 ms 25,692 KB
testcase_06	AC	32 ms 27,268 KB
testcase_07	AC	32 ms 25,584 KB
testcase_08	AC	31 ms 25,588 KB
testcase_09	AC	288 ms 47,672 KB
testcase_10	AC	287 ms 49,460 KB
testcase_11	AC	291 ms 49,464 KB
testcase_12	AC	299 ms 49,720 KB
testcase_13	AC	297 ms 49,212 KB
testcase_14	AC	230 ms 49,848 KB
testcase_15	AC	236 ms 49,844 KB
testcase_16	AC	291 ms 48,820 KB
testcase_17	AC	304 ms 49,464 KB
testcase_18	AC	209 ms 47,404 KB
testcase_19	AC	198 ms 50,232 KB
testcase_20	AC	197 ms 52,020 KB
testcase_21	AC	213 ms 30,656 KB
testcase_22	AC	217 ms 30,264 KB
testcase_23	AC	205 ms 30,388 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

raw source code

using System;
using System.IO;
using System.Collections.Generic;
using System.Linq;
using System.Numerics;
using System.Diagnostics;
using System.Runtime.CompilerServices;

namespace FertiLib.Contest.C
{
	static class Program
	{
		public static void Solve(Scanner cin)
		{
			var (n, _d, _w) = cin.ReadValue<int, int, int>();
			var (a, b) = cin.ReadValueArray<int, int>(_d);
			var (c, d) = cin.ReadValueArray<int, int>(_w);

			var automobile = new UnionFind(n);
			var pedestrian = new UnionFind(n);
			for (int i = 0; i < _d; i++)
			{
				automobile.Unite(a[i] - 1, b[i] - 1);
			}
			for (int i = 0; i < _w; i++)
			{
				pedestrian.Unite(c[i] - 1, d[i] - 1);
			}
			for (int i = 0; i < n; i++)
			{
				automobile.Find(i);
				pedestrian.Find(i);
			}

			var autoSize = new long[n];
			var pedeSize = new long[n];
			for (int i = 0; i < n; i++)
			{
				autoSize[automobile.Find(i)]++;
				pedeSize[pedestrian.Find(i)]++;
			}

			var autoToPede = new Set<int>[n];
			for (int i = 0; i < n; i++)
			{
				autoToPede[i] = new Set<int>();
			}
			for (int i = 0; i < n; i++)
			{
				autoToPede[automobile.Find(i)].Add(pedestrian.Find(i));
			}
			long ans = 0;
			for (int i = 0; i < n; i++)
			{
				if (autoSize[i] == 0) continue;
				var sum = 0L;
				foreach (var e in autoToPede[i])
				{
					sum += pedeSize[e];
				}
				sum *= autoSize[i];
				ans += sum;
			}
			ans -= n;
			Console.WriteLine(ans);
		}

		public static void Main(string[] args)
		{
			var sw = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false };
			Console.SetOut(sw);
			var cin = new Scanner();
			Solve(cin);
			Console.Out.Flush();
		}

		public static void YESNO(bool condition) => Console.WriteLine(condition ? "YES" : "NO");
		public static void YesNo(bool condition) => Console.WriteLine(condition ? "Yes" : "No");
		public static void yesno(bool condition) => Console.WriteLine(condition ? "yes" : "no");

		public static bool Chmax<T>(ref T a, T b) where T : IComparable<T>
		{
			if (a.CompareTo(b) >= 0) return false;
			a = b;
			return true;
		}
		public static bool Chmin<T>(ref T a, T b) where T : IComparable<T>
		{
			if (a.CompareTo(b) <= 0) return false;
			a = b;
			return true;
		}
	}

	public class Set<T> : IEnumerable<T> where T : IComparable<T>
	{
		Node root;
		readonly IComparer<T> comparer;
		readonly Node nil;
		public bool IsMultiSet { get; set; }

		public Set() : this(Comparer<T>.Default) { }
		public Set(IComparer<T> comparer)
		{
			nil = new Node(default);
			root = nil;
			this.comparer = comparer;
		}
		public Set(Comparison<T> comparison) : this(Comparer<T>.Create(comparison)) { }

		public T this[int index]
		{
			get
			{
				if (index < 0 || root.Count < index) throw new ArgumentOutOfRangeException();
				return Find(root, index);
			}
		}

		public bool Add(T x) => Insert(ref root, x);
		public bool Remove(T x) => Erase(ref root, x);
		public void RemoveAt(int index)
		{
			if (index < 0 || Count < index) throw new ArgumentOutOfRangeException();
			EraseAt(ref root, index);
		}
		public bool Contains(T x) => this.Count == 0 ? false : EqualRange(x) > 0;
		public int LowerBound(T x) => BinarySearch(root, x, false);
		public int UpperBound(T x) => BinarySearch(root, x, true);
		public int EqualRange(T x) => UpperBound(x) - LowerBound(x);

		public void Clear() => root = nil;

		public T Min() => root.Min.Value;
		public T Max() => root.Max.Value;

		public T[] Items
		{
			get
			{
				var a = new T[root.Count];
				var k = 0;
				Walk(root, a, ref k);
				return a;
			}
		}
		public int Count => root.Count;
		public int Height => root.Height;

		public IEnumerator<T> GetEnumerator() { return Items.AsEnumerable().GetEnumerator(); }
		System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator() => GetEnumerator();

		bool Insert(ref Node p, T x)
		{
			if (p.Count == 0)
			{
				p = new Node(x);
				p.Left = p.Right = nil;
				p.Update();
				return true;
			}
			bool ret;
			var t = comparer.Compare(p.Value, x);
			if (t > 0)
			{
				ret = Insert(ref p.Left, x);
			}
			else if (t < 0)
			{
				ret = Insert(ref p.Right, x);
			}
			else
			{
				if (IsMultiSet == true) ret = Insert(ref p.Left, x);
				else return false;
			}
			Balance(ref p);
			return ret;
		}

		bool Erase(ref Node p, T x)
		{
			if (p.Count == 0) return false;
			var t = comparer.Compare(p.Value, x);
			bool ret;
			if (t < 0) ret = Erase(ref p.Right, x);
			else if (t > 0) ret = Erase(ref p.Left, x);
			else
			{
				ret = true;
				if (p.Right.Count == 0) { p = p.Left; return true; }
				if (p.Left.Count == 0) { p = p.Right; return true; }

				p.Value = p.Left.Max.Value;
				Erase(ref p.Left, p.Left.Max.Value);
			}
			Balance(ref p);

			return ret;
		}

		void EraseAt(ref Node p, int index)
		{
			var count = p.Left.Count;
			if (index < count) EraseAt(ref p.Left, index);
			else if (index > count) EraseAt(ref p.Right, index - count - 1);
			else
			{
				if (p.Left.Count == 0) { p = p.Right; return; }
				if (p.Right.Count == 0) { p = p.Left; return; }

				p.Value = p.Left.Max.Value;
				EraseAt(ref p.Left, index - 1);
			}

			Balance(ref p);
		}

		void Walk(Node t, T[] a, ref int k)
		{
			if (t.Count == 0) return;
			Walk(t.Left, a, ref k);
			a[k++] = t.Value;
			Walk(t.Right, a, ref k);
		}

		void Balance(ref Node p)
		{
			var balance = p.Left.Height - p.Right.Height;
			if (balance < -1)
			{
				if (p.Right.Left.Height - p.Right.Right.Height > 0) RotateR(ref p.Right);
				RotateL(ref p);
			}
			else if (balance > 1)
			{
				if (p.Left.Left.Height - p.Left.Right.Height < 0) RotateL(ref p.Left);
				RotateR(ref p);
			}
			else p.Update();
		}

		T Find(Node p, int index)
		{
			if (index < p.Left.Count) return Find(p.Left, index);
			if (index > p.Left.Count) return Find(p.Right, index - p.Left.Count - 1);
			return p.Value;
		}

		void RotateL(ref Node p)
		{
			var r = p.Right;
			p.Right = r.Left;
			r.Left = p;
			p.Update();
			r.Update();
			p = r;
		}

		void RotateR(ref Node p)
		{
			var l = p.Left;
			p.Left = l.Right;
			l.Right = p;
			p.Update();
			l.Update();
			p = l;
		}

		int BinarySearch(Node p, T x, bool isUpperBound)
		{
			if (p.Count == 0) throw new NullReferenceException();

			var node = p;

			var ret = 0;

			while (p.Count != 0)
			{
				var cmp = p.Value.CompareTo(x);
				if (cmp > 0 || (!isUpperBound && cmp == 0))
				{
					p = p.Left;
				}
				else
				{
					ret += p.Left.Count + 1;
					p = p.Right;
				}
			}

			return ret;
		}

		public override string ToString()
		{
			return string.Join(", ", Items);
		}

		class Node
		{
			public Node Left, Right;
			public T Value { get; set; }

			public int Count { get; private set; }
			public int Height { get; private set; }

			public Node Min
			{
				get
				{
					if (Left.Count == 0) return this;
					else return Left.Min;
				}
			}

			public Node Max
			{
				get
				{
					if (Right.Count == 0) return this;
					else return Right.Max;
				}
			}

			public Node(T value) => Value = value;

			public void Update()
			{
				Count = Left.Count + Right.Count + 1;
				Height = Math.Max(Left.Height, Right.Height) + 1;
			}

			public override string ToString() => $"Count = {Count}, Value = {Value}";
		}
	}

	public class UnionFind
	{
		public int Count { get; private set; }
		int[] parent;
		public int[] size;

		public UnionFind(int n)
		{
			parent = Enumerable.Range(0, n).ToArray();
			size = Enumerable.Repeat(1, n).ToArray();
		}

		public void Unite(int x, int y)
		{
			x = Find(x);
			y = Find(y);
			if (x == y) return;
			if (size[x] > size[y])
			{
				parent[y] = x;
				size[x] += size[y];
			}
			else
			{
				parent[x] = y;
				size[y] += size[x];
			}
		}

		public int Find(int x)
		{
			if (parent[x] == x) return x;
			else return parent[x] = Find(parent[x]);
		}

		public bool IsSame(int x, int y) => Find(x) == Find(y);
	}

	static class Extention
	{
		public static string Join<T>(this IEnumerable<T> x, string separator = "") => string.Join(separator, x);

		public static int UpperBound<T>(this IList<T> list, T value) => list.BinarySearch(value, true, 0, list.Count, Comparer<T>.Default);
		public static int LowerBound<T>(this IList<T> list, T value) => list.BinarySearch(value, false, 0, list.Count, Comparer<T>.Default);
		public static int BinarySearch<T>(this IList<T> list, T value, bool isUpperBound, int index, int length, Comparer<T> comparer)
		{
			var ng = index - 1;
			var ok = index + length;
			while (ok - ng > 1)
			{
				var mid = ng + (ok - ng) / 2;
				var res = comparer.Compare(list[mid], value);
				if (res < 0 || (isUpperBound && res == 0)) ng = mid;
				else ok = mid;
			}
			return ok;
		}
	}

	class Scanner
	{
		string[] s;
		int i;

		char[] separator = new char[] { ' ' };

		public Scanner()
		{
			s = new string[0];
			i = 0;
		}

		public string Read() => ReadString();

		public string ReadString()
		{
			if (i < s.Length) return s[i++];
			string st = Console.ReadLine();
			while (st == "") st = Console.ReadLine();
			s = st.Split(separator, StringSplitOptions.RemoveEmptyEntries);
			if (s.Length == 0) return ReadString();
			i = 0;
			return s[i++];
		}

		public string[] ReadStringArray(int N)
		{
			string[] Array = new string[N];
			for (int i = 0; i < N; i++)
			{
				Array[i] = ReadString();
			}
			return Array;
		}

		public int ReadInt() => int.Parse(ReadString());

		public int[] ReadIntArray(int N, int add = 0)
		{
			int[] Array = new int[N];
			for (int i = 0; i < N; i++)
			{
				Array[i] = ReadInt() + add;
			}
			return Array;
		}

		public long ReadLong() => long.Parse(ReadString());

		public long[] ReadLongArray(int N, long add = 0)
		{
			long[] Array = new long[N];
			for (int i = 0; i < N; i++)
			{
				Array[i] = ReadLong() + add;
			}
			return Array;
		}

		public double ReadDouble() => double.Parse(ReadString());

		public double[] ReadDoubleArray(int N, double add = 0)
		{
			double[] Array = new double[N];
			for (int i = 0; i < N; i++)
			{
				Array[i] = ReadDouble() + add;
			}
			return Array;
		}

		public T1 ReadValue<T1>() => (T1)Convert.ChangeType(ReadString(), typeof(T1));

		public (T1, T2) ReadValue<T1, T2>()
		{
			var inputs = ReadStringArray(2);
			var v1 = (T1)Convert.ChangeType(inputs[0], typeof(T1));
			var v2 = (T2)Convert.ChangeType(inputs[1], typeof(T2));
			return (v1, v2);
		}

		public (T1, T2, T3) ReadValue<T1, T2, T3>()
		{
			var inputs = ReadStringArray(3);
			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 (T1, T2, T3, T4) ReadValue<T1, T2, T3, T4>()
		{
			var inputs = ReadStringArray(4);
			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 (T1, T2, T3, T4, T5) ReadValue<T1, T2, T3, T4, T5>()
		{
			var inputs = ReadStringArray(5);
			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 (T1, T2, T3, T4, T5, T6) ReadValue<T1, T2, T3, T4, T5, T6>()
		{
			var inputs = ReadStringArray(6);
			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 (T1, T2, T3, T4, T5, T6, T7) ReadValue<T1, T2, T3, T4, T5, T6, T7>()
		{
			var inputs = ReadStringArray(7);
			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 T1[] ReadValueArray<T1>(int N)
		{
			var v1 = new T1[N];
			for (int i = 0; i < N; i++)
			{
				v1[i] = ReadValue<T1>();
			}
			return v1;
		}

		public (T1[], T2[]) ReadValueArray<T1, T2>(int N)
		{
			var (v1, v2) = (new T1[N], new T2[N]);
			for (int i = 0; i < N; i++)
			{
				var (t1, t2) = ReadValue<T1, T2>();
				v1[i] = t1;
				v2[i] = t2;
			}
			return (v1, v2);
		}

		public (T1[], T2[], T3[]) ReadValueArray<T1, T2, T3>(int N)
		{
			var (v1, v2, v3) = (new T1[N], new T2[N], new T3[N]);
			for (int i = 0; i < N; i++)
			{
				var (t1, t2, t3) = ReadValue<T1, T2, T3>();
				v1[i] = t1;
				v2[i] = t2;
				v3[i] = t3;
			}
			return (v1, v2, v3);
		}

		public (T1[], T2[], T3[], T4[]) ReadValueArray<T1, T2, T3, T4>(int N)
		{
			var (v1, v2, v3, v4) = (new T1[N], new T2[N], new T3[N], new T4[N]);
			for (int i = 0; i < N; i++)
			{
				var (t1, t2, t3, t4) = ReadValue<T1, T2, T3, T4>();
				v1[i] = t1;
				v2[i] = t2;
				v3[i] = t3;
				v4[i] = t4;
			}
			return (v1, v2, v3, v4);
		}

		public (T1[], T2[], T3[], T4[], T5[]) ReadValueArray<T1, T2, T3, T4, T5>(int N)
		{
			var (v1, v2, v3, v4, v5) = (new T1[N], new T2[N], new T3[N], new T4[N], new T5[N]);
			for (int i = 0; i < N; i++)
			{
				var (t1, t2, t3, t4, t5) = ReadValue<T1, T2, T3, T4, T5>();
				v1[i] = t1;
				v2[i] = t2;
				v3[i] = t3;
				v4[i] = t4;
				v5[i] = t5;
			}
			return (v1, v2, v3, v4, v5);
		}

		public (T1[], T2[], T3[], T4[], T5[], T6[]) ReadValueArray<T1, T2, T3, T4, T5, T6>(int N)
		{
			var (v1, v2, v3, v4, v5, v6) = (new T1[N], new T2[N], new T3[N], new T4[N], new T5[N], new T6[N]);
			for (int i = 0; i < N; i++)
			{
				var (t1, t2, t3, t4, t5, t6) = ReadValue<T1, T2, T3, T4, T5, T6>();
				v1[i] = t1;
				v2[i] = t2;
				v3[i] = t3;
				v4[i] = t4;
				v5[i] = t5;
				v6[i] = t6;
			}
			return (v1, v2, v3, v4, v5, v6);
		}

		public (T1[], T2[], T3[], T4[], T5[], T6[], T7[]) ReadValueArray<T1, T2, T3, T4, T5, T6, T7>(int N)
		{
			var (v1, v2, v3, v4, v5, v6, v7) = (new T1[N], new T2[N], new T3[N], new T4[N], new T5[N], new T6[N], new T7[N]);
			for (int i = 0; i < N; i++)
			{
				var (t1, t2, t3, t4, t5, t6, t7) = ReadValue<T1, T2, T3, T4, T5, T6, T7>();
				v1[i] = t1;
				v2[i] = t2;
				v3[i] = t3;
				v4[i] = t4;
				v5[i] = t5;
				v6[i] = t6;
				v7[i] = t7;
			}
			return (v1, v2, v3, v4, v5, v6, v7);
		}
	}
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード