ruby - 通过二维数组生成类继承关系树

Question

我最近在整理公司项目中的类继承关系，使用 Ruby 抽取出了一个类及其父类的组合，类似
[[B,A], [C,A], [E,D], [F,B] ...]
其中 [B,A] 中 B 为子类，A为父类，根类不只有一个，比如上例中的 A，D。我希望能能生成这些类的关系图:

黄舟 · Answer

先写个生成各个类下的子类。

a = [['B','A'], ['C','A'], ['E','D'], ['F','B'], ['G','E']]
b = ['A','B','C','D','E','F','G'] 
b.map{|e| a.map{|d|  d[0] if d[-1] == e}}

输出：

[["B", "C", nil, nil, nil],
 [nil, nil, nil, "F", nil],
 [nil, nil, nil, nil, nil],
 [nil, nil, "E", nil, nil],
 [nil, nil, nil, nil, "G"],
 [nil, nil, nil, nil, nil],
 [nil, nil, nil, nil, nil]]

每行代表一个类的子类集合。