Thanks! I like the Pair
destruction and zip().sumOf()
approach. I'm relatively new to Kotlin, so this is a good learning experience. ๐
proved_unglue
joined 3 months ago
Kotlin
No ๐ for Kotlin here?
import kotlin.math.abs
fun part1(input: String): Int {
val diffs: MutableList<Int> = mutableListOf()
val pair = parse(input)
pair.first.sort()
pair.second.sort()
pair.first.forEachIndexed { idx, num ->
diffs.add(abs(num - pair.second[idx]))
}
return diffs.sum()
}
fun part2(input: String): Int {
val pair = parse(input)
val frequencies = pair.second.groupingBy { it }.eachCount()
var score = 0
pair.first.forEach { num ->
score += num * frequencies.getOrDefault(num, 0)
}
return score
}
private fun parse(input: String): Pair<MutableList<Int>, MutableList<Int>> {
val left: MutableList<Int> = mutableListOf()
val right: MutableList<Int> = mutableListOf()
input.lines().forEach { line ->
if (line.isNotBlank()) {
val parts = line.split("\\s+".toRegex())
left.add(parts[0].toInt())
right.add(parts[1].toInt())
}
}
return left to right
}
Kotlin
A bit late to the party, but here you go.
import kotlin.math.abs
fun part1(input: String): Int {
return solve(input, ::isSafe)
}
fun part2(input: String): Int {
return solve(input, ::isDampSafe)
}
private fun solve(input: String, condition: (List<Int>) -> Boolean): Int {
var safeCount = 0
input.lines().forEach { line ->
if (line.isNotBlank()) {
val nums = line.split("\\s+".toRegex()).map { it.toInt() }
safeCount += if (condition(nums)) 1 else 0
}
}
return safeCount
}
private fun isSafe(list: List<Int>): Boolean {
val safeDiffs = setOf(1, 2, 3)
var incCount = 0
var decCount = 0
for (idx in 0..<list.lastIndex) {
if (!safeDiffs.contains(abs(list[idx] - list[idx + 1]))) {
return false
}
if (list[idx] <= list[idx + 1]) incCount++
if (list[idx] >= list[idx + 1]) decCount++
}
return incCount == 0 || decCount == 0
}
private fun isDampSafe(list: List<Int>): Boolean {
if (isSafe(list)) {
return true
} else {
for (idx in 0..list.lastIndex) {
val shortened = list.toMutableList()
shortened.removeAt(idx)
if (isSafe(shortened)) {
return true
}
}
}
return false
}
Nice, sometimes a few extra linebreaks can do the trick...
Kotlin
fun part1(input: String): Int {
val pattern = "mul\\((\\d{1,3}),(\\d{1,3})\\)".toRegex()
var sum = 0
pattern.findAll(input).forEach { match ->
val first = match.groups[1]?.value?.toInt()!!
val second = match.groups[2]?.value?.toInt()!!
sum += first * second
}
return sum
}
fun part2(input: String): Int {
val pattern = "mul\\((\\d{1,3}),(\\d{1,3})\\)|don't\\(\\)|do\\(\\)".toRegex()
var sum = 0
var enabled = true
pattern.findAll(input).forEach { match ->
if (match.value == "do()") enabled = true
else if (match.value == "don't()") enabled = false
else if (enabled) {
val first = match.groups[1]?.value?.toInt()!!
val second = match.groups[2]?.value?.toInt()!!
sum += first * second
}
}
return sum
}
Kotlin