在F#中创建复合迭代器

Question

我正在实现类似跳棋的游戏,我需要一个列举给定配置的所有合法移动的序列.

我有以下函数,直接从C#翻译:

seq {
    for y1 = 0 to BOARDSIZE-1 do
        for x1 = 0 to BOARDSIZE-1 do
             for dy = -2 to 2 do
                 for dx = -2 to 2 do
                     let x2 = x1 + dx;
                     let y2 = y1 + dy;
                     let currentMove = new MoveStruct(x1, y1, x2, y2);
                     if (currentMove.SomeCondition = true) then
                             yield currentMove;
   }

它很有效,但它很笨拙,并不是"F#方式",更不用说我有一种潜在的怀疑,即我在这里所做的并不是性能最佳.

我想要的是"将其展平"为使用"遍历所有单元格"的组合,"迭代来自此单元格的所有有效移动".

以下是我希望结合的功能:

let AllCells =
    seq {
        for y=0 to BOARDSIZE-1 do
            for x=0 to BOARDSIZE-1 do
                yield (x,y);
    };

和

let LegalMovesAround(x1,y1) = 
    seq {
      if board.[x1, y1] = WHITE then
        for dy = -2 to 2 do
          for dx = -2 to 2 do
                let x2 = x1 + dx;
                let y2 = y1 + dy;
                let currentMove = new MoveStruct(x1, y1, x2, y2);
                if (currentMove.DetermineMoveType <> MoveType.ILLEGAL 
                    && board.[x2, y2] = NONE) then
                        yield currentMove;
     }

我将不遗余力地告诉你我各种尝试的细节,因为它们都没有成功.但是长话短说,我能想到的最好的是一个迭代器,它返回每个yield的seq,而不是我正在寻找的扁平版本,这将返回一个简单的MoveStruct.

任何人都知道如何组合AllCells和LegalMovesAround(x,y)？

此致,Aleks

Answer 1

您应该能够按原样组合它们，然后展平，如下所示：

 let validMoves = 
    AllCells 
    |> Seq.collect LegalMovesAround
    |> Seq.distinct

但就性能而言，它可能不是最佳解决方案。

编辑：根据托马斯评论修复了示例代码