在SQL中使用停止条件递归SELECT?

时间:2016-10-05 15:46:56

标签: sql postgresql recursion recursive-query

我的名为元素的表格如下所示:

 id | successor | important
----------------------------
  1 | NULL      | 0
  2 | 4         | 1
  3 | 5         | 0
  4 | 8         | 0
  5 | 6         | 1
  6 | 7         | 0
  7 | NULL      | 0
  8 | 10        | 1
  9 | 10        | 0
 10 | NULL      | 0

我从元素的ID开始。每个元素可以具有或不具有后续元素。因此,给定任何元素ID,我可以根据其后继者和后继者继承者构建来自0..n元素的元素链,依此类推。

假设我的起始ID是2.这导致以下链:

2 -> 4 -> 8 -> 10

现在我想问一下这个问题:一个特定的元素链是否包含至少一个重要的元素== 1?

在伪代码中,在没有不必要的检查的情况下实现此功能的函数可能如下所示:

boolean chainIsImportant(element)
{
    if (element.important == 1) {
        return true;
    }

    if (element.successor != NULL) {
        return chainIsImportant(element.successor);
    }

    return false;
}

我想这可以通过WITH RECURSIVE来实现,对吗?一旦找到具有重要== 1的元素,我该如何停止递归?

2 个答案:

答案 0 :(得分:3)

这通常通过聚合相关列并在CTE的递归部分中添加连接条件来完成:

with recursive all_elements as (
  select id, successor, important, array[important] as important_elements
  from elements
  where successor is null
  union all
  select c.id, c.successor, c.important, p.important_elements||c.important
  from elements c
     join all_elements p on c.successor = p.id
  where 1 <> all(p.important_elements)
)
select *
from all_elements;

请注意,条件是“翻转”的,因为where子句定义了应包含的那些行。

答案 1 :(得分:1)

CREATE TABLE booltree
        ( id INTEGER NOT NULL PRIMARY KEY
        , successor INTEGER REFERENCES booltree(id)
        , important Boolean NOT NULL
        );

INSERT INTO booltree(id , successor , important) VALUES
  ( 1, NULL     , False)
  ,(2, 4        , True)
  ,(3, 5        , False)
  ,(4, 8        , False)
  ,(5, 6        , True)
  ,(6, 7        , False)
  ,(7, NULL     , False)
  ,(8, 10       , True)
  ,(9, 10       , False)
 ,(10, NULL     , False)
        ;

-- SELECT * FROM booltree;

WITH RECURSIVE rec AS (
        SELECT id, important
        FROM booltree
        WHERE successor IS NULL
        UNION ALL
        SELECT bt.id, GREATEST(rec.important, bt.important) AS important
        FROM booltree bt
        JOIN rec ON bt.successor = rec.id
        )
SELECT id, important
FROM rec
ORDER BY important, id;

结果:

CREATE TABLE
INSERT 0 10
 id | important 
----+-----------
  1 | f
  6 | f
  7 | f
  9 | f
 10 | f
  2 | t
  3 | t
  4 | t
  5 | t
  8 | t
(10 rows)

注意:恕我直言,一旦找到真正的重要性,就无法停止递归(基本上,因为RECURSIVE UNIONS中不允许LEFT JOINS) 但是如果你正在寻找一个给定的id(或者它们的一组),那么也许你可以使用它作为开始条件,并向上搜索树。