組合數取模(楊輝三角+Lucas定理+模合數)
阿新 • • 發佈:2018-12-31
/* (1) 1 <= m <= n <= 1000 和 1 <= p <= 10^9 ( p可以是任何數 ) 這個問題比較簡單,組合數的計算可以靠 楊輝三角 ,那麼由於和的範圍小,直接兩層迴圈即可。 */ long long C[maxn][maxn]; void Comb(int n, int m, int p){ memset(C, 0, sizeof(C)); C[0][0] = 1; for(int i = 0; i <= n; i++){ C[i][0] = C[i][i] = 1; for(int j = 1; j < i; j++) C[i][j] = (C[i-1][j-1] + C[i-1][j]) % p; } }
/* 1 <= m <= n <= 10^18 和 2 <= p <= 10^5 (p 是素數) Lucas定理 */ long long mod_pow(long long n, long long k, long long p) { if (k == 0) return 1; if (k == 1) return n; long long ans = mod_pow(n * n % p, k>>1, p); if (k&1) ans = ans * n % p; return ans; } long long Comb(long long n, long long m, long long p) { if (m > n) return 0; m = min(m, n - m); long long zi = 1, mu = 1; for (long long i = 0; i < m; i++) { zi = zi * (n - i) % p; mu = mu * (i + 1) % p; } return zi * mod_pow(mu, p - 2, p) % p; } long long Lucas(long long n, long long m, long long p) { if (m == 0) return 1; return Comb(n % p, m % p, p) * Lucas(n / p, m / p, p) % p; } /* 非遞迴的快速冪取模 long long mod_pow(long long x, long long k, long long MOD) { long long ans = 1; while (k) { if (k&1) ans = ans * x % MOD; x = x * x % MOD; k >>= 1; } return ans; } */
/* 1 <= m,n <= 10^6 和 p <= 10^9 (p不一定是素數) */ #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const long long maxn = 500005; int prime[maxn]; // 第i個素數(從0開始計數) bool is_prime[maxn+1]; // is_prime[i]為true表示i是素數 int getprime(int n){ //返回值是n以內素數的個數 int p = 0; for(int i = 0; i <= n; i++) is_prime[i] = true; is_prime[0] = is_prime[1] = false; for(int i = 2; i <= n; i++){ if(is_prime[i]){ prime[p++] = i; for(int j = 2*i; j <= n; j += i) is_prime[j] = false; } } return p; } long long count(long long x, long long y){ long long ret = 0; while(x / y){ ret += x/y; x /= y; } return ret; } long long mod_pow(long long n, long long k, long long p) { if (k == 0) return 1; if (k == 1) return n; long long ans = mod_pow(n * n % p, k>>1, p); if (k&1) ans = ans * n % p; return ans; } long long solve(long long n, long long m, long long p){ long long ans = 1; for(long long i = 0; prime[i] <= n; i++){ long long cnt = count(n, prime[i]) - count(m, prime[i]) - count(n-m, prime[i]); ans = ans * mod_pow(prime[i], cnt, p) % p; if(ans == 0) break; } return ans; } // 打素數表的時候要注意, 最大的素數 要 大於 輸入的n int main(){ long long n, m, p; getprime(maxn); while(scanf("%I64d%I64d%I64d",&n, &m, &p)!=EOF){ printf("%I64d\n",solve(n, m, p)); } return 0; }