package com.manalili.advent

import kotlin.math.abs

class Day06(val input: List<String>){
    var coordinates: List<ChronalCoordinate>
    var height: IntRange
    var width: IntRange
    init {
        coordinates = input.map {
            val (x, y) = it.split(", ")
            ChronalCoordinate(x.toInt(), y.toInt())
        }
        width = IntRange(coordinates.minBy { it.w }!!.w, coordinates.maxBy { it.w }!!.w)
        height = IntRange(coordinates.minBy { it.h }!!.h, coordinates.maxBy { it.h }!!.h)

    }

    fun largestArea(): Int {
        for (i in height) {
            for (j in width) {
                val coordinateList = coordinates.map { it to it.distance(j, i) }
                    .groupBy { it.second }
                    .minBy { it.key }

                if (coordinateList!!.value.size == 1) {
                    coordinateList.value.first().first.also { it.units++ }
                }
            }
        }
        return coordinates.filter { !it.isInfinite(height, width) }.maxBy { it.units }!!.units
    }

    fun regionDistanceAcross(limit: Int): Int {
        val area = mutableMapOf<Pair<Int, Int>, Int>()
        for (i in height) {
            for (j in width) {
                area[i to j] = coordinates.sumBy { it.distance(j, i) }
            }
        }
        return area.filterValues { it <  limit}.count()
    }
}

data class ChronalCoordinate(val w: Int, val h: Int){
    var units: Int = 0

    fun distance(x: Int, y: Int): Int {
        return abs(this.w - x) + abs(this.h - y)
    }

    fun isInfinite(height: IntRange, width: IntRange): Boolean{
        return this.let { w == width.first || w == width.last || h == height.first || h == height.last }
    }
}