#include <bits/stdc++.h> using namespace std; int mod = 2137; int add(int a, int b) { a += b; return a < mod ? a : a - mod; } int sub(int a, int b) { a -= b; return a < 0 ? a + mod : a; } int mul(int a, int b) { return int((long long) a * b % mod); } int pot(int a, int b) { int r = 1; for(; b; b >>= 1, a = mul(a, a)) if(b & 1) r = mul(r, a); return r; } int n, m; vector<vector<int>> dp, s, p; int main() { scanf("%d%d%d", &n, &m, &mod); dp.resize(n, vector<int>(m)); s.resize(n, vector<int>(m)); p.resize(n, vector<int>(m)); for(int i = 0; i < m; i++) { dp[0][i] = i + 1; } for(int i = 0; i < m; i++) { p[0][i] = dp[0][i]; s[0][i] = dp[0][m - 1 - i]; } for(int i = 1; i < m; i++) { p[0][i] = add(p[0][i], p[0][i - 1]); } for(int i = m - 2; i >= 0; i--) { s[0][i] = add(s[0][i], s[0][i + 1]); } for(int i = 1; i < n; i++) { int sum = 0; for(int j = 0; j < m; j++) { sum = add(sum, dp[i - 1][j]); } int pref_sum = 0; for(int j = 0; j < m; j++) { dp[i][j] = sub(mul(sum, add(j, 1)), pref_sum); if(j < m - 1) dp[i][j] = sub(dp[i][j], mul(s[i - 1][j + 1], add(j, 1))); pref_sum = add(pref_sum, p[i - 1][j]); } for(int j = 0; j < m; j++) { s[i][j] = dp[i][m - 1 - j]; p[i][j] = dp[i][j]; } for(int j = m - 2; j >= 0; j--) { s[i][j] = add(s[i][j], s[i][j + 1]); } for(int j = 1; j < m; j++) { p[i][j] = add(p[i][j], p[i][j - 1]); } } int ans = 0; for(int i = 0; i < m; i++) { ans = add(ans, dp[n - 1][i]); } printf("%d\n", ans); }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 | #include <bits/stdc++.h> using namespace std; int mod = 2137; int add(int a, int b) { a += b; return a < mod ? a : a - mod; } int sub(int a, int b) { a -= b; return a < 0 ? a + mod : a; } int mul(int a, int b) { return int((long long) a * b % mod); } int pot(int a, int b) { int r = 1; for(; b; b >>= 1, a = mul(a, a)) if(b & 1) r = mul(r, a); return r; } int n, m; vector<vector<int>> dp, s, p; int main() { scanf("%d%d%d", &n, &m, &mod); dp.resize(n, vector<int>(m)); s.resize(n, vector<int>(m)); p.resize(n, vector<int>(m)); for(int i = 0; i < m; i++) { dp[0][i] = i + 1; } for(int i = 0; i < m; i++) { p[0][i] = dp[0][i]; s[0][i] = dp[0][m - 1 - i]; } for(int i = 1; i < m; i++) { p[0][i] = add(p[0][i], p[0][i - 1]); } for(int i = m - 2; i >= 0; i--) { s[0][i] = add(s[0][i], s[0][i + 1]); } for(int i = 1; i < n; i++) { int sum = 0; for(int j = 0; j < m; j++) { sum = add(sum, dp[i - 1][j]); } int pref_sum = 0; for(int j = 0; j < m; j++) { dp[i][j] = sub(mul(sum, add(j, 1)), pref_sum); if(j < m - 1) dp[i][j] = sub(dp[i][j], mul(s[i - 1][j + 1], add(j, 1))); pref_sum = add(pref_sum, p[i - 1][j]); } for(int j = 0; j < m; j++) { s[i][j] = dp[i][m - 1 - j]; p[i][j] = dp[i][j]; } for(int j = m - 2; j >= 0; j--) { s[i][j] = add(s[i][j], s[i][j + 1]); } for(int j = 1; j < m; j++) { p[i][j] = add(p[i][j], p[i][j - 1]); } } int ans = 0; for(int i = 0; i < m; i++) { ans = add(ans, dp[n - 1][i]); } printf("%d\n", ans); } |