题目描述
天天开始弹奏钢琴了。他的钢琴由 N个琴键组成,每个键有一个权值xi,当天天开始弹琴时,他同时按K个不同的键。但是这个钢琴十分古怪,在他同时按了K个键之后,我们只能听到具有最大权值的那个键。现在天天准备试试这个钢琴K个键的所有组合,他想要知道将要听到琴声的权值和。因为结果会很大,所以输出mod 1000000007之后的结果。
输入
第一行两个整数N,K。(1<= N <=100000,1<= K <=50). 下一行由 N 个整数xi组成(0<= xi <=1000000000).
输出
一个整数 表示权值和sum%P的结果
样例输入 Copy
5 3
2 4 2 3 4
样例输出 Copy
39
可以将n个琴建的权值进行排序,然后定一取前面的k-1个,所以就转化成了组合数问题
sum=an*C(k-1,n-1);
#include<iostream>
#include<map>
#include<algorithm>
using namespace std;
typedef long long ll;
const int mod = 1000000007;
ll a[100010];
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;
}
ll jc(ll a) //阶乘
{
ll res = 1;
for (int i = 1; i <= a; i++)
res = res*i%mod;
return res;
}
ll c(ll a,ll b) //算方案
{
ll res = 1;
res = res*jc(b)%mod;
res = res*fpow(jc(a), mod - 2) % mod; //求逆元
res = res*fpow(jc(b - a), mod - 2) % mod;
return res;
}
int main(){
int n, k;
ll ans = 0;
cin >> n >> k;
for (int i = 1; i <= n; i++)
cin >> a[i];
sort(a + 1, a + 1 + n);
for (int i = n; i >= 1; i--)
{
if (i-1 == k - 2) //当不能再选取k-1个时
break;
else
{
ans = ans + c(k - 1, i -1)*a[i]%mod;
}
}
cout << ans % mod << endl;
return 0;
}