package main

import (
	"bufio"
	"fmt"
	"log"
	"os"
	"strings"
)

const Time = "Time:      "
const Distance = "Distance:  "

type Paper struct {
	matrix    [][]int
	bigMatrix [][]int
}

func readInput(file *os.File) Paper {
	scanner := bufio.NewScanner(file)
	var paper Paper

	for scanner.Scan() {
		line := scanner.Text()
		if line == "" {
			break
		}

		if strings.HasPrefix(line, Time) {
			line = strings.TrimLeft(line, Time)
		} else {
			line = strings.TrimLeft(line, Distance)
		}

		parts := strings.Split(line, " ")
		if len(parts) == 0 {
			log.Fatalf("Wrong input: %s", line)
		}

		var numbers []int
		var numberString string
		for i := range parts {
			if parts[i] == "" {
				continue
			}

			numberString += parts[i]

			var number int
			n, err := fmt.Sscanf(parts[i], "%d", &number)
			if n != 1 || err != nil {
				log.Fatalf("Failed to read number: %s\n%s", parts[i], err)
			}

			numbers = append(numbers, number)
		}

		var bigNumber int
		n, err := fmt.Sscanf(numberString, "%d", &bigNumber)
		if n != 1 || err != nil {
			log.Fatalf("Failed to read number: %s\n%s", numberString, err)
		}

		paper.bigMatrix = append(paper.bigMatrix, []int{bigNumber})

		paper.matrix = append(paper.matrix, numbers)
		if len(paper.matrix) == 2 {
			break
		}
	}

	return paper
}

func part1(matrix [][]int) int {
	result := 1

	for i := range matrix[0] {
		min := matrix[1][i] / matrix[0][i]
		for {
			if min*(matrix[0][i]-min) > matrix[1][i] {
				break
			}

			min++
		}

		result *= matrix[0][i] - min - min + 1
	}

	return result
}

func main() {
	if len(os.Args) < 2 {
		log.Fatal("You need to specify a file!")
	}

	filePath := os.Args[1]
	file, err := os.Open(filePath)
	if err != nil {
		log.Fatalf("Failed to open %s!\n", filePath)

	}

	paper := readInput(file)
	fmt.Println("Part1:", part1(paper.matrix))
	fmt.Println("Part2:", part1(paper.bigMatrix))
}