Submission #493910

#	Time	Username	Problem	Language	Result	Execution time	Memory
493910		khoabright	Kangaroo (CEOI16_kangaroo)	C++17	100 / 100	35 ms	31608 KiB

kangaroo

#include <bits/stdc++.h>
using namespace std;

#define int long long 
#define ll long long
#define pii pair<int, int>
#define ff first
#define ss second
#define pb push_back
#define rep(i, a, b) for (int i = (int)a; i <= (int)b; ++i)
#define rep1(i, a, b) for (int i = (int)a; i >= (int)b; --i)
#define all(x) x.begin(), x.end()

const int N = 2e3 + 5;
const int mod = 1e9 + 7;

int dp[N][N];

main() {
    int n, a, b; cin >> n >> a >> b;
    dp[1][1] = 1;
    rep(i, 1, n - 1) rep(j, 1, n) {
        if (i + 1 == a || i + 1 == b) {
            (dp[i + 1][j] += dp[i][j]) %= mod;
            (dp[i + 1][j + 1] += dp[i][j]) %= mod;
        }

        else {
            // merge two components
            (dp[i + 1][j - 1] += dp[i][j] * (j - 1)) %= mod;

            // create a new component
            (dp[i + 1][j + 1] += (dp[i][j] * (j + 1 - (i + 1 > a) - (i + 1 > b)))) %= mod;
        }
    }

    cout << dp[n][1];
}

Compilation message (stderr)

kangaroo.cpp:19:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   19 | main() {
      | ^~~~

• Subtask 1: 6 / 6
• Subtask 2: 30 / 30
• Subtask 3: 15 / 15
• Subtask 4: 49 / 49