DMOJ · goodset

Good Sets

This C++ solution uses dynamic programming for DMOJ goodset Good Sets. Read the reasoning, inspect the code, or try your own test case below.

goodsetDynamic programmingC++21 lines
Solution102of 248
Open official problem ↗ Download C++ file ↓ Search the library → Open full Code Lab ↗ Report an issue ↗

Approach

Dynamic programming

Good Set spacing rule and subset count match the dynamic program.

Dynamic programming

Problem and code

Useful links.

Written by benbenyaojifen. Try the problem first, then compare your approach with the code.

Open official problem ↗View exact source file ↗
Implementation

good_set.cpp

C++

    #include <bits/stdc++.h>
     
    using namespace std;
    const long long MOD = 1e9 + 7;
     
    int main() {
        ios::sync_with_stdio(0);
        cin.tie(0);
        long long n, k;
        cin >> n >> k;
        vector<long long> dp(n + 1);
        dp[0] = 1; // empty set
        for (long long i = 1; i <= n; i++) {
            long long skip = dp[i - 1];
            long long take = (i >= k ? dp[i - k] : 1);
            dp[i] = (skip + take) % MOD;
        }
        cout << dp[n] % MOD << '\n';
        return 0;
    }
     
        

Did this explanation save you time? I'm a Grade 11 student building this free library to make difficult algorithms easier to understand.

Buy me a coffee ↗Keep studying →

Test this problem

Run your code here.

Paste your code, run a test case, compare the output, or trace selected values.

Full trace, comparison & stress testing ↗
StatusReady
Output
No run yet.
Diagnostics
No diagnostics yet.

Each run is isolated and has strict limits. Passing one test does not guarantee the judge will accept the solution.