在Java 8中添加后缀

Question

我需要一个函数，该函数返回一个包含{XXX11, XXX12, XXX13, XXX14, XXX21, ... , XXX43, XXX44}其中XXX是函数的整数参数。 Java 8风格的优雅方式是什么？ 如果前缀是一位，我已经做了如下操作：

/**
 * I need to get {<base>11, <base>12, ..., <base>44} but can't...
 * @return {<base>1, <base>2, <base>3, <base>4}
 */
Set<Integer> addOnlyOnePrefix(int base){
  return IntStream.range(1, 4).map(prefix -> base*10 + prefix)
      .boxed().collect(Collectors.toSet());
}

Answer 1

最简单的方法可能只是生成16个值：

Set<Integer> addOnlyTwoPrefixes(int base) {
  return IntStream.range(0, 16)
      .map(prefix ->
          base * 100 +             // Leading digits
          10 + 10 * (prefix / 4) + // Penultimate digit
          1 + (prefix % 4))        // Last digit
      .boxed()
      .collect(Collectors.toSet());
}

10+和1+应考虑prefix / 4和prefix % 4的范围是0-3，而不是1-4。

Answer 2

您可以flatMap 。

首先生成一个IntStream的1..4并将其乘以10。然后flatMap另一个IntStream的1..4给出11,12..21,22..

将base乘以100并将其添加到IntStream ：

Set<Integer> addTwoPrefixes(final int base) {
    final int prefix = base * 100;
    return IntStream
            .rangeClosed(1, 4).map(i -> i * 10)
            .flatMap(i -> IntStream.rangeClosed(1, 4).map(j -> i + j))
            .map(i -> base + i)
            .boxed()
            .collect(toCollection(LinkedHashSet::new));
}

注意，我已经收集到LinkedHashSet来保留顺序。

Answer 3

您可以使用递归来编写添加n个前缀的方法：

static IntStream addNPrefixes(int base, int n) {
    return n == 0 ? IntStream.of(base) : addNPrefixes(base, n - 1)
            .flatMap(b -> IntStream.rangeClosed(1, 4).map(prefix -> b * 10 + prefix));
}

然后，您想要的方法将如下所示：

static Set<Integer> addTwoPrefixesThenCollectToSet(int base) {
    return addNPrefixes(base, 2).boxed().collect(Collectors.toSet());
}

更改第一个方法很容易，如果您愿意，也可以传入前缀生成函数：

static IntStream addNPrefixes(int base, int n, Supplier<IntStream> prefixes) {
    return n == 0 ? IntStream.of(base) : addNPrefixes(base, n - 1, prefixes)
            .flatMap(b -> prefixes.get().map(prefix -> b * 10 + prefix));
}