enumerate/enumerate_cartesian_product.cpp

• View this file on GitHub
• Last update: 2026-01-29 02:59:39+08:00

Code

//enumerate product of [0, rs_i)
template<typename F>
requires invocable<F, vector<int>>
void enumerate_cartesian_product(vector<int> rs, F f) {
  vector<int> a(size(rs));
  auto dfs = [&](int i, auto &self) -> void {
    if (i == ssize(rs)) {
      f(a);
    } else {
      for(int j = 0; j < rs[i]; j++) {
        a[i] = j;
        self(i + 1, self);
      }
    }
  };
  dfs(0, dfs);
}

#line 1 "enumerate/enumerate_cartesian_product.cpp"
//enumerate product of [0, rs_i)
template<typename F>
requires invocable<F, vector<int>>
void enumerate_cartesian_product(vector<int> rs, F f) {
  vector<int> a(size(rs));
  auto dfs = [&](int i, auto &self) -> void {
    if (i == ssize(rs)) {
      f(a);
    } else {
      for(int j = 0; j < rs[i]; j++) {
        a[i] = j;
        self(i + 1, self);
      }
    }
  };
  dfs(0, dfs);
}

Back to top page