codeforces 938E MaxHistory 组合数学

题意

给出一个长为n的序列,对于这个序列的任意一个排列,求 ∑ f a \sum f_a fa的值。

题解

我们枚举每一个 a i a_i ai然后,计算 a i a_i ai出现了多少次,然后把 a i ∗ n u m a_i*num ainum加入到ans里面。
考虑 a i a_i ai什么时候出现。
a i a_i ai出现的位置前面的 a j a_j aj必须严格小于 a i a_i ai,这样 a i a_i ai才能被取到,而且 a i a_i ai后面一定要有比它大的数,这样才能 a i a_i ai取到。
假设小于 a i a_i ai的数有 m m m个,那么 n u m m = ∑ r = 0 m C m r ∗ r ! ∗ ( n − 1 − r ) ! num_m = \sum_{r = 0}^mC_m^r*r!*(n-1-r)! numm=r=0mCmrr!(n1r)!
整个的公式就是:
∑ i = 1 n a i ∗ n u m i \sum_{i = 1}^na_i*num_i i=1nainumi
但这样暴力求的话时间复杂度是 O ( n 2 ) O(n^2) O(n2),因此必须运用公式变换上述式子,可以证明 n u m i num_i numi可以在 O ( 1 ) O(1) O(1)的时间求出来,具体求解方法见下图。

题解
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define pr(x) cout<<#x<<":"<<x<<endl
const int maxn = 1e6+10;
typedef long long ll;
const ll mod = 1e9+7;
ll a[maxn],fac1[maxn],fac2[maxn];
int n;
int main(){
    scanf("%d",&n);
    for(int i = 0;i < n;++i)
        scanf("%lld",&a[i]);
    sort(a,a+n);
    fac1[0] = 1;
    fac2[n+1] = 1;
    fac2[n] = n;
    for(int i = 1;i <= n;++i)
        fac1[i] = fac1[i-1]*i%mod;
    for(int i = n-1;i >= 1;--i)
        fac2[i] = fac2[i+1]*i%mod;
    ll ans = 0;
    for(int i = 0,j = 1;i < n && a[i] != a[n-1];i = j){
        for(j = i;a[i] == a[j];j++);
        ll num = j-i;
        ans = (ans + num*((a[i]*fac1[n-i-1]%mod)*fac2[n-i+1]))%mod;
    }
    cout<<ans<<endl;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值