enumerate/multiset.hpp

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

Verified with

• test/mytest/enumerate_multisets.test.cpp

Code

// [0, K) 内の広義単調増加列

template <typename F>
void enumerate_multiset(int K, int N, F query) {
  vc<int> A(N);
  auto dfs = [&](auto& dfs, int p, int s) -> void {
    if (p == N) {
      query(A);
      return;
    }
    FOR(x, s, K) {
      A[p] = x;
      dfs(dfs, p + 1, x);
    }
  };
  dfs(dfs, 0, 0);
}

#line 1 "enumerate/multiset.hpp"
// [0, K) 内の広義単調増加列

template <typename F>
void enumerate_multiset(int K, int N, F query) {
  vc<int> A(N);
  auto dfs = [&](auto& dfs, int p, int s) -> void {
    if (p == N) {
      query(A);
      return;
    }
    FOR(x, s, K) {
      A[p] = x;
      dfs(dfs, p + 1, x);
    }
  };
  dfs(dfs, 0, 0);
}

Back to top page