F - Problem Makes Problem (LightOJ - 1102)
阿新 • • 發佈:2018-02-14
快速冪 stream 解題思路 ase ble oid iostream 劃分 pow
- 題目大意
求n有順序的劃分為k個數的方案數。
- 解題思路
很顯然是一個組合數,用隔板法, 再用費馬小定理求逆元,再利用快速冪即可求出。
- 代碼
#include<iostream> #define mod 1000000007 using namespace std; const int MAX = 2000000; long long num[MAX]; void zh() { num[0] = 1; for (int i = 1; i <MAX ; i++) num[i] = num[i - 1] * i%mod; } long long powMod(long long a, long long n) { long long ans = 1; for (; n > 0; n >>= 1) { if (n & 1) ans = ans * a%mod; a = a * a%mod; } return ans; } long long Comb(long long n, long long m) { if (n < m) return 0; return num[n] * powMod(num[m] * num[n - m] % mod, mod - 2) % mod; } int main() { int t; long long n, k; zh(); cin >> t; for(int i=1;i<=t;i++) { cin >> n>> k; cout << "Case " << i << ": "; cout << Comb(n+k-1,k-1) << endl; } return 0; }
F - Problem Makes Problem (LightOJ - 1102)