競技プログラミングの鉄則
次の問題へ
前の問題へ
A45 Card Elimination
C#のソース
using System;
using System.Collections.Generic;
using System.Linq;
class Program
{
static string InputPattern = "InputX";
static List<string> GetInputList()
{
var WillReturn = new List<string>();
if (InputPattern == "Input1") {
WillReturn.Add("4 B");
WillReturn.Add("WBBR");
//Yes
}
else {
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
}
return WillReturn;
}
static void Main()
{
List<string> InputList = GetInputList();
string[] SplitArr = InputList[0].Split(' ');
string AStr = InputList[1];
int GoalVal = 0;
if (SplitArr[1] == "R") GoalVal = 2;
if (SplitArr[1] == "B") GoalVal = 1;
int SumMod = 0;
foreach (char EachChar in AStr) {
if (EachChar == 'B') SumMod++;
if (EachChar == 'R') SumMod += 2;
SumMod %= 3;
}
if (SumMod == GoalVal) {
Console.WriteLine("Yes");
}
else {
Console.WriteLine("No");
}
}
}
解説
白を0
青を1
赤を2
とすれば、mod 3の世界での足し算だと分かります。
足し算は交換法則が成り立つので
mod 3での総和を求めれば、解も分かります。