patrick/advent-of-code-2023
1
0
Fork 0
Code Issues Pull Requests 2 Packages Releases Activity

Part 2

Browse Source
This commit is contained in:
Smaug123
2023-12-12 19:04:41 +00:00
parent 8401566fb9
commit 20a45caf8f
3 changed files with 106 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

111
AdventOfCode2023.FSharp/AdventOfCode2023.FSharp.Lib/Day12.fs
View File

@@ -7,61 +7,125 @@ open System.Globalization
[<RequireQualifiedAccess>]
module Day12 =
let rec solve (line : ReadOnlySpan<char>) (groups : IReadOnlyList<int>) (currentGroupIndex : int) =
let rec solve
(dict : Dictionary<int * int, 'a>)
(line : ReadOnlySpan<char>)
(groups : IReadOnlyList<int>)
(currentGroupIndex : int)
: 'a
=
if line.Length = 0 then
if currentGroupIndex = groups.Count then LanguagePrimitives.GenericOne else LanguagePrimitives.GenericZero
if currentGroupIndex = groups.Count then
LanguagePrimitives.GenericOne
else
LanguagePrimitives.GenericZero
elif currentGroupIndex = groups.Count then
if line.Contains '#' then
LanguagePrimitives.GenericZero
else
LanguagePrimitives.GenericOne
else
match dict.TryGetValue ((line.Length, currentGroupIndex)) with
| true, v -> v
| false, _ ->
let remaining =
let mutable remaining = -1
for i = currentGroupIndex to groups.Count - 1 do
remaining <- remaining + groups.[i] + 1
remaining
if remaining > line.Length then
dict.Add ((line.Length, currentGroupIndex), LanguagePrimitives.GenericZero)
LanguagePrimitives.GenericZero
else
match line.[0] with
| '#' ->
if currentGroupIndex >= groups.Count then LanguagePrimitives.GenericZero else
if currentGroupIndex >= groups.Count then
LanguagePrimitives.GenericZero
else
let mutable isOk = true
for i = 1 to groups.[currentGroupIndex] - 1 do
if isOk && (i >= line.Length || (line.[i] <> '#' && line.[i] <> '?')) then
isOk <- false
if not isOk then LanguagePrimitives.GenericZero else
if groups.[currentGroupIndex] < line.Length then
if not isOk then
LanguagePrimitives.GenericZero
else if groups.[currentGroupIndex] < line.Length then
if line.[groups.[currentGroupIndex]] = '#' then
LanguagePrimitives.GenericZero
else
solve (line.Slice (groups.[currentGroupIndex] + 1)) groups (currentGroupIndex + 1)
solve dict (line.Slice (groups.[currentGroupIndex] + 1)) groups (currentGroupIndex + 1)
else
solve ReadOnlySpan<_>.Empty groups (currentGroupIndex + 1)
| '.' ->
solve (line.Slice 1) groups currentGroupIndex
solve dict ReadOnlySpan<_>.Empty groups (currentGroupIndex + 1)
| '.' -> solve dict (line.Slice 1) groups currentGroupIndex
| '?' ->
let ifDot = solve (line.Slice 1) groups currentGroupIndex
let afterMark = line.IndexOfAnyExcept ('?', '#')
if afterMark >= 0 && groups.[currentGroupIndex] > afterMark then
// this group would extend into a dot if this were filled in!
let firstHash = line.IndexOf '#'
if firstHash >= 0 && firstHash < afterMark then
// this group *is* filled in, contradiction
LanguagePrimitives.GenericZero
else
solve dict (line.Slice afterMark) groups currentGroupIndex
else
let ifDot = solve dict (line.Slice 1) groups currentGroupIndex
dict.TryAdd ((line.Length - 1, currentGroupIndex), ifDot) |> ignore
let ifHash =
if currentGroupIndex >= groups.Count then LanguagePrimitives.GenericZero else
if currentGroupIndex >= groups.Count then
LanguagePrimitives.GenericZero
else
let mutable isOk = true
for i = 1 to groups.[currentGroupIndex] - 1 do
if isOk && (i >= line.Length || (line.[i] <> '#' && line.[i] <> '?')) then
isOk <- false
if not isOk then LanguagePrimitives.GenericZero else
if groups.[currentGroupIndex] < line.Length then
if groups.[currentGroupIndex] < line.Length && line.[groups.[currentGroupIndex]] = '#' then
if not isOk then
LanguagePrimitives.GenericZero
else if groups.[currentGroupIndex] < line.Length then
if
groups.[currentGroupIndex] < line.Length
&& line.[groups.[currentGroupIndex]] = '#'
then
LanguagePrimitives.GenericZero
else
solve (line.Slice (groups.[currentGroupIndex] + 1)) groups (currentGroupIndex + 1)
solve dict (line.Slice (groups.[currentGroupIndex] + 1)) groups (currentGroupIndex + 1)
else
solve ReadOnlySpan<_>.Empty groups (currentGroupIndex + 1)
solve dict ReadOnlySpan<_>.Empty groups (currentGroupIndex + 1)
ifDot + ifHash
| _ ->
if currentGroupIndex = groups.Count then LanguagePrimitives.GenericOne else LanguagePrimitives.GenericZero
if currentGroupIndex = groups.Count then
LanguagePrimitives.GenericOne
else
LanguagePrimitives.GenericZero
let part1 (s : string) =
use mutable lines = StringSplitEnumerator.make '\n' s
let mutable answer = 0uL
let arr = ResizeArray ()
for line in lines do
if not line.IsEmpty then
arr.Clear ()
use ints = StringSplitEnumerator.make' ',' (line.Slice (line.IndexOf ' ' + 1))
for int in ints do
arr.Add (Int32.Parse (int, NumberStyles.None, CultureInfo.InvariantCulture))
let solved = solve line arr 0
let solved = solve (Dictionary ()) line arr 0
answer <- answer + solved
answer
@@ -71,19 +135,24 @@ module Day12 =
let mutable answer = 0uL
let arr = ResizeArray ()
for line in lines do
if not line.IsEmpty then
arr.Clear ()
let spaceIndex = line.IndexOf ' '
for _ = 0 to 4 do
use ints = StringSplitEnumerator.make' ',' (line.Slice (spaceIndex + 1))
for int in ints do
arr.Add (Int32.Parse (int, NumberStyles.None, CultureInfo.InvariantCulture))
let sliced = line.Slice(0, spaceIndex).ToString ()
let line = String.Concat (sliced, '?', sliced, '?', sliced, '?', sliced, '?', sliced)
let solved = solve (line.AsSpan()) arr 0
let line =
String.Concat (sliced, '?', sliced, '?', sliced, '?', sliced, '?', sliced)
let solved = solve (Dictionary ()) (line.AsSpan ()) arr 0
printfn $"%s{line} : %i{solved}"
answer <- answer + solved

6
AdventOfCode2023.FSharp/AdventOfCode2023.FSharp/Program.fs
View File

@@ -219,7 +219,11 @@ module Program =
sw.Restart ()
let data = Day11.parse input
sw.Stop ()
Console.Error.WriteLine ((1_000.0 * float sw.ElapsedTicks / float Stopwatch.Frequency).ToString () + "ms parse")
Console.Error.WriteLine (
(1_000.0 * float sw.ElapsedTicks / float Stopwatch.Frequency).ToString ()
+ "ms parse"
)
sw.Restart ()
let part1 = Day11.solve data 2uL

4
AdventOfCode2023.FSharp/Test/TestDay12.fs
View File

@@ -16,9 +16,7 @@ module TestDay12 =
[<Test>]
let part2Sample () =
sample
|> Day12.part2
|> shouldEqual 525152uL
sample |> Day12.part2 |> shouldEqual 525152uL
[<Test>]
let part1Actual () =