打怪兽

lyk在玩一个叫做“打怪兽”的游戏。
游戏的规则是这样的。
lyk一开始会有一个初始的能量值。每次遇到一个怪兽,若lyk的能量值>=怪兽的能量值,那么怪兽将会被打败,lyk的能量值增加1,否则lyk死亡,游戏结束。
若怪兽全部打完,游戏也将会结束。
共有n个怪兽,由于lyk比较弱,它一开始只有0点能量值。
n个怪兽排列随机,也就是说共有n!种可能,lyk想知道结束时它能量值的期望。
由于小数点比较麻烦,所以你只需要输出期望*n!关于1000000007取模后的值就可以了!

例如有两个怪兽,能量值分别为{0,1},那么答案为2,因为游戏结束时有两种可能,lyk的能量值分别为0和2。期望为1,1*2!=2,所以答案为2。
Input
第一行一个数n(1<=n<=100000)。
接下来一行n个数ai表示怪兽的能量(0<=ai<n)。
Output
一行表示答案
Input示例
2
0 1
Output示例
2
#include <iostream>  
#include <cstring>  
#include <algorithm>  
using namespace std;  
      
typedef long long int ll;
const int MAXN = 100005;  
const ll MOD = 1000000007;  
      
int n, energy, rem;  
int input[MAXN];
int pos[MAXN];
ll factorial[MAXN];  
ll result, way;  
      
int main() 
{  
    factorial[0] = 1;  
    for (int i = 1; i < MAXN; ++i) 
	{  
        factorial[i] = (factorial[i - 1] * i) % MOD;  
    }

	cin >> n;
    for (int i = 0; i < n; ++i) 
	{  
		cin >> input[i];
    }  
    sort(input, input + n); 

    input[n] = 0x3f3f3f3f;  
    for (int i = 0, j = 0; i <= n; ++i) 
	{  
        while (i >= input[j]) 
	    {  
            ++j;  
        }  
        pos[i] = j - 1;  
    }  
    result = energy = 0;  
    way = 1;  
    while (pos[energy] >= energy) 
	{  
        rem = n - pos[energy] - 1;  
        result = (result + ((((way * rem) % MOD) * factorial[n - energy - 1]) % MOD) * energy) % MOD;  
        way = (way * (pos[energy] - energy + 1)) % MOD;  
        ++energy;  
    }  
    if (energy == n) 
	{  
        result = (result + way * energy) % MOD;  
    }  
    else 
	{  
        rem = n - energy;  
        result = (result + ((((way * rem)%MOD) * factorial[rem - 1]) % MOD) * energy) % MOD;  
    }  
	cout << result << endl;
    
	return 0;  
}  


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值