enumerate/product.hpp

• View this file on GitHub
• Last update: 2023-01-23 21:04:48+09:00
• Include: #include "enumerate/product.hpp"

Verified with

• test/mytest/enumerate_products.test.cpp
• test/mytest/fibonacci_search.test.cpp
• test_atcoder/abc233c.test.cpp
• test_atcoder/arc153f.test.cpp

Code

// [0, A0) x [0, A1) x ...
template <typename F>
void enumerate_product(vc<int> A, F query) {
  int N = len(A);
  auto dfs = [&](auto& dfs, vc<int>& p) -> void {
    int n = len(p);
    if (n == N) return query(p);
    FOR(x, A[n]) {
      p.eb(x);
      dfs(dfs, p);
      p.pop_back();
    }
  };
  vc<int> p;
  dfs(dfs, p);
}

#line 1 "enumerate/product.hpp"
// [0, A0) x [0, A1) x ...
template <typename F>
void enumerate_product(vc<int> A, F query) {
  int N = len(A);
  auto dfs = [&](auto& dfs, vc<int>& p) -> void {
    int n = len(p);
    if (n == N) return query(p);
    FOR(x, A[n]) {
      p.eb(x);
      dfs(dfs, p);
      p.pop_back();
    }
  };
  vc<int> p;
  dfs(dfs, p);
}

Back to top page