namespace aoc2021;

/// <summary>
/// Day 8: <see href="https://adventofcode.com/2021/day/8"/>
/// </summary>
public sealed class Day08 : Day
{
    private static readonly List<char[]> PossibleMappings =
        "abcdefg".ToCharArray().Permute().Select(m => m.ToArray()).ToList();
    
    public Day08() : base(8, "Seven Segment Search")
    {
    }

    private static int MatchDigit(string lit) =>
        lit switch
        {
            "1110111" => 0,
            "0010010" => 1,
            "1011101" => 2,
            "1011011" => 3,
            "0111010" => 4,
            "1101011" => 5,
            "1101111" => 6,
            "1010010" => 7,
            "1111111" => 8,
            "1111011" => 9,
            _ => -1
        };

    private static int Decode(string line)
    {
        var s = line.Split(" | ").Select(s => s.Split(' ')).ToList();
        int i;
        char[] lit;
        for (i = 0; i < PossibleMappings.Count; i++)
        {
            var matches = 0;
            var failed = false;
            foreach (var signal in s[0])
            {
                lit = "0000000".ToCharArray();
                foreach (var c in signal)
                    lit[Array.IndexOf(PossibleMappings[i], c)] = '1';
                
                var match = MatchDigit(new(lit));
                if (match == -1)
                {
                    failed = true;
                    break;
                }

                matches++;
            }

            if (failed) continue;
            if (matches == 10) break;
        }

        if (i == PossibleMappings.Count)
        {
            throw new("no mapping found");
        }

        var digits = new StringBuilder();
        foreach (var o in s[1])
        {
            lit = "0000000".ToCharArray();
            foreach (var c in o)
                lit[Array.IndexOf(PossibleMappings[i], c)] = '1';
            
            digits.Append(MatchDigit(new(lit)));
        }

        return int.Parse(digits.ToString());
    }

    public override object Part1() =>
        Input
            .Select(line => line.Split(" | ")[1].Split(' '))
            .Select(outputs => outputs.Count(o => new[] { 2, 3, 4, 7 }.Contains(o.Length)))
            .Sum();

    public override object Part2() => Input.Select(Decode).Sum();
}