32 lines
761 B
Text
32 lines
761 B
Text
import Lean.Util.SCC
|
|
|
|
open Lean.SCC
|
|
|
|
def checkSCC (expected : List (List Nat)) (vertices : List Nat) (successorsOf : Nat → List Nat) : IO Unit := do
|
|
let r := scc vertices successorsOf
|
|
IO.println r
|
|
unless expected == r do
|
|
throw $ IO.userError ("expected result " ++ toString expected)
|
|
|
|
#eval checkSCC [[2], [1, 3], [4]] [1, 2, 3, 4] fun x => match x with
|
|
| 1 => [2, 3]
|
|
| 3 => [1]
|
|
| 4 => [1, 2]
|
|
| _ => []
|
|
|
|
#eval checkSCC [[1], [2], [3], [4]] [1, 2, 3, 4] fun x => match x with
|
|
| _ => []
|
|
|
|
#eval checkSCC [[1, 2, 3, 4]] [1, 2, 3, 4] fun x => match x with
|
|
| 1 => [2]
|
|
| 2 => [3]
|
|
| 3 => [4]
|
|
| 4 => [1]
|
|
| _ => []
|
|
|
|
#eval checkSCC [[1, 2, 4], [3]] [1, 2, 3, 4] fun x => match x with
|
|
| 1 => [2]
|
|
| 2 => [4]
|
|
| 3 => [4]
|
|
| 4 => [1]
|
|
| _ => []
|