Skip to content

Commit 44c1163

Browse files
committed
fix hardcoded values; create a val for MAX_POWER
1 parent bcb077a commit 44c1163

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/main/scala/org/codefork/aoc2018/Day11.scala

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@ object Day11 {
66

77
val INPUT = 7857
88

9+
val MAX_POWER = 4
10+
911
case class Square(x: Int, y: Int, squareSize: Int)
1012

1113
case class SquarePower(x: Int = 0, y: Int = 0, size: Int = 0, power: Int = 0)
@@ -42,14 +44,14 @@ object Day11 {
4244
*/
4345
@tailrec
4446
final def increasingSquares(cur: Square, last: SquarePower, largest: SquarePower): SquarePower = {
45-
if (cur.x + cur.squareSize - 1 <= 300 && cur.y + cur.squareSize - 1 <= 300) {
47+
if (cur.x + cur.squareSize - 1 <= width && cur.y + cur.squareSize - 1 <= height) {
4648
val sumOfEdges = cur.x.to(cur.x + cur.squareSize - 1).map(x_ => grid(x_, cur.y + cur.squareSize - 1)).sum +
4749
cur.y.to(cur.y + cur.squareSize - 1).map(y_ => grid(cur.x + cur.squareSize - 1, y_)).sum -
4850
grid(cur.x + cur.squareSize - 1, cur.y + cur.squareSize - 1)
4951
val newPower = last.power + sumOfEdges
5052
val newLast = SquarePower(cur.x, cur.y, cur.squareSize, newPower)
5153
val newLargest = if (newPower > largest.power) newLast else largest
52-
val potentialPowerOfNextSquare = newPower + (4 * (2 * (cur.squareSize + 1) - 1))
54+
val potentialPowerOfNextSquare = newPower + (MAX_POWER * (2 * (cur.squareSize + 1) - 1))
5355
if (potentialPowerOfNextSquare >= newLargest.power) {
5456
increasingSquares(
5557
cur = cur.copy(squareSize = cur.squareSize + 1),

0 commit comments

Comments
 (0)