AtCoderのABC
次のABCの問題へ
前のABCの問題へ
ABC270-C Simple path
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("5 2 5");
WillReturn.Add("1 2");
WillReturn.Add("1 3");
WillReturn.Add("3 4");
WillReturn.Add("3 5");
//2 1 3 5
}
else if (InputPattern == "Input2") {
WillReturn.Add("6 1 2");
WillReturn.Add("3 1");
WillReturn.Add("2 5");
WillReturn.Add("1 2");
WillReturn.Add("4 1");
WillReturn.Add("2 6");
//1 2
}
else {
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
}
return WillReturn;
}
// 隣接リスト
static Dictionary<int, List<int>> mToNodeListDict = new Dictionary<int, List<int>>();
static int mX;
static int mY;
static void Main()
{
List<string> InputList = GetInputList();
int[] wkArr = { };
Action<string> SplitAct = pStr =>
wkArr = pStr.Split(' ').Select(pX => int.Parse(pX)).ToArray();
SplitAct(InputList[0]);
mX = wkArr[1];
mY = wkArr[2];
foreach (string EachStr in InputList.Skip(1)) {
SplitAct(EachStr);
int FromNode = wkArr[0];
int ToNode = wkArr[1];
if (mToNodeListDict.ContainsKey(FromNode) == false) {
mToNodeListDict[FromNode] = new List<int>();
}
if (mToNodeListDict.ContainsKey(ToNode) == false) {
mToNodeListDict[ToNode] = new List<int>();
}
mToNodeListDict[FromNode].Add(ToNode);
mToNodeListDict[ToNode].Add(FromNode);
}
ExecEulerTour();
var AnswerList = new List<int>();
foreach (JyoutaiDef EachJyoutai in mEulerTourJyoutaiList) {
AnswerList.Add(EachJyoutai.Node);
if (EachJyoutai.Node == mY) break;
}
var CntDict = new Dictionary<int, int>();
foreach (int EachInt in AnswerList) {
if (CntDict.ContainsKey(EachInt) == false) {
CntDict[EachInt] = 0;
}
CntDict[EachInt]++;
}
var AnswerList2 = new List<int>();
foreach (int EachInt in AnswerList) {
if (CntDict[EachInt] < 2) {
AnswerList2.Add(EachInt);
}
}
Console.WriteLine(IntEnumJoin(" ", AnswerList2));
}
struct JyoutaiDef
{
internal int Node;
internal int Level;
internal bool IsStart;
internal bool IsEnd;
}
// オイラーツアーを行い、下記の2通りのタイミングでノードをAddする
// 1 最初の訪問時
// 2 子の部分木を訪問完了時
static List<JyoutaiDef> mEulerTourJyoutaiList = new List<JyoutaiDef>();
static void ExecEulerTour()
{
var Stk = new Stack<JyoutaiDef>();
JyoutaiDef WillPush;
WillPush.Node = mX;
WillPush.Level = 0;
WillPush.IsStart = false; WillPush.IsEnd = true;
Stk.Push(WillPush);
WillPush.IsStart = true; WillPush.IsEnd = false;
Stk.Push(WillPush);
var VisitedSet = new HashSet<int>();
VisitedSet.Add(mX);
while (Stk.Count > 0) {
JyoutaiDef Popped = Stk.Pop();
mEulerTourJyoutaiList.Add(Popped);
if (Popped.IsEnd) {
continue;
}
if (mToNodeListDict.ContainsKey(Popped.Node)) {
foreach (int EachToNode in mToNodeListDict[Popped.Node]) {
if (VisitedSet.Add(EachToNode)) {
WillPush.Node = EachToNode;
WillPush.Level = Popped.Level + 1;
WillPush.IsStart = false; WillPush.IsEnd = true;
Stk.Push(WillPush);
WillPush.IsStart = true; WillPush.IsEnd = false;
Stk.Push(WillPush);
}
}
}
}
}
// セパレータとInt型の列挙を引数として、結合したstringを返す
static string IntEnumJoin(string pSeparater, IEnumerable<int> pEnum)
{
string[] StrArr = Array.ConvertAll(pEnum.ToArray(), pX => pX.ToString());
return string.Join(pSeparater, StrArr);
}
}
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>();
string wkStr;
while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
return WillReturn;
}
// 隣接リスト
static Dictionary<int, List<int>> mToNodeListDict = new Dictionary<int, List<int>>();
static int mX;
static int mY;
static void Main()
{
List<string> InputList = GetInputList();
int[] wkArr = { };
Action<string> SplitAct = pStr =>
wkArr = pStr.Split(' ').Select(pX => int.Parse(pX)).ToArray();
SplitAct(InputList[0]);
mX = wkArr[1];
mY = wkArr[2];
foreach (string EachStr in InputList.Skip(1)) {
SplitAct(EachStr);
int FromNode = wkArr[0];
int ToNode = wkArr[1];
if (mToNodeListDict.ContainsKey(FromNode) == false) {
mToNodeListDict[FromNode] = new List<int>();
}
if (mToNodeListDict.ContainsKey(ToNode) == false) {
mToNodeListDict[ToNode] = new List<int>();
}
mToNodeListDict[FromNode].Add(ToNode);
mToNodeListDict[ToNode].Add(FromNode);
}
var Que = new Queue<JyoutaiDef>();
JyoutaiDef WillEnqueue;
WillEnqueue.CurrNode = mX;
Que.Enqueue(WillEnqueue);
var VisitedSet = new HashSet<int>();
// 親ノード[ノード]なDict
var ParentNodeDict = new Dictionary<int, int>();
while (Que.Count > 0) {
JyoutaiDef Dequeued = Que.Dequeue();
if (Dequeued.CurrNode == mY) {
break;
}
foreach (int EachToNode in mToNodeListDict[Dequeued.CurrNode]) {
if (VisitedSet.Add(EachToNode)) {
ParentNodeDict[EachToNode] = Dequeued.CurrNode;
WillEnqueue.CurrNode = EachToNode;
Que.Enqueue(WillEnqueue);
}
}
}
var AnswerList = new List<int>() { mY };
while (true) {
int CurrNode = AnswerList.Last();
if (CurrNode == mX) break;
AnswerList.Add(ParentNodeDict[CurrNode]);
}
AnswerList.Reverse();
Console.WriteLine(IntEnumJoin(" ", AnswerList));
}
struct JyoutaiDef
{
internal int CurrNode;
}
// セパレータとInt型の列挙を引数として、結合したstringを返す
static string IntEnumJoin(string pSeparater, IEnumerable<int> pEnum)
{
string[] StrArr = Array.ConvertAll(pEnum.ToArray(), pX => pX.ToString());
return string.Join(pSeparater, StrArr);
}
}
解説
XからYまでのオイラーツアーを行って、
1回しか登場しないノードを空白区切りで出力すれば良いです。
別解として、
BFSで、ノードごとの親ノードをDictに保存してから、
ゴールからスタートまでの経路を求める方法もあります。