package day03

import readInput

fun main() {

    fun part1(input: List<String>): Int {
        var totalSum = 0
        for (s in input) {
            val half = s.length / 2
            val parts = s.chunked(half)
            val set1 = parts[0].toSet()
            val set2 = parts[1].toSet()
            for (c in set1) {
                if (set2.contains(c)) {
                    if (c.isLowerCase()) {
                        totalSum += (c.code - 96)
                        break
                    } else {
                        totalSum += (c.code - 38)
                        break
                    }
                }
            }
        }
        return totalSum
    }

    fun part2(input: List<String>): Int {
        var totalSum = 0
        var index = 0
        while (index < input.size) {
            val s1 = input[index].toSet()
            val s2 = input[index + 1].toSet()
            val s3 = input[index + 2].toSet()
            var found = false
            while (!found) {
                for (c in s1) {
                    if (s2.contains(c) && s3.contains(c)) {
                        totalSum += if (c.isLowerCase()) {
                            (c.code - 96)
                        } else {
                            (c.code - 38)
                        }
                        found = true
                        break
                    }
                }
                if (found)
                    break
            }
            index += 3
        }

        return totalSum
    }

    val testInput = "v<KEY>\n" +
            "<KEY>" +
            "PmmdzqPrVvPwwTWBwg\n" +
            "wMqvLMZHhHMvwLHjbvcjnnSBnvTQFn\n" +
            "ttgJtRGJQctTZtZT\n" +
            "CrZsJsPPZsGzwwsLwLmpwMDw"

    println(part1(testInput.split("\n")))
    println(part2(testInput.split("\n")))

    val input = readInput("03")
    println(part1(input))
    println(part2(input))
}
