birdulon
/
AdventOfCode
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Clean up scala

This commit is contained in:
Luke Hubmayer-Werner 2022-12-05 20:14:48 +10:30
parent 69461192ca
commit ae3f6655a9
1 changed files with 7 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
2022/day5.scala
View File

@ -16,13 +16,12 @@ extension (s: String) def partition2(sep: String = "\n\n", sep2 : String = "\n")
(ss(0).split(sep2), ss(1).split(sep2)) (ss(0).split(sep2), ss(1).split(sep2))
extension (aos: Seq[String]) def transpose(reverseX: Boolean = false, reverseY: Boolean = false): Seq[String] = extension (aos: Seq[String]) def transpose(reverseX: Boolean = false, reverseY: Boolean = false): Seq[String] =
val mapping: Function[Int, String] = val mapping: Function[Int, String] = if reverseY
if reverseY then (i => aos.reverse.map(_.substring(i,i+1)).reduceLeft(_+_)) then (i => aos.reverse.map(_.substring(i,i+1)).reduceLeft(_+_))
else (i => aos.map(_.substring(i,i+1)).reduceLeft(_+_)) else (i => aos.map(_.substring(i,i+1)).reduceLeft(_+_))
if reverseX then if reverseX
(0 until aos(0).length).reverse.map(mapping) then (0 until aos(0).length).reverse.map(mapping)
else else (0 until aos(0).length).map(mapping)
(0 until aos(0).length).map(mapping)
def getStacksString(stacks: Map[Int, Stack[Char]]): String = stacks.keys.toBuffer.sorted.map(stacks(_).last.toString).reduceLeft(_+_) def getStacksString(stacks: Map[Int, Stack[Char]]): String = stacks.keys.toBuffer.sorted.map(stacks(_).last.toString).reduceLeft(_+_)
@ -37,23 +36,14 @@ def getStacksString(stacks: Map[Int, Stack[Char]]): String = stacks.keys.toBuffe
var stacks: Map[Int, Stack[Char]] = stacks_s.map((k,v) => (k, v.toCharArray.to(Stack))).toMap var stacks: Map[Int, Stack[Char]] = stacks_s.map((k,v) => (k, v.toCharArray.to(Stack))).toMap
for order <- orders do for order <- orders do
val (amount, i_source, i_dest) = (order(0), order(1), order(2)) val (amount, source, dest) = (order(0), stacks(order(1)), stacks(order(2)))
val source = stacks(i_source)
val dest = stacks(i_dest)
// println(s"Moving $amount from ${source.mkString} to ${dest.mkString}")
// print(s"Doing amount $amount : ")
for _ <- 1 to amount do for _ <- 1 to amount do
dest.append(source.removeLast()) dest.append(source.removeLast())
// print(". ")
// print("\n")
// println(s"Moved $amount from ${source.mkString} to ${dest.mkString}")
println(s"Part 1: ${getStacksString(stacks)}") println(s"Part 1: ${getStacksString(stacks)}")
stacks = stacks_s.map((k,v) => (k, v.toCharArray.to(Stack))).toMap stacks = stacks_s.map((k,v) => (k, v.toCharArray.to(Stack))).toMap
for order <- orders do for order <- orders do
val (amount, i_source, i_dest) = (order(0), order(1), order(2)) val (amount, source, dest) = (order(0), stacks(order(1)), stacks(order(2)))
val source = stacks(i_source)
val dest = stacks(i_dest)
dest.appendAll(source.takeRight(amount)) dest.appendAll(source.takeRight(amount))
source.dropRightInPlace(amount) source.dropRightInPlace(amount)
println(s"Part 2: ${getStacksString(stacks)}") println(s"Part 2: ${getStacksString(stacks)}")