天天的钢琴（数论）

题目描述

天天开始弹奏钢琴了。他的钢琴由 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;
}