Postgres递归函数唯一值

时间:2014-02-02 20:07:19

标签: sql postgresql recursion graph

我有一个图表结构

Node(pid integer)
Neighbor(pid integer, pidn integer)

Node是微不足道的,我应该说Neighbor为每个节点存储其邻居列表。这是我正在测试的图表(Neighbor关系的内容):

PID | PIDN
==========
1   | 2
1   | 3
2   | 1
2   | 3
2   | 4
2   | 5
3   | 1
3   | 2
4   | 2
4   | 6
5   | 2
6   | 4

我想得到一个节点的所有邻居的集合,其度数小于固定数,所以我执行以下查询:

WITH RECURSIVE search_graph(root, depth) AS (
        SELECT n.pidn, 1
        FROM node p, neighbor n
        WHERE p.pid = n.pid
        AND p.pid = 1
      UNION
        SELECT nxt.pidn, sg.depth + 1
        FROM neighbor nxt, search_graph sg
        WHERE sg.root = nxt.PID
        AND sg.depth < 3
)
SELECT * FROM search_graph s;

起始节点为1,最大深度为3(如果您错过了它们)。我得到以下结果:

Node | Depth
============
2    | 1
3    | 1
1    | 2
3    | 2
4    | 2
5    | 2
2    | 2
2    | 3
3    | 3
1    | 3
4    | 3
5    | 3
6    | 3

因为它扩展了每个节点的所有子节点,包括访问过的子节点

0                               1
1               2                               3
2       1    3    4    5                    1       2
3      2 3  1 2  2 6   2                   2 3   1 3 4 5

虽然我想排除被访问的孩子,但产生:

Node | Depth
============
2    | 1
3    | 1
4    | 2
5    | 2
6    | 3

如果没有访问节点,我需要一种方法将结果添加到search_graph。

1 个答案:

答案 0 :(得分:4)

您是否阅读过有关WITH RECURISVE的Postgresql文档?

图表中有一些示例,其中一个看起来像是您问题的解决方案:

  

如果链接关系包含循环,则此循环将循环。因为我们需要“深度”   输出,只需将UNION ALL更改为UNION就不会消除循环。相反,我们需要   识别我们是否在遵循&gt;链接的特定路径时再次到达同一行。我们在易于循环的查询中添加两列路径和循环:

WITH RECURSIVE search_graph(id, link, data, depth, path, cycle) AS (
    SELECT g.id, g.link, g.data, 1,
      ARRAY[g.id],
      false
    FROM graph g
  UNION ALL
    SELECT g.id, g.link, g.data, sg.depth + 1,
      path || g.id,
      g.id = ANY(path)
    FROM graph g, search_graph sg
    WHERE g.id = sg.link AND NOT cycle
)
SELECT * FROM search_graph;