如何使用LoadingCache将递归转换为迭代?

时间:2012-06-22 09:03:57

标签: java caching recursion tree guava

我完全重写了这个问题,因为原来的问题无法解决。为了保持简单,我使用Fibonacci数字作为玩具示例。

trivial recursive cached computation以非常长的堆栈跟踪结束,正如预期的那样。这就是为什么我想要一个像IterativeLoadingCache这样的抽象类, 我可以通过类似

之类的东西扩展here
@Override
protected Integer computeNonRecursivelly(Integer key) {
    final Integer x1 = getOrEnqueue(key-1);
    final Integer x2 = getOrEnqueue(key-2);
    if (x1==null) return null;
    if (x2==null) return null;
    return x1+x2;
}

并且在不使用递归的情况下将关注所有缓存和计算。

真的没有寻找斐波纳契数的有效计算。我需要一些允许使用缓存和递归函数的东西,其中递归深度可以任意高。

我已经有了一种解决方案,但它效率很低而且非常难看,所以我希望得到一些好的建议。如果有人需要它或者已经实现它,我也很好奇。

2 个答案:

答案 0 :(得分:4)

既然你已经改写了你的问题,这是一个新答案。

首先,我认为computeNonRecursivelly的实现仍然是递归的,因为getOrEnqueue会调用它。

我认为你不能直接使用Cache,因为你需要在你的计算中有2个步骤:一个表示所需值的依赖关系,另一个表示依赖关系是满足。它只有在你从未有过循环依赖时才会起作用(它与递归中的要求相同)。

这样,您可以对尚未包含在缓存中的依赖项(及其依赖项等)进行排队,然后按正确的顺序计算它们。有点像:

public abstract class TwoStepCacheLoader<K, V> extends CacheLoader<K, V> {
    public abstract Set<K> getDependencies(K key);
}

public class TwoStepCache<K, V> extends ForwardingLoadingCache<K, V> {
    private final TwoStepCacheLoader<K, V> loader;
    private LoadingCache<K, V> cache;

    public TwoStepCache(TwoStepCacheLoader<K, V> loader) {
        this.loader = loader;
        cache = CacheBuilder.newBuilder().build(loader);
    }

    @Override
    public V get(K key) 
            throws ExecutionException {
        V value = cache.getIfPresent(key);
        if (value != null) {
            return value;
        }

        Deque<K> toCompute = getDependenciesToCompute(key);
        return computeDependencies(toCompute);
    }

    private Deque<K> getDependenciesToCompute(K key) {
        Set<K> seen = Sets.newHashSet(key);
        Deque<K> dependencies = new ArrayDeque<K>(seen), toCompute = new ArrayDeque<K>(seen);
        do {
            for (K dependency : loader.getDependencies(dependencies.remove())) {
                if (seen.add(dependency) && // Deduplication in the dependencies
                    cache.getIfPresent(dependency) == null) {
                    // We need to compute it.
                    toCompute.push(dependency);
                    // We also need its dependencies.
                    dependencies.add(dependency);
                }
            }
        } while (!dependencies.isEmpty());
        return toCompute;
    }

    private V computeDependencies(Deque<K> toCompute)
            throws ExecutionException {
        V value;
        do {
            value = cache.get(toCompute.pop());
        } while (!toCompute.isEmpty());
        // The last computed value is for our key.
        return value;
    }

    @Override
    public V getUnchecked(K key) {
        try {
            return get(key);
        } catch (ExecutionException e) {
            throw new UncheckedExecutionException(e.getCause());
        }
    }

    @Override
    protected LoadingCache<K, V> delegate() {
        return cache;
    }
}

现在,您可以实现一个安全地调用缓存的TwoStepCacheLoader

public class Fibonacci {
    private LoadingCache<Integer, Integer> cache = new TwoStepCache<Integer, Integer>(new FibonacciCacheLoader());


    public int fibonacci(int n) {
        return cache.getUnchecked(n);
    }


    private class FibonacciCacheLoader extends TwoStepCacheLoader<Integer, Integer> {
        @Override
        public Set<Integer> getDependencies(Integer key) {
            if (key <= 1) {
                return ImmutableSet.of();
            }
            return ImmutableSet.of(key - 2, key - 1);
        }


        @Override
        public Integer load(Integer key)
                throws Exception {
            if (key <= 1) {
                return 1;
            }
            return cache.get(key - 2) + cache.get(key - 1);
        }
    }
}

我对它进行了单元测试,似乎运行正常。

答案 1 :(得分:3)

编辑:更改了实现,以便在多个线程中将相同的Expression作为参数传递时允许单个计算。

不要使用LoadingCache,只需将结果缓存在eval中(一旦修改为使用迭代而不是递归):

public Node eval(final Expression e) {
    if (e==null) return null;
    return cache.get(e, new Callable<Node>() {
        @Override
        public Node call() {
            final Node n0 = eval(leftExpression(e));
            final Node n1 = eval(rightExpression(e));
            return new Node(n0, n1);
        }
    });
}

private final Cache<Expression, Node> cache
= CacheBuilder.newBuilder().build();