#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define L long long
#define MP make_pair
#define REP(i, n) for(int i = 0; i < n; ++i)
#define REPR(i, n) for(int i = n - 1; i >= 0; --i)
#define FOR(i, a, b) for(int i = a; i < b; ++i)
#define FORR(i, a, b) for(int i = b - 1; i >= a; --i)
#define EB emplace_back
#define ST first
#define ND second
#define S size()
#define RS resize
using VI = vector<int>;
using VVI = vector<VI>;
using PI = pair<int, int>;
using VPI = vector<PI>;
using VVPI = vector<VPI>;
using VL = vector<L>;
using VVL = vector<VL>;
using VB = vector<bool>;
using VC = vector<char>;
L p;
void mod(L &a)
{
if (a >= p)
a -= p;
}
void mod2(L &a)
{
if (a >= p)
a %= p;
}
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int n, m;
cin >> n >> m >> p;
L sp = p * p;
VVL pref(n, VL(m + 1)), pref2(n, VL(m + 1));
FOR(i, 1, m + 1)
{
pref[0][i] = pref[0][i - 1] + (L)i;;
mod2(pref[0][i]);
pref2[0][i] = pref2[0][i - 1] + pref[0][i];
mod(pref2[0][i]);
}
FOR(i, 1, n)
FOR(d, 1, m + 1)
{
pref[i][d] = pref[i][d - 1] + (L)d * (pref[i - 1][m] - pref[i - 1][m - d]) - pref2[i - 1][d - 1] + sp;
mod2(pref[i][d]);
pref2[i][d] = pref2[i][d - 1] + pref[i][d];
mod(pref2[i][d]);
}
cout << pref[n - 1][m];
}
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 | #include <bits/stdc++.h> using namespace std; #define endl '\n' #define L long long #define MP make_pair #define REP(i, n) for(int i = 0; i < n; ++i) #define REPR(i, n) for(int i = n - 1; i >= 0; --i) #define FOR(i, a, b) for(int i = a; i < b; ++i) #define FORR(i, a, b) for(int i = b - 1; i >= a; --i) #define EB emplace_back #define ST first #define ND second #define S size() #define RS resize using VI = vector<int>; using VVI = vector<VI>; using PI = pair<int, int>; using VPI = vector<PI>; using VVPI = vector<VPI>; using VL = vector<L>; using VVL = vector<VL>; using VB = vector<bool>; using VC = vector<char>; L p; void mod(L &a) { if (a >= p) a -= p; } void mod2(L &a) { if (a >= p) a %= p; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, m; cin >> n >> m >> p; L sp = p * p; VVL pref(n, VL(m + 1)), pref2(n, VL(m + 1)); FOR(i, 1, m + 1) { pref[0][i] = pref[0][i - 1] + (L)i;; mod2(pref[0][i]); pref2[0][i] = pref2[0][i - 1] + pref[0][i]; mod(pref2[0][i]); } FOR(i, 1, n) FOR(d, 1, m + 1) { pref[i][d] = pref[i][d - 1] + (L)d * (pref[i - 1][m] - pref[i - 1][m - d]) - pref2[i - 1][d - 1] + sp; mod2(pref[i][d]); pref2[i][d] = pref2[i][d - 1] + pref[i][d]; mod(pref2[i][d]); } cout << pref[n - 1][m]; } |
English