某考试 T2 sum

 

    为什么其他人都是插值套插值啊,,,,就我是XJB做的吗2333

    k次多项式的前缀和可以表示成k+1次多项式,用两次这个玩意就可以发现g可以表示成一个k+2次多项式。

   然后我的做法是把g用拉格朗日插值+暴力多项式乘法乘出来,通过其他方法合并g(a),g(a+d),,,,,g(a+nd)。

   假设我们现在把g的多项式的每个指数的系数都求出来了,设x^i前的系数是b[i],

    那么答案显然就是b[0] * a^0 + b[1] * a^1 + b[2] * a^2 +....b[k+2] * a^(k+2)  + b[0] * (a+d)^0 + b[1] * (a+d)^1 + b[2] * (a+d)^2 +....b[k+2] * (a+d)^(k+2)+......

    然后我们可以把次数一样的合并到一个类里,提一个b[]出来,里面的求和只需要把括号拆了之后推一推,然后你就会发现还需要 组合数 和自然幂数前缀和。

    所以我们在之前预处理一下组合数 还有 1到n 的 1-k+2 次前缀和,就可以直接计算了。

 

    不过这个算法的瓶颈不在这里,这些计算的复杂度只是O(K^2)的。

    前面的暴力多项式乘法才是这个算法的瓶颈,是O(K^3)的,不过足够通过本题了2333 (反正好像直接插值套插值也要 O(N^3)吧)。

 

    至于前面的拉格朗日插值部分我就不说了,反正就算我说了会的还是会不会的还是不会2333。

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#define ll long long
using namespace std;
const int maxn=405;
const int ha=1234567891;
 
inline ll add(ll x,ll y){
    x+=y;
    return x>=ha?x-ha:x;
}
 
struct node{
    ll tp,a[maxn];
     
    inline void clear(){
        tp=0;
        memset(a,0,sizeof(a));
    }
     
    node operator +(const node &u)const{
        node r; r.clear();
        r.tp=max(tp,u.tp);
        for(int i=0;i<=r.tp;i++) r.a[i]=add(a[i],u.a[i]);
        return r;
    }
     
    node operator *(const node &u)const{
        node r; r.clear();
        r.tp=tp+u.tp;
        for(int i=0;i<=tp;i++)
            for(int j=0;j<=u.tp;j++) r.a[i+j]=add(r.a[i+j],a[i]*u.a[j]%ha);
        return r;
    }
     
    node operator *(const ll &u)const{
        node r; r.clear();
        r.tp=tp;
        for(int i=0;i<=tp;i++) r.a[i]=a[i]*u%ha;
        return r;
    }
}qz[maxn],hz[maxn],ANS;
ll K,A,N,D,T,L,jc[maxn];
ll f[maxn],g[maxn],ni[maxn];
ll C[maxn][maxn],sum[maxn];
 
inline ll ksm(ll x,ll y){
    ll an=1;
    for(;y;y>>=1,x=x*x%ha) if(y&1) an=an*x%ha;
    return an;
}
 
inline void prework(){
    L=K+3;
    for(int i=0;i<=L+1;i++) qz[i].clear(),hz[i].clear();
    ANS.clear();
}
 
inline void init(){
    C[0][0]=1;
    for(int i=1;i<=400;i++){
        C[i][0]=1;
        for(int j=1;j<=i;j++) C[i][j]=add(C[i-1][j-1],C[i-1][j]);
    }
    jc[0]=1,ni[0]=1;
    for(int i=1;i<=400;i++) jc[i]=jc[i-1]*(ll)i%ha,ni[i]=ksm(jc[i],ha-2);
}
 
inline ll calc(ll tmp){
    ll tot=0,INV=ksm(A,ha-2),base=ksm(A,tmp),d=1;
    for(int i=0;i<=tmp;i++,base=base*INV%ha,d=d*D%ha){
        tot=add(tot,sum[i]*C[tmp][i]%ha*base%ha*d%ha);
    }
    return tot;
}
 
inline void solve(){
    for(int i=1;i<=L;i++) f[i]=add(f[i-1],ksm(i,K));
    for(int i=1;i<=L;i++) g[i]=add(g[i-1],f[i]);
    //calc g
    qz[0].a[0]=hz[L+1].a[0]=1;
    for(int i=1;i<=L;i++){
        qz[i].a[0]=ha-i,qz[i].a[1]=1,qz[i].tp=1;
        hz[i]=qz[i];
    }
    for(int i=1;i<L;i++) qz[i]=qz[i-1]*qz[i];
    for(int i=L;i>1;i--) hz[i]=hz[i+1]*hz[i];
     
    for(int i=1;i<=L;i++) ANS=ANS+(qz[i-1]*hz[i+1])*(g[i]*ni[i-1]%ha*ni[L-i]%ha*(((L-i)&1)?ha-1:1)%ha);
    //g is OK!
     
    ll ans=0; sum[0]=N+1;
    for(int i=1;i<=ANS.tp;i++){
        sum[i]=ksm(N+1,i+1);
        for(int j=0;j<i;j++) sum[i]=add(sum[i],ha-C[i+1][i+1-j]*sum[j]%ha);
        sum[i]=sum[i]*ksm(i+1,ha-2)%ha;
    }
 
    for(int i=0;i<=ANS.tp;i++) ans=add(ans,ANS.a[i]*calc(i)%ha);
    printf("%lld\n",ans);
}
 
int main(){
//  freopen("sum.in","r",stdin);
//  freopen("sum.out","w",stdout);
     
    scanf("%lld",&T);
    init();
    while(T--){
        scanf("%lld%lld%lld%lld",&K,&A,&N,&D);
        prework();
        solve();
    }
    return 0;
}

  

posted @   蒟蒻JHY  阅读(185)  评论(0编辑  收藏  举报
编辑推荐:
· 开发中对象命名的一点思考
· .NET Core内存结构体系(Windows环境)底层原理浅谈
· C# 深度学习:对抗生成网络(GAN)训练头像生成模型
· .NET 适配 HarmonyOS 进展
· .NET 进程 stackoverflow异常后,还可以接收 TCP 连接请求吗?
阅读排行:
· 本地部署 DeepSeek:小白也能轻松搞定!
· 基于DeepSeek R1 满血版大模型的个人知识库,回答都源自对你专属文件的深度学习。
· 在缓慢中沉淀,在挑战中重生!2024个人总结!
· 大人,时代变了! 赶快把自有业务的本地AI“模型”训练起来!
· Tinyfox 简易教程-1:Hello World!
点击右上角即可分享
微信分享提示