aboutsummaryrefslogtreecommitdiff
path: root/aoc-2022-dotnet/Day07/Program.fs
diff options
context:
space:
mode:
Diffstat (limited to 'aoc-2022-dotnet/Day07/Program.fs')
-rw-r--r--aoc-2022-dotnet/Day07/Program.fs78
1 files changed, 78 insertions, 0 deletions
diff --git a/aoc-2022-dotnet/Day07/Program.fs b/aoc-2022-dotnet/Day07/Program.fs
new file mode 100644
index 0000000..89cd728
--- /dev/null
+++ b/aoc-2022-dotnet/Day07/Program.fs
@@ -0,0 +1,78 @@
+open System.IO
+open FParsec
+
+let fileSizeThreshold = 100_000
+let totalDiskSpace = 70_000_000
+let requiredUnusedSpace = 30_000_000
+
+type Node =
+ | File of int // file size
+ | Dir of string // dir name
+
+type Command =
+ | CD of string // dest
+ | LS of Node list // nodes in current dir
+
+let parseCommands input =
+ let pcd = pstring "$ cd " >>. restOfLine true |>> CD
+ let pfile = pint32 .>> pchar ' ' .>> restOfLine true |>> File
+ let pdir = pstring "dir " >>. restOfLine true |>> Dir
+ let pnode = pfile <|> pdir
+
+ let pls =
+ pstring "$ ls" >>. skipNewline >>. many pnode
+ |>> LS
+
+ let pcmd = pcd <|> pls
+ let pinput = many pcmd
+
+ match run pinput input with
+ | Success (result, _, _) -> result
+ | _ -> failwith "Invalid input format!"
+
+let combine =
+ function
+ | (_, "/") -> []
+ | (_ :: t, "..") -> t
+ | ([], "..") -> failwith "Can't go above root directory!"
+ | (path, dest) -> dest :: path
+
+let buildFilesystem commands =
+ let rec helper path =
+ function
+ | (CD dir) :: t -> helper (combine (path, dir)) t
+ | (LS list) :: t -> (path, list) :: helper path t
+ | [] -> []
+
+ commands |> helper [] |> Map.ofList
+
+let rec directorySize fileSystem path =
+ fileSystem
+ |> Map.find path
+ |> List.sumBy (function
+ | Dir dir -> directorySize fileSystem <| combine (path, dir)
+ | File size -> size)
+
+let part1 = Seq.filter ((>=) fileSizeThreshold) >> Seq.sum
+
+let part2 sizes =
+ let occupiedSpace = Seq.max sizes // root directory size
+ let unusedSpace = totalDiskSpace - occupiedSpace
+ let missingSpace = requiredUnusedSpace - unusedSpace
+ sizes |> Seq.filter ((<=) missingSpace) |> Seq.min
+
+let solution reduceSizes input =
+ let fileSystem = input |> parseCommands |> buildFilesystem
+
+ fileSystem
+ |> Map.keys
+ |> Seq.map (directorySize fileSystem)
+ |> reduceSizes
+
+let test = File.ReadAllText("test.txt")
+assert (solution part1 test = 95437)
+assert (solution part2 test = 24933642)
+
+let input = File.ReadAllText("input.txt")
+printfn "%d" <| solution part1 input
+printfn "%d" <| solution part2 input