From a903c0a3970cd72dbd8c29929a6fa2ec7a513b13 Mon Sep 17 00:00:00 2001
From: Dnomd343 <dnomd343@gmail.com>
Date: Sun, 3 Nov 2024 15:47:37 +0800
Subject: [PATCH] feat: minimum-step demo of group `169-0u`

---
 graph/build.py | 83 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 83 insertions(+)
 create mode 100755 graph/build.py

diff --git a/graph/build.py b/graph/build.py
new file mode 100755
index 0000000..3104f17
--- /dev/null
+++ b/graph/build.py
@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+
+import pickle
+import igraph as ig
+
+from klotski import FastCal
+from klotski import CommonCode
+from klotski import Group, GroupUnion
+
+
+def build_step_map(code: str) -> dict[CommonCode, int]:
+    cal = FastCal(CommonCode(code))
+    cal.build_all()
+    data = cal.exports()
+    result = {}
+    for step_num, cases in enumerate(data):
+        for case in cases:
+            result[case] = step_num
+    return result
+
+
+def build_min_step_scope(group: Group, targets: list[str]) -> dict[CommonCode, dict[CommonCode, int]]:
+    targets = [CommonCode(x) for x in targets]
+    all_solution = set([x for x in group.cases() if str(x).startswith('D')])
+    not_solved = all_solution - set(targets)
+
+    step_data = {x: {} for x in group.cases() if x not in not_solved}
+    for target in targets:
+        for code, step_num in build_step_map(target).items():
+            if code in not_solved:
+                continue
+            step_data[code][target] = step_num
+
+    min_step_data = {}
+    for code, step_info in step_data.items():
+        min_step = min([x for x in step_info.values()])
+        min_step_cases = [x for x, y in step_info.items() if y == min_step]
+        min_step_data[code] = (min_step, min_step_cases)
+
+    result = {}
+    for target in targets:
+        tmp = [(x, info[0]) for x, info in min_step_data.items() if target in info[1]]
+        result[target] = dict(sorted(tmp, key=lambda x: (x[1], x[0])))
+    return result
+
+
+def build_graph(nodes: dict[CommonCode, int]) -> ig.Graph:
+    edges = []
+    for code, step in nodes.items():
+        for next_case in code.next_cases():
+            if next_case not in nodes:
+                continue
+            assert step - nodes[next_case] in [-1, 0, 1]
+            if nodes[next_case] - step == 1:
+                edges.append((code, next_case))
+
+    graph = ig.Graph(n=len(nodes))
+    graph.vs['code'] = [x.to_string(shorten=False)[:7] for x in nodes]
+    graph.vs['step'] = list(nodes.values())
+    nodes_index = {y: x for x, y in enumerate(nodes)}
+    graph.add_edges([(nodes_index[x], nodes_index[y]) for x, y in edges])
+    return graph
+
+
+def save_graph(name: str, graph: ig.Graph) -> None:
+    with open(f'{name}.pkl', 'wb') as fp:
+        pickle.dump(graph, fp)
+
+
+def main() -> None:
+    targets = [
+        'DAAF4CC', 'DABFD20', 'DAEF720', 'DAEFD20', 'DAFED20', 'DBAB4CC', 'DBAFD20', 'DEAB4CC', 'DEAFD20', 'DEEB60C', 'DFAA4CC', 'DFAED20',
+        'DAA7F30', 'DAB9F30', 'DABDF80', 'DAE9F30', 'DAEDF80', 'DAFA730', 'DAFB780', 'DBADF80', 'DBAF780', 'DBBE930', 'DEADF80', 'DFAB780',
+        'DAF60EC', 'DBE60EC', 'DBE70AC', 'DBF682C', 'DFA70AC', 'DFE90AC',
+    ]
+    group = GroupUnion(169).groups()[1]
+    data = build_min_step_scope(group, targets)
+    for target in targets:
+        save_graph(target, build_graph(data[CommonCode(target)]))
+
+
+if __name__ == '__main__':
+    main()