A bit less verbose

diff --git a/AdventOfCode2017/Day7.fs b/AdventOfCode2017/Day7.fs
index 869c0a9..1dd4728 100644 (file)
--- a/AdventOfCode2017/Day7.fs
+++ b/AdventOfCode2017/Day7.fs
@@ -44,11 +44,11 @@ let rec findUnbalanced (tower : Tower) : (Tower * int) option =
     let rec weight tower =
         tower.Weight + (tower.Above |> Seq.map weight |> Seq.sum)
 
-    let towersByWeight = tower.Above |> Seq.groupBy weight
+    let towersByWeight = tower.Above |> Seq.groupBy weight |> Seq.sortBy (snd >> Seq.length)
 
     if towersByWeight |> Seq.length > 1 then
-        let unbalanced = towersByWeight |> Seq.minBy (snd >> Seq.length)
-        let others = towersByWeight |> Seq.maxBy (snd >> Seq.length)
+        let unbalanced = towersByWeight |> Seq.item 0
+        let others = towersByWeight |> Seq.item 1
         let delta = fst others - fst unbalanced
         let unbalanced' = unbalanced |> snd |> Seq.head
         findUnbalanced unbalanced' |> Option.orElse (Some (unbalanced', unbalanced'.Weight + delta))