算法：从多个有序列表中搜索增量整数序列

Question

我试图找到最有效的方法，从排序整数数组（列表）的数组（或列表）中找到整数序列。假设我有这个阵列

{
{1, 3, 4, 9, 15},
{5, 10, 13},
{2, 6, 11, 17},
{7, 8, 12}
}

然后解应该返回连续递增数的数组数组

{ {4, 5, 6, 7}, {9, 10, 11, 12} }

在结果中，4和9来自第一个子数组，5和10来自第二个子数组，依此类推。约束条件是，最终输出中的所有数组必须具有与输入中的数组数相同的固定长度，即，输入中的所有数组必须在结果中的每个数组中提供1个元素。

我们可以假设输入数组已排序，并且输入数组中没有重复项，例如，在示例中，如果9在数组1中，我可以假设9不会在任何其他输入数组中。

有什么有效的方法可以做到吗？除了强制数组之外，我想不出任何东西。使用任何数据结构和算法都是可以接受的。

谢谢你。

Answer 1

>

从第一个数组中选取第一个值，并将其另存为prevValue。

在每个输入数组中构建一个位置（索引）数组，并将其初始化为0。

迭代输入数组：

>

将prevValue更新为找到的值。

现在，位置数组存储下一个递增值序列的位置。将值保存到结果。

重复步骤3和4，直到完成。

作为代码，应该是这样的：

private static int[][] findIncrementalSequences(int[][] input) {
    int[] pos = new int[input.length];
    int prevValue = Integer.MIN_VALUE;
    List<int[]> results = new ArrayList<>();
    for (;;) {
        int[] result = new int[pos.length];
        for (int i = 0; i < pos.length; i++) {
            while (pos[i] < input[i].length && input[i][pos[i]] <= prevValue)
                pos[i]++;
            if (pos[i] == input[i].length)
                return results.toArray(new int[results.size()][]);
            prevValue = result[i] = input[i][pos[i]];
        }
        results.add(result);
    }
}

测试

int[][] input = { {1, 3, 4, 9, 15},
                  {5, 10, 13},
                  {2, 6, 11, 17},
                  {7, 8, 12} };
System.out.println(Arrays.deepToString(findIncrementalSequences(input)));

输出

[[1, 5, 6, 7], [9, 10, 11, 12]]

使现代化

如果每个子结果必须是连续递增值，则可以修改代码以在输入[0]处重新启动，如果下一个数字不完全比前一个值高1。

下面的代码也略有更改，以删除重复的数组查找。

private static int[][] findConsecutiveSequences(int[][] input) {
    int[] pos = new int[input.length];
    int nextValue = Integer.MIN_VALUE, value;
    List<int[]> results = new ArrayList<>();
    for (;;) {
        int[] result = new int[pos.length];
        for (int i = 0; i < pos.length; i++) {
            for (;;) {
                if (pos[i] == input[i].length)
                    return results.toArray(new int[results.size()][]);
                if ((value = input[i][pos[i]]) >= nextValue)
                    break;
                pos[i]++;
            }
            if (i == 0 || value == nextValue) {
                result[i] = value;
                nextValue = value + 1;
            } else {
                i = -1; // Restart with input[0]
            }
        }
        results.add(result);
    }
}

输出

[[4, 5, 6, 7], [9, 10, 11, 12]]