牛客 Chino with Equation(不定方程，组合数)

链接：https://ac.nowcoder.com/acm/problem/23868
来源：牛客网
 

题目描述

Chino的数学很差，因此Cocoa非常担心。今天，Cocoa要教Chino解不定方程。
众所周知，不定方程的解有0个或者若干个。
给出方程:

Cocoa想知道这个不定方程的正整数解和非负整数解各有几个。
题目对Chino来说太难啦，你能帮一帮Chino吗？

输入描述:

两个正整数m, n

输出描述:

题目要求的答案，即正整数解的个数和非负整数解的个数 。由于答案可能会很大，你只需要输出答案 mod(109 + 7) 即可。

 

示例1

输入

复制

4 7

输出

复制

20 120

不定方程可以看成是n个1 进行组合，例如m=7，n=4时，1111    111   ， 1  111111可以算成其中的两种情况

所以可以抽象转化成组合数问题，正整数的解为C(n-1,m-1),非负整数的解为C(n+m-1,m-1)，在计算时，因为取模，

所以分母要用逆元来求.

 

#include<iostream>
using namespace std;
const int mod = 1e9 + 7;
typedef long long ll;
ll jc(ll n)   // 算阶乘
{
	ll res = 1;
	for (int i = 1; i <= n; i++)
	{
		res = res*i%mod;
	}
	return res;
}
ll fpow(ll a, ll b)
{
	ll res = 1;
	while (b)
	{
		if (b & 1)
			res = res*a%mod;
		b >>= 1;
		a = a*a%mod;
	}
	return res%mod;
}
ll f(ll n, ll m){
	ll ans = 1;  
	ans = ans*jc(n) % mod;   //分子
	ans = ans*fpow(jc(m), mod - 2) % mod;   //求分母逆元
	ans = ans*fpow(jc(n - m), mod - 2) % mod;
	return ans%mod;
}
int main(){
	ll m, n;
	cin >> m >> n;
	cout << f(n - 1, m - 1) << " " << f(n + m - 1, m - 1); 
	return 0;
}