链接: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;
}