[TJOI2018]教科书般的亵渎

Description
小豆喜欢玩游戏, 现在他在玩一个游戏遇到这样的场面,每个怪的血量为ai,且每个怪物血量均不相同, 小豆手里有无限张“亵渎”。亵渎的效果是对所有的怪造成1点伤害,如果有怪死亡,则再次施放该法术。我们认为血量为0怪物死亡。小豆使用一张“亵渎”会获得一定的分数,分数计算如下,在使用一张“亵渎”之后,每一个被亵渎造成伤害的怪会产生x^k,其中x是造成伤害前怪的血量为x和需要杀死所有怪物所需的“亵渎”的张数k。

Input
第一行输入一个T,表示有多少组测试数据。
每组组测试数据第一行为n,m,表示有当前怪物最高的血量n,和m种没有出现的血量。
接下来m行,每行1个数ai,表示场上没有血量为ai的怪物。
T ≤10 n ≤ 10^13 m ≤ 50

Output
一共T行,每行一个数,第i行表示第i组测试数据中小豆的最后可以获得的分数
因为这个分数会很大,需要模10^9 + 7。

Sample Input
2
10 1
5
4 2
1
2

Sample Output
415
135


MD……看题看半天系列

\(k\)一直是固定的,为\(m+1\)

每次得分是\(\sum\limits_{i=1}^{n}i^k-\sum\limits_{j=1}^ma_j^k\),然后\(n=n-a_1\)\(a_1\)消失,之后的所有\(a_j\)依次向前挪

所以……这题瓶颈在求幂和,求幂和参考这篇博客即可

/*program from Wolfycz*/
#include
#include
#include
#include
#include
#define inf 0x7f7f7f7f
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline char gc(){
    static char buf[1000000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
templateinline T frd(T x){
    int f=1; char ch=gc();
    for (;ch<'0'||ch>'9';ch=gc())   if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=gc()) x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}
templateinline T read(T x){
    int f=1;char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar())  if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar())    x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}
inline void print(int x){
    if (x<0)    putchar('-'),x=-x;
    if (x>9)    print(x/10);
    putchar(x%10+'0');
}
templateinline T min(T x,T y){return xinline T max(T x,T y){return x>y?x:y;}
const int N=50,p=1e9+7;
int prime[N+10],f[N+10],fac[N+10],inv[N+10];
bool inprime[N+10];
int mlt(int a,int b){
    int res=1;
    for (;b;b>>=1,a=1ll*a*a%p)  if (b&1)    res=1ll*res*a%p;
    return res;
}
void prepare(int k){
    int tot=0; f[1]=fac[0]=inv[0]=inv[1]=1,k++;
    for (int i=2;i<=k;i++){
        if (!inprime[i])    prime[++tot]=i,f[i]=mlt(i,k-1);
        for (int j=1;j<=tot&&i*prime[j]<=k;j++){
            inprime[i*prime[j]]=1;
            f[i*prime[j]]=1ll*f[i]*f[prime[j]]%p;
            if (i%prime[j]==0)  break;
        }
    }
    for (int i=1;i<=k;i++)  f[i]=(f[i]+f[i-1])%p;
    
    for (int i=1;i<=k;i++)  fac[i]=1ll*fac[i-1]*i%p;
    for (int i=2;i<=k;i++)  inv[i]=1ll*(p-p/i)*inv[p%i]%p;
    for (int i=1;i<=k;i++)  inv[i]=1ll*inv[i-1]*inv[i]%p;
}
int calc(int k,ll n){
    static int Pre[N+10],Suf[N+10];
    n%=p,k++,Pre[0]=n,Suf[k]=n-k; int res=0;
    for (int i=1;i<=k;i++)  Pre[i]=1ll*Pre[i-1]*(n-i)%p;
    for (int i=k-1;~i;i--)  Suf[i]=1ll*Suf[i+1]*(n-i)%p;
    for (int i=0;i<=k;i++){
        int tmp=1ll*f[i]*inv[i]%p*inv[k-i]%p;
        if (i!=0)   tmp=1ll*tmp*Pre[i-1]%p;
        if (i!=k)   tmp=1ll*tmp*Suf[i+1]%p;
        res=(res+((k-i)&1?-1:1)*tmp)%p;
    }
    return (res+p)%p;
}
int main(){
    for (int T=read(0);T;T--){
        static ll A[N+10];
        ll n=read(0ll); int m=read(0),k=m+1;
        for (int i=1;i<=m;i++)  A[i]=read(0ll);
        std::sort(A+1,A+1+m);
        prepare(k); int Ans=0;
        for (int i=1;i<=m;i++){
            Ans=(Ans+calc(k,n))%p;
            for (int j=i+1;j<=m;j++)    Ans=(Ans-mlt(A[j],k))%p,A[j]-=A[i];
            Ans=(Ans-mlt(A[i],k)),n-=A[i];
        }
        Ans=(Ans+calc(k,n))%p;
        printf("%d\n",(Ans+p)%p);
    }
    return 0;
}

转载于:https://www.cnblogs.com/Wolfycz/p/10630246.html

你可能感兴趣的:(ui)