我有一个图表结构
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。
答案 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;