Mathematica 中的树数据结构

Question

我主要使用mathematica 作为数学工作台和编写相对较小的临时程序。然而，我正在设计一个系统，打算在 Mathematica 中进行编程。我需要将数据存储在树中，并搜索和遍历树。尽管我知道如何实现树，但我更喜欢标准的、经过测试的代码。我在 Mathematica 用户 wiki 上查看了基本数据结构有哪些类型的包。尽管 Mathematica 文档中有一个小例子，但我还没有找到。

现在我的问题：

1. 某处是否有可用的数据结构（开源）包？

2. 您在数据结构方面使用了什么方法？逐步开发自己的 util 包？

（不是一个问题，只是一个评论。也许...缺乏（大量可用的）开源软件包是 Mathematica 没有应有的动力的原因。恐怕是先有鸡还是先有蛋的问题。）

Answer 1

在 Mathematica 中，您所做的大部分工作都是基于表达式。表达式自然具有树结构。对于深度优先遍历（这可能是最常见的），您可以使用诸如 Scan、Map、Cases 等函数。更传统的语言是没有简单的方法来保存表达式树中单个节点的身份，因为 Mathematica 中没有指针。此外，当您只需要在几个地方修改表达式时，Mathematica 中惯用的许多表达式操作都会复制整个表达式，因为表达式是不可变的。

使用不可变的 Mathematica 表达式作为树仍然有几个优点。一是，因为它们是不可变的，所以只需查看它们就很容易理解它们存储的内容（状态和行为不混合）。另一个是有高效且通用的函数，例如 Map、MapIndexed 或 Scan 来遍历它们。例如，访问者设计模式是不可见 - 它只是Map[f,tree ,Infinity]，内置于语言中。此外，还有诸如 Cases、Replace、ReplaceAll 等内置函数，允许您编写非常简洁和声明性的代码解构树，找到具有特定语法或满足某些条件的树片段等。由于树不限于仅从列表构建并且从不同头构建，因此可以有效地使用它来编写非常简洁的树处理代码。最后，本着 探索性和自下而上的编程，这缩短了开发周期并最终带来更好的设计。

也就是说，您当然可以实现“有状态”（可变）树数据结构。我怀疑，尚未完成的真正原因通常是与构建、修改和遍历这样的树相关的性能影响，因为它在每一步都会经历完整的符号评估过程（请参阅这篇文章了解更多详细信息）。有关如何在 Mathematica 上下文中使用二叉搜索树来获得相当高效的代码的 2 个示例，请参阅我的帖子 此处（通用符号设置）和此处（在编译代码的上下文中）。对于在 Mathematica 中惯用地构建数据结构的一般方法，我推荐 Roman Maeder 的书籍：“Mathematica 中的编程”、“Mathematica 程序员 I&II”，尤其是“Mathematica 中的计算机科学”。在后者中，他详细讨论了如何在 Mathematica 中实现二叉搜索树。 编辑 正如 @Simon 提到的，@Daniel Lichtblau 的演讲也是一个很好的资源，它展示了如何构建数据结构并使其高效。

关于在 Mathematica 中实现包含某些状态的数据结构的一般方法，以下是从我在 此 Mathgroup 线程 - 它实现“对”数据结构。

Unprotect[pair, setFirst, getFirst, setSecond, getSecond, new, delete];
ClearAll[pair, setFirst, getFirst, setSecond, getSecond, new, delete];
Module[{first, second},
  first[_] := {};
  second[_] := {};
  pair /: new[pair[]] := pair[Unique[]];
  pair /: pair[tag_].delete[] := (first[tag] =.; second[tag] =.);
  pair /: pair[tag_].setFirst[value_] := first[tag] = value;
  pair /: pair[tag_].getFirst[] := first[tag];
  pair /: pair[tag_].setSecond[value_] := second[tag] = value;
  pair /: pair[tag_].getSecond[] := second[tag];
  Format[pair[x_Symbol]] := "pair[" <> ToString[Hash[x]] <> "]";
];
Protect[pair, setFirst, getFirst, setSecond, getSecond, new, delete]; 

以下是如何使用它：

pr = new[pair[]];
pr.setFirst[10];
pr.setSecond[20];
{pr.getFirst[], pr.getSecond[]}

{10, 20}

创建新对对象的列表：

pairs = Table[new[pair[]], {10}]

{"pair[430427975]", "pair[430428059]", "pair[430428060]", "pair[430428057]",
"pair[430428058]", "pair[430428063]", "pair[430428064]", "pair[430428061]", 
"pair[430428062]", "pair[430428051]"}

设置字段：

Module[{i},
 For[i = 1, i <= 10, i++,
  pairs[[i]].setFirst[10*i];
  pairs[[i]].setSecond[20*i];]]

检查字段：

#.getFirst[] & /@ pairs

{10, 20, 30, 40, 50, 60, 70, 80, 90, 100}

#.getSecond[] & /@ pairs

{20, 40, 60, 80, 100, 120, 140, 160, 180, 200} 

在我提到的帖子中有更详细的讨论。以这种方式创建的“对象”的一个大问题是它们没有自动垃圾收集，这可能是顶级 Mathematica 本身实现的 OOP 扩展没有真正起飞的主要原因之一。

Mathematica 有几个 OOP 扩展，例如 Roman Maeder 的 classes.m 包（源代码在他的“Mathematica Programmer”书中）、Objectica 商业包、和其他几个。但是，在 Mathematica 本身提供有效的机制（可能基于某种指针或引用机制）来构建可变数据结构（如果发生这种情况）之前，与此类数据结构的顶级实现相关的性能可能会受到很大影响在MMA。此外，由于 mma 基于不变性作为核心思想之一，因此使可变数据结构与 Mathematica 编程的其他习惯很好地契合并不容易。

编辑

这是一个与上面示例类似的基本状态树实现：

Module[{parent, children, value},
  children[_] := {};
  value[_] := Null;
  node /: new[node[]] := node[Unique[]];
  node /: node[tag_].getChildren[] := children[tag];
  node /: node[tag_].addChild[child_node, index_] := 
        children[tag] = Insert[children[tag], child, index];
  node /: node[tag_].removeChild[index_] := 
        children[tag] = Delete[children[tag], index];
  node /: node[tag_].getChild[index_] := children[tag][[index]];
  node /: node[tag_].getValue[] := value[tag];
  node /: node[tag_].setValue[val_] := value[tag] = val;
];

一些使用示例：

In[68]:= root = new[node[]]

Out[68]= node[$7]

In[69]:= root.addChild[new[node[]], 1]

Out[69]= {node[$8]}

In[70]:= root.addChild[new[node[]], 2]

Out[70]= {node[$8], node[$9]}

In[71]:= root.getChild[1].addChild[new[node[]], 1]

Out[71]= {node[$10]}

In[72]:= root.getChild[1].getChild[1].setValue[10]

Out[72]= 10

In[73]:= root.getChild[1].getChild[1].getValue[]

Out[73]= 10

有关使用此可变树数据结构的一个重要示例，请参阅我的这篇文章。它还将这种方法与更加重用 Mathematica 本地数据结构和函数的方法进行比较，并很好地说明了本文开头讨论的要点。

Answer 2

我主要将mathematica用作数学工作台和编写相对较小的临时程序。

Mathematica 在这方面确实很擅长。

您在数据结构方面使用了什么方法？逐步开发自己的util包？

我避免在 Mathematica 中创建自己的数据结构，因为它无法有效地处理它们。具体来说，Mathematica 中的一般数据结构往往比其他地方慢 10-1,000 倍，这极大地限制了它们的实际用途。例如， Mathematica 比 F# 慢 100 倍计算红黑树的深度范围。

使用列表进行逻辑编程就是一个例子，其中 Mathematica 通常比其他编译语言慢几个数量级。以下 Mathematica 程序使用链表来解决 n 皇后问题：

safe[{x0_, y0_}][{x1_, y1_}] := 
 x0 != x1 && y0 != y1 && x0 - y0 != x1 - y1 && x0 + y0 != x1 + y1

filter[_, {}] := {}
filter[p_, {h_, t_}] := If[p[h], {h, filter[p, t]}, filter[p, t]]

search[n_, nqs_, qs_, {}, a_] := If[nqs == n, a + 1, a]
search[n_, nqs_, qs_, {q_, ps_}, a_] := 
 search[n, nqs, qs, ps, 
  search[n, nqs + 1, {q, qs}, filter[safe[q], ps], a]]

ps[n_] := 
 Fold[{#2, #1} &, {}, Flatten[Table[{i, j}, {i, n}, {j, n}], 1]]

solve[n_] := search[n, 0, {}, ps[n], 0]

这是等效的 F#：

let safe (x0, y0) (x1, y1) =
  x0<>x1 && y0<>y1 && x0-y0<>x1-y1 && x0+y0<>x1+y1

let rec filter f = function
  | [] -> []
  | x::xs -> if f x then x::filter f xs else filter f xs

let rec search n nqs qs ps a =
  match ps with
  | [] -> if nqs=n then a+1 else a
  | q::ps ->
      search n (nqs+1) (q::qs) (filter (safe q) ps) a
      |> search n nqs qs ps

let ps n =
  [ for i in 1..n do
      for j in 1..n do
        yield i, j ]

let solve n = search n 0 [] (ps n) 0

solve 8

解决 8 皇后问题使用 Mathematica 需要 10.5 秒，使用 F# 需要 0.07 秒。因此，在这种情况下，F# 比 Mathematica 快 150 倍。

Stack Overflow 问题Mathematica“链接列表”和性能给出了一个更极端的例子。将该 Mathematica 代码简单地转换为 F# 可以得到一个等效程序，其运行速度比 Mathematica 快 4,000 到 200,000 倍：

let rand = System.Random()
let xs = List.init 10000 (fun _ -> rand.Next 100)
Array.init 100 (fun _ ->
  let t = System.Diagnostics.Stopwatch.StartNew()
  ignore(List.length xs)
  t.Elapsed.TotalSeconds)

具体来说，Mathematica 执行一次迭代需要 0.156 秒到 16 秒，而 F# 需要 42 微秒到 86 微秒。

如果我真的想留在 Mathematica，那么我会把我所做的一切都塞进 Mathematica 的一些内置数据结构中，例如 Dispatch。