feat(cses): more dp
This commit is contained in:
parent
3153490991
commit
e0a8333591
32 changed files with 1052 additions and 0 deletions
90
cses/dynamic-programming/dice-combinations.cc
Normal file
90
cses/dynamic-programming/dice-combinations.cc
Normal file
|
|
@ -0,0 +1,90 @@
|
|||
#include <bits/stdc++.h> // {{{
|
||||
|
||||
// https://codeforces.com/blog/entry/96344
|
||||
|
||||
#pragma GCC optimize("O2,unroll-loops")
|
||||
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
|
||||
|
||||
using namespace std;
|
||||
|
||||
using i32 = int32_t;
|
||||
using u32 = uint32_t;
|
||||
using i64 = int64_t;
|
||||
using u64 = uint64_t;
|
||||
using d64 = double;
|
||||
using d128 = long double;
|
||||
template <typename T>
|
||||
using vec = std::vector<T>;
|
||||
template <typename T, size_t N>
|
||||
using arr = std::array<T, N>;
|
||||
template <typename T1, typename T2>
|
||||
using pai = std::pair<T1, T2>;
|
||||
|
||||
template <typename T>
|
||||
constexpr T MIN = std::numeric_limits<T>::min();
|
||||
|
||||
template <typename T>
|
||||
constexpr T MAX = std::numeric_limits<T>::max();
|
||||
|
||||
template <typename T>
|
||||
[[nodiscard]] static T sc(auto&& x) {
|
||||
return static_cast<T>(x);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
[[nodiscard]] static T sz(auto&& x) {
|
||||
return static_cast<T>(x.size());
|
||||
}
|
||||
|
||||
static void NO() {
|
||||
std::cout << "NO\n";
|
||||
}
|
||||
|
||||
static void YES() {
|
||||
std::cout << "YES\n";
|
||||
}
|
||||
|
||||
#define all(x) (x).begin(), (x).end()
|
||||
#define rall(x) (x).rbegin(), (x).rend()
|
||||
|
||||
#ifdef LOCAL
|
||||
#define db(...) std::print(__VA_ARGS__)
|
||||
#define dbln(...) std::println(__VA_ARGS__)
|
||||
#else
|
||||
#define db
|
||||
#define dbln
|
||||
#endif
|
||||
// }}}
|
||||
|
||||
constexpr u64 MOD = 1e9 + 7;
|
||||
|
||||
void solve() {
|
||||
u64 n;
|
||||
cin >> n;
|
||||
|
||||
u64 six = 6;
|
||||
vec<u64> toss(n + 1, 0);
|
||||
toss[0] = 1;
|
||||
for (u64 i = 1; i <= n; ++i) {
|
||||
for (u64 j = 1; j <= min(six, i); ++j) {
|
||||
toss[i] = (toss[i] + toss[i - j]) % MOD;
|
||||
}
|
||||
}
|
||||
|
||||
cout << toss[n] << '\n';
|
||||
}
|
||||
|
||||
int main() { // {{{
|
||||
cin.tie(nullptr)->sync_with_stdio(false);
|
||||
cin.exceptions(cin.failbit);
|
||||
|
||||
int tc = 1;
|
||||
// cin >> tc;
|
||||
|
||||
for (int t = 0; t < tc; ++t) {
|
||||
solve();
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
// }}}
|
||||
Loading…
Add table
Add a link
Reference in a new issue