#include <cstdio>
#include <vector>
int main() {
int N,M;
long long int P;
scanf("%d %d %lld", &N, &M, &P);
std::vector<long long int> psum, csum, psumsum, csumsum;
std::vector<std::vector<long long int> > dp;
dp.resize(N+1);
dp[0].resize(M+1);
psum.resize(M+1);
psumsum.resize(M+1);
csum.resize(M+1);
csumsum.resize(M+1);
psum[M] = 1;
psumsum[M] = 1;
for (int i=1; i<=N; ++i) {
dp[i].resize(M+1);
for (int j=1; j<=M; ++j) {
dp[i][j] = (j*psum[M] + j*P - j*psum[M-j]) % P;
dp[i][j] = (dp[i][j] + P - psumsum[j-1]) % P;
csum[j] = (csum[j-1] + dp[i][j]) % P;
csumsum[j] = (csum[j] + csumsum[j-1]) %P;
}
std::swap(psum, csum);
std::swap(psumsum, csumsum);
}
printf("%lld\n",psum[M]);
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 | #include <cstdio> #include <vector> int main() { int N,M; long long int P; scanf("%d %d %lld", &N, &M, &P); std::vector<long long int> psum, csum, psumsum, csumsum; std::vector<std::vector<long long int> > dp; dp.resize(N+1); dp[0].resize(M+1); psum.resize(M+1); psumsum.resize(M+1); csum.resize(M+1); csumsum.resize(M+1); psum[M] = 1; psumsum[M] = 1; for (int i=1; i<=N; ++i) { dp[i].resize(M+1); for (int j=1; j<=M; ++j) { dp[i][j] = (j*psum[M] + j*P - j*psum[M-j]) % P; dp[i][j] = (dp[i][j] + P - psumsum[j-1]) % P; csum[j] = (csum[j-1] + dp[i][j]) % P; csumsum[j] = (csum[j] + csumsum[j-1]) %P; } std::swap(psum, csum); std::swap(psumsum, csumsum); } printf("%lld\n",psum[M]); return 0; } |
English