package main

import "core:container/queue"
import "core:fmt"
import "core:math"
import "core:os"
import "core:slice"
import "core:strconv"
import "core:strings"
import "core:time"
import "core:unicode/utf8"

import util "../util"

find_max_jolts :: proc(line: ^string, battery_count: int) -> i64 {
	stack := make([dynamic]rune, 0, battery_count)
	for ch, i in line {
		for len(stack) > 0 && stack[len(stack) - 1] < ch && len(stack) + len(line) - i > battery_count {
			pop(&stack)
		}
		if len(stack) < battery_count do append(&stack, ch)
	}
	jolts, _ := strconv.parse_i64(utf8.runes_to_string(stack[:]))
	return jolts
}

part_1 :: proc(lines: ^[]string) {
	result: i64 = 0
	for &line in lines do result += find_max_jolts(&line, 2)
	fmt.printfln("  [Result] %v", result)
}

part_2 :: proc(lines: ^[]string) {
	result: i64 = 0
	for &line in lines do result += find_max_jolts(&line, 12)
	fmt.printfln("  [Result] %v", result)
}

main :: proc() {
	context.allocator = context.temp_allocator
	defer free_all(context.temp_allocator)

	INPUT :: #load("input.txt", string)
	lines := strings.split(INPUT, "\n")
	lines = lines[:len(lines) - 1]

	fmt.println("[Part 1]")
	start := time.tick_now()
	part_1(&lines)
	duration := time.tick_diff(start, time.tick_now())
	fmt.printfln("  [Time]   %fms\n", time.duration_milliseconds(duration))

	fmt.printfln("[Part 2]")
	start = time.tick_now()
	part_2(&lines)
	duration = time.tick_diff(start, time.tick_now())
	fmt.printfln("  [Time]   %fms", time.duration_milliseconds(duration))
}