ftx456789

发布于 2018-07-25 11:07

牛客网暑期ACM多校训练营（第一场） F

先思考n个数字相同的情况，即

$\sum_{x_1}^{i}\sum_{x_2}^{i}\sum_{x_3}^{i}\cdots \sum_{x_n}^{i}max(x_1,x_2,x_3,\cdots,x_n)$

如果我们知道了 $1\sim i$ 中每个数字作为最大值出现的个数的话我们最后的答案就是
$ans=\sum_{j=1}^{i}j*num[j]$
对每个数最为最大值的数量打表会发现每个数为最大值的数量刚好与自然数有关,有这样的关系

$\begin{matrix} 1 &1^n \\ 2 &2^n-1^n \\ 3 &3^n-(2^n-1)-1^n \\ \vdots & \vdots \\ i & i^n-((i-1)^n-(i-2)^n-\cdots -1^n)-((i-2)^n-\cdots -1^n)-\cdots -1^n \end{matrix}$

我们设num[i]为i为最大值出现的数量，那么我们再记作S[i]为出现的数的数量的和

即有 $S[i]=num[1]+num[2]+\cdots+num[i]=i^n$

那么我们再看看原来的ans就可以变为

$ans=\sum_{j=1}^{i}j*num[j]=i*S[i]-\sum_{j=1}^{i-1}S[j]=i*i^n-(1^n+2^n+3^n+\cdots+(i-1)^n)$

再回到原来的问题上，题目中给的n个是不一定相同的数，我们其实会发现答案和给的顺序没有关系，所以我们把数从小到大排序
$\begin{matrix} a_1 &a_2 &a_3 & \cdots &a_n \end{matrix}$
同样的我们也可以发现一个规律

$\begin{matrix} 1 &1^n \\ 2 &2^n-1^n \\ 3 &3^n-(2^n-1)-1^n \\ \vdots & \vdots \\ a_1 & a_1^n-((a_1-1)^n-(a_1-2)^n-\cdots -1^n)-((a_1-2)^n-\cdots -1^n)-\cdots -1^n \\ a_1+1 & a_1*(a_1+1)^{n-1}-S[a_1]\\ \vdots & \vdots\\ a_2 & a_1*a_2^{n-1}-S[a_2-1]\\ \vdots & \vdots\\ a_3 & a_1*a_2*a_3^{n-2}-S[a_3-1]\\ \vdots & \vdots\\ a_n & a_1*a_2*a_3*\cdots*a_n-S[a_n-1] \end{matrix}$

同样的有 $S[a_i]=a_1*a_2*\cdots*a_i^{n-i+1}$ ，最后的答案和我们一开始的推出的形式是一样的即

$ans=\sum_{j=1}^{a_n}j*num[j]=a_n*S[a_n]-\sum_{j=1}^{a_n-1}S[j]$

现在要解决的问题就是如何快速求出S[j]

我们可以分段求和处理 $S[1]\sim S[a_1]$ ， $S[a_1+1]\sim S[a_2]$ ， $\cdots$ ， $S[a_{n-1}+1]\sim S[a_n]$

举个例子就可以理解了,如求解 $S[a_i+1]\sim S[a_{i+1}]$

$S[a_i+1]=a_1*a_2*\cdots*a_i*(a_{i}+1)^{n-i},S[a_i+2]=a_1*a_2*\cdots*a_i*(a_i+2)^{n-i}$

$S[a_i+3]=a_1*a_2*\cdots*a_i*(a_i+3)^{n-i},\cdots,S[a_{i+1}]=a_1*a_2*\cdots*a_i*a_{i+1}^{n-i}$

那么就有 $\sum S[a_i+1]\sim S[a_{i+1}]$

$\sum S[a_i+1]\sim S[a_{i+1}]=a_1*a_2*\cdots*a_i*((a_{i}+1)^{n-i}+(a_{i}+2)^{n-i}+\cdots+a_{i+1}^{n-i})=a_1*a_2*\cdots*a_i\sum_{j=a_i+1}^{a_{i+1}}j^{n-i}$

$\sum S[a_i]\sim S[a_{i+1}]$ 可以用两个自然数幂次和相减来得到

$\sum S[a_i+1]\sim S[a_{i+1}]=a_1*a_2*\cdots*a_i(\sum_{j=1}^{a_{i+1}}j^{n-i}-\sum_{j=1}^{a_{i}}j^{n-i})$

剩下的就是如何快速求出自然数次幂和了，有一个基于伯努利数的自然数求和公式

具体的实现可以参考https://blog.csdn.net/acdreamers/article/details/38929067

给出求解自然数和的模版

typedef long long LL;
const LL mod = 1000000007;
const int N = 1005;

LL C[N][N];
LL B[N],Inv[N];
LL Tmp[N];
LL n;
void Init()
{
    //预处理组合数
    for(int i=0; i<N; i++)
    {
        C[i][0] = C[i][i] = 1;
        if(i == 0) continue;
        for(int j=1; j<i; j++)
            C[i][j] = (C[i-1][j] % mod + C[i-1][j-1] % mod) % mod;
    }
    //预处理逆元
    Inv[1] = 1;
    for(int i=2; i<N; i++)
        Inv[i] = (mod - mod / i) * Inv[mod % i] % mod;
    //预处理伯努利数
    B[0] = 1;
    for(int i=1; i<N; i++)
    {
        LL ans = 0;
        if(i == N - 1) break;
        for(int j=0; j<i; j++)
        {
            ans += C[i+1][j] * B[j];
            ans %= mod;
        }
        ans *= -Inv[i+1];
        ans = (ans % mod + mod) % mod;
        B[i] = ans;
    }
}

LL sum(int n,int k)
{
    Tmp[0] = 1;
    for(int i=1; i<N; i++)
        Tmp[i] = Tmp[i-1] * (n + 1) % mod;
    LL ans = Inv[k+1];
    LL sum = 0;
    for(int i=1; i<=k+1; i++)
    {
        sum += C[k+1][i] * Tmp[i] % mod * B[k+1-i] % mod;
        sum %= mod;
    }
    ans *= sum;
    ans %= mod;
    return ans;
}

也就有代码

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
using namespace std;
typedef long long LL;
const LL mod = 1000000007;
const int N = 1005;

LL C[N][N];
LL B[N],Inv[N];
LL Tmp[N];
LL n;

void Init()
{
    //预处理组合数
    for(int i=0; i<N; i++)
    {
        C[i][0] = C[i][i] = 1;
        if(i == 0) continue;
        for(int j=1; j<i; j++)
            C[i][j] = (C[i-1][j] % mod + C[i-1][j-1] % mod) % mod;
    }
    //预处理逆元
    Inv[1] = 1;
    for(int i=2; i<N; i++)
        Inv[i] = (mod - mod / i) * Inv[mod % i] % mod;
    //预处理伯努利数
    B[0] = 1;
    for(int i=1; i<N; i++)
    {
        LL ans = 0;
        if(i == N - 1) break;
        for(int j=0; j<i; j++)
        {
            ans += C[i+1][j] * B[j];
            ans %= mod;
        }
        ans *= -Inv[i+1];
        ans = (ans % mod + mod) % mod;
        B[i] = ans;
    }
}

LL sum(int n,int k)
{
    Tmp[0] = 1;
    for(int i=1; i<N; i++)
        Tmp[i] = Tmp[i-1] * (n + 1) % mod;
    LL ans = Inv[k+1];
    LL sum = 0;
    for(int i=1; i<=k+1; i++)
    {
        sum += C[k+1][i] * Tmp[i] % mod * B[k+1-i] % mod;
        sum %= mod;
    }
    ans *= sum;
    ans %= mod;
    return ans;
}
long long a[1005];
int main()
{
    Init();
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
            scanf("%lld",&a[i]);
        if(n==1)
        {
            printf("%lld\n",sum(a[0],1));
            continue;
        }
        sort(a,a+n);
        long long ans=a[n-1],s=1;
        for(int i=0;i<n;i++)
            ans=ans*a[i]%mod;
        for(int i=0;i<n;i++)
        {
            if(i==0)
            ans=(ans-s*(sum(a[i],n-i)))%mod;
            else if(a[i]!=a[n-1])
            {
                ans=(ans-s*(sum(a[i],n-i)-sum(a[i-1],n-i)))%mod;
            }
            else if(i==n-1&&a[i-1]!=a[n-1])
            {
                ans=(ans-s*(sum(a[i]-1,n-i)-sum(a[i-1],n-i)))%mod;
            }
             s=s*a[i]%mod;
        }
        printf("%lld\n",(ans+mod)%mod);
    }
    return 0;
}

2024最新求职资料大礼包领取

大家都在关注

校招日程表笔试日历 ai模拟面试面试宝典剑指offer 知识点专项练习

已采纳

采纳

精彩回帖

精彩

全部评论

(0) 回帖

加载中...

话题同步到我的动态回帖

牛客网暑期ACM多校训练营（第一场） F

已采纳

精彩回帖

全部评论

本文相关内容

等你来战

热门推荐