/ Uiua / 2023 / 12.ua
12.ua
 1  &fras ⊣ &args
 2  
 3  ⊜(⍜(°□⊏1)(⊜⋕≠@,.)⊜□≠@ .)≠@\n.
 4  
 5  Take! ← (
 6    ⊙(⟜⊏⊙,-1)
 7    /×◡[
 8      ⊃(≥0◌
 9      | ≤⊙◌
10      | ¬∊⊙@.⬚@?↙¯⊃(⋅⋅∘|↙⊙⋅⋅∘)
11      | ⨬(1|≠@#⊏)≥0.-1-⊃(⋅⋅∘|∘))]
12    ⨬(0|^0-1-⊃(⋅⋅∘|⟜⋅∘))
13  )
14  Skip! ← ⨬(0|^0-1)≠@#◡(⊏-1⊙⋅∘)
15  Dp ← |4 memo(
16    ⨬(=0◌|+⊃Take!Dp Skip!Dp)>0.
17  )
18  
19  Solve! ← /+≡(Dp◡∩⧻ ^0 ∩°□°⊟)
20  ⊃Solve!(⍜(⊂@?)(♭↯5) ⊙(♭↯5)) Solve!∘