Submission #26960942

---

Source Code Expand

#ifndef BZ
#pragma GCC optimize "-O3"
#endif
#include <bits/stdc++.h>

#define ALL(v) (v).begin(), (v).end()
#define rep(i, l, r) for (int i = (l); i < (r); ++i)

using ll = long long;
using ld = long double;
using ull = unsigned long long;

using namespace std;


ll MOD;
/*
ll pw(ll a, ll b) {
	ll ans = 1; while (b) {
		while (!(b & 1)) b >>= 1, a = (a * a) % MOD;
		ans = (ans * a) % MOD, --b;
	} return ans;
}
*/

const int N = 5100;

int n, m;

ll dp[N][N][2];

void add(ll& a, ll b) {
	a = (a + b) % MOD;
}

int main() {
	ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	cout.setf(ios::fixed), cout.precision(20);
	cin >> n >> m >> MOD;
	dp[0][0][0] = 1;
	dp[0][0][1] = 1;
	for (int i = 0; i < n; ++i) {
		for (int j = 0; j <= i; ++j) {
			if (i + 2 <= n) {
				add(dp[i + 2][j + 1][0], dp[i][j][0]); // 00
				add(dp[i + 2][j + 1][1], dp[i][j][0]); // 00
				add(dp[i + 2][j][1], dp[i][j][1]); // 00
				add(dp[i + 2][j + 1][0], dp[i][j][0]); // 01
				add(dp[i + 2][j + 1][1], dp[i][j][1]); // 01
			} else {
				add(dp[i + 1][j][0], dp[i][j][0]); // 0$
				add(dp[i + 1][j][1], dp[i][j][1]); // 0$
			}
			add(dp[i + 1][j + 1][0], dp[i][j][0]); // 1
			add(dp[i + 1][j + 1][1], dp[i][j][1]); // 1
		}
	}

	ll ans = 0;
	for (int k = 3; k <= m; ++k) {
		add(ans, dp[n][k][1]);
	}

	add(ans, 1); // 2 2 2 2

	if (m == n - 1 && m > 2) {
		add(ans, 1);
	}

	cout << ans << "\n";

	return 0;
}

Submission Info

Judge Result