#include <bits/stdc++.h>
#define ll long long int
#define pb push_back
#define st first
#define nd second
#define pii pair<int,int>
#define mp make_pair
#define pll pair<long long,long long>
using namespace std;
ll n,m;
vector<vector<ll> > gora,dol;
int mod;
int main()
{
ios_base::sync_with_stdio(0); cin.tie(0);
cin>>n>>m>>mod;
gora.resize(n + 1);
dol.resize(n + 1);
for(int i=1;i<=n;i++)
{
gora[i].resize(m + 1,0);
dol[i].resize(m + 1,0);
}
vector<ll> prefix_dolkow;
for(int i=1;i<=m;i++)
{
gora[1][i] = (m - i + 1);
dol[1][i] = i;
}
prefix_dolkow.resize(m + 1,0);
prefix_dolkow[0] = 0;
for(int i=1;i<=m;i++)
{
prefix_dolkow[i] = prefix_dolkow[i - 1] + dol[1][i];
prefix_dolkow[i] %= mod;
}
int alles = prefix_dolkow[m];
vector<ll> sufix_gorek;
sufix_gorek.resize(m + 2,0);
sufix_gorek[m + 1] = 0;
for(int i=m;i>=0;i--)
{
sufix_gorek[i] = sufix_gorek[i + 1] + gora[1][i];
sufix_gorek[i] %= mod;
}
for(int i=2;i<=n;i++)
{
ll sum = 0;
for(int j=m;j>=1;j--)
{
ll akt = alles - prefix_dolkow[j - 1];
if(akt < 0) akt += mod;
sum += sufix_gorek[j + 1];
sum %= mod;
gora[i][j] = akt * (m - j + 1);
gora[i][j] %= mod;
gora[i][j] -= sum;
if(gora[i][j] < 0) gora[i][j] += mod;
}
for(int j=1;j<=m;j++)
{
dol[i][j] = gora[i][m + 1 - j];
}
for(int j=1;j<=m;j++)
{
prefix_dolkow[j] = prefix_dolkow[j - 1] + dol[i][j];
prefix_dolkow[j] %= mod;
}
for(int j=m;j>=0;j--)
{
sufix_gorek[j] = sufix_gorek[j + 1] + gora[i][j];
sufix_gorek[j] %= mod;
}
alles = 0;
for(int j=1;j<=m;j++)
{
alles += gora[i][j];
alles %= mod;
}
}
ll ans = 0;
for(int i=1;i<=m;i++)
{
ans += gora[n][i];
ans %= mod;
}
cout<<ans;
return 0;
}
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 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 | #include <bits/stdc++.h> #define ll long long int #define pb push_back #define st first #define nd second #define pii pair<int,int> #define mp make_pair #define pll pair<long long,long long> using namespace std; ll n,m; vector<vector<ll> > gora,dol; int mod; int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin>>n>>m>>mod; gora.resize(n + 1); dol.resize(n + 1); for(int i=1;i<=n;i++) { gora[i].resize(m + 1,0); dol[i].resize(m + 1,0); } vector<ll> prefix_dolkow; for(int i=1;i<=m;i++) { gora[1][i] = (m - i + 1); dol[1][i] = i; } prefix_dolkow.resize(m + 1,0); prefix_dolkow[0] = 0; for(int i=1;i<=m;i++) { prefix_dolkow[i] = prefix_dolkow[i - 1] + dol[1][i]; prefix_dolkow[i] %= mod; } int alles = prefix_dolkow[m]; vector<ll> sufix_gorek; sufix_gorek.resize(m + 2,0); sufix_gorek[m + 1] = 0; for(int i=m;i>=0;i--) { sufix_gorek[i] = sufix_gorek[i + 1] + gora[1][i]; sufix_gorek[i] %= mod; } for(int i=2;i<=n;i++) { ll sum = 0; for(int j=m;j>=1;j--) { ll akt = alles - prefix_dolkow[j - 1]; if(akt < 0) akt += mod; sum += sufix_gorek[j + 1]; sum %= mod; gora[i][j] = akt * (m - j + 1); gora[i][j] %= mod; gora[i][j] -= sum; if(gora[i][j] < 0) gora[i][j] += mod; } for(int j=1;j<=m;j++) { dol[i][j] = gora[i][m + 1 - j]; } for(int j=1;j<=m;j++) { prefix_dolkow[j] = prefix_dolkow[j - 1] + dol[i][j]; prefix_dolkow[j] %= mod; } for(int j=m;j>=0;j--) { sufix_gorek[j] = sufix_gorek[j + 1] + gora[i][j]; sufix_gorek[j] %= mod; } alles = 0; for(int j=1;j<=m;j++) { alles += gora[i][j]; alles %= mod; } } ll ans = 0; for(int i=1;i<=m;i++) { ans += gora[n][i]; ans %= mod; } cout<<ans; return 0; } |
English