Submission #891950

#	Time	Username	Problem	Language	Result	Execution time	Memory
891950		josanneo22	Kangaroo (CEOI16_kangaroo)	C++17	100 / 100	29 ms	39516 KiB

kangaroo

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

const int mod = 1E9 + 7;
const int _N = 2500;
i64 dp[_N][_N];
/*
    dp[i][j] = 已经排了i个数,连通块=j
    放入某个块: dp[i][j]=dp[i-1][j]*j
    新加一个块: dp[i][j]=dp[i-1][j-1]*j
    合并两个块: dp[i][j]=dp[i-1][j+1]*j
    若i==A || i==B, 我们不可以用来合并而且也不能随便加入块:
        dp[i][j] = dp[i-1][j]只能加入最后一个块/最前一个块
        dp[i][j] = dp[i-1][j-1]放上最后一个块/最前一个块
*/
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);

    int N, A, B; cin >> N >> A >> B;
    dp[1][1] = 1;
    for (int i = 2; i <= N; i++) {
        for (int j = 1; j <= N; j++) {
            if (i == A || i == B) {
                dp[i][j] += dp[i - 1][j];
                dp[i][j] += dp[i - 1][j - 1];
                dp[i][j] %= mod;
                continue;
            }
            int cnt = (i > A) + (i > B);
            dp[i][j] += dp[i - 1][j - 1] * (j - cnt);//不能放头，不能放尾
            dp[i][j] %= mod;
            dp[i][j] += dp[i - 1][j + 1] * j;
            dp[i][j] %= mod;
        }
    }
    cout << dp[N][1] << '\n';
}

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