牛客网普及组第二次比赛C涨薪

涨薪

在这里插入图片描述
这道题初看比较绕,但只要想通一个点就比较简单:给哪些员工涨薪

很容易想到用贪心的思想,将员工的初始工资从大到小排序,每次给第1到第x个员工涨3倍工资,给第x+1到第x+y个员工涨2倍工资。

(方案有很多种,例如每次换着给不同的员工涨薪来保留更多员工,但显然不是最优的,有兴趣的可以自己证明)

那么显然,前1到x个员工会涨m次3倍工资,前x+1到x+y个员工会涨m次2倍工资,而后面的员工则不需要考虑。(除非年数小于2,我就没考虑这种情况,呜呜~~)这就可以用快速幂来优化了。

(快速幂还不会的,自己去搜博客吧)

代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#define ll long long 
using namespace std;
const ll mod=1e9+7;
ll n,m,x,y;
ll ans,a[100005];
ll ksm(ll k)
{
    
    
	ll p=m,s=1;
	while(p)
	{
    
    
		if(p&1)
		s=(s*k)%mod;
		k=(k*k)%mod;
		p>>=1;
	}
	return s%mod;
}
int main()
{
    
    
	scanf("%d%d%d%d",&n,&m,&x,&y);
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	sort(a+1,a+n+1,greater<int>());
	for(int i=1;i<=x;i++)
	ans=(ans+(a[i]*ksm(3))%mod)%mod;
	for(int i=x+1;i<=x+y;i++)
	ans=(ans+(a[i]*ksm(2))%mod)%mod;
	cout<<ans;
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45383207/article/details/109275714