HDU-5332(前缀和优化dp/CDQ+NTT)

HDU-5332(CDQ+NTT/前缀和优化dp)

考虑依次求出\(i\)个点的答案

假设当前有\(i-1\)个点,枚举第\(i\)个点前面的点数\(j\),则\(dp_i=dp_{i-j-1}\cdot (j+1)^2\cdot C(i-1,i-j-1)\cdot j!\)

直接转移是\(O(n^2)\)的,可以看到是一个\(dp\)转移与差值有关,所以可以用\(CDQ\)分治+\(NTT\)解决

关于这种简单粗暴的做法,模板题HDU-5730 题解

以下是暴力的代码

#include<bits/stdc++.h>
using namespace std;

#define reg register
typedef long long ll;
#define rep(i,a,b) for(reg int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(reg int i=a,i##end=b;i>=i##end;--i)

template <class T> inline void cmin(T &a,T b){ ((a>b)&&(a=b)); }
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); } 

char IO;
int rd(){
    int s=0,f=0;
    while(!isdigit(IO=getchar())) if(IO=='-') f=1;
    do s=(s<<1)+(s<<3)+(IO^'0');
    while(isdigit(IO=getchar()));
    return f?-s:s;
}

const int N=(1<<18)+4,P=998244353;

int n=1e5;
ll dp[N];
ll A[N],B[N];
ll Inv[N],Fac[N];
int rev[N];
ll qpow(ll x,ll k) {
    ll res=1;
    for(;k;k>>=1,x=x*x%P) if(k&1) res=res*x%P;
    return res;
}

void NTT(int n,ll *a,int f){
    rep(i,1,n-1) if(rev[i]>i) swap(a[i],a[rev[i]]);
    for(reg int i=1;i<n;i<<=1) {
        int len=i*2;
        ll w=qpow(f==1?3:(P+1)/3,(P-1)/i/2);
        for(reg int l=0;l<n;l+=len) {
            ll e=1;
            for(reg int j=l;j<l+i;++j,e=e*w%P) {
                ll t=a[j+i]*e;
                a[j+i]=(a[j]-t)%P;
                a[j]=(a[j]+t)%P;
            }
        }
    }
    if(f==-1) {
        ll base=qpow(n,P-2);
        rep(i,0,n-1) A[i]=(A[i]*base%P+P)%P;
    }
}

void Solve(int l,int r){
    if(l==r) return;
    int mid=(l+r)>>1;
    Solve(l,mid);
    int R=1,c=-1;
    while(R<=r-l+1) R<<=1,c++;
    rep(i,1,R) rev[i]=(rev[i>>1]>>1)|((i&1)<<c);
    rep(i,l,mid) A[i-l]=dp[i]*Inv[i]%P;
    rep(i,1,r-l+1) B[i]=1ll*i*i%P;
    NTT(R,A,1),NTT(R,B,1);
    rep(i,0,R) A[i]=A[i]*B[i]%P;
    NTT(R,A,-1);
    rep(i,mid+1,r) dp[i]=(dp[i]+A[i-l]*Fac[i-1])%P;
    rep(i,0,R) A[i]=B[i]=0;
    Solve(mid+1,r);
}

int main(){
    Fac[0]=Fac[1]=Inv[0]=Inv[1]=1;
    rep(i,2,n) {
        Fac[i]=Fac[i-1]*i%P;
        Inv[i]=(P-P/i)*Inv[P%i]%P;
    }
    rep(i,2,n) Inv[i]=Inv[i-1]*Inv[i]%P;
    dp[0]=1;
    Solve(0,n);
    while(~scanf("%d",&n)) printf("%lld\n",dp[n]);
}

\[ \ \]

下面是\(O(n)\)做法

观察转移\(dp_i=dp_{i-j-1}\cdot (j+1)^2\cdot C(i-1,i-j-1)\cdot j!\)

变形一下\(dp_i=dp_j\cdot (i-j)^2\cdot \frac{(i-1)!}{j!}\)

考虑前缀和转移

可以看到阶乘的问题可以通过参数分离很好地解决,就是\((i-j)^2\)的问题

当\(i\rightarrow i+1\)时,\((i-j)^2\rightarrow (i+1-j)^2\)

提出来看一下,就是\(x^2\rightarrow(x+1)^2=x^2+2x+1\)

所以我们可以记录\(a=\sum x^2dp_j,b=\sum xdp_j,c=\sum dp_j\)

所以每次\(i\)增加,\(a=a+2b+c,b=b+c\)即可

#include<bits/stdc++.h>
using namespace std;

#define reg register
typedef long long ll;
#define rep(i,a,b) for(reg int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(reg int i=a,i##end=b;i>=i##end;--i)

template <class T> inline void cmin(T &a,T b){ ((a>b)&&(a=b)); }
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); } 

char IO;
int rd(){
    int s=0,f=0;
    while(!isdigit(IO=getchar())) if(IO=='-') f=1;
    do s=(s<<1)+(s<<3)+(IO^'0');
    while(isdigit(IO=getchar()));
    return f?-s:s;
}

const double PI=acos(-1);

const int N=1e5+10,P=998244353;

int n=1e5;
ll Fac[N],Inv[N];
ll dp[N];
int main(){
    Fac[0]=Fac[1]=Inv[0]=Inv[1]=1;
    rep(i,2,n) {
        Fac[i]=Fac[i-1]*i%P;
        Inv[i]=(P-P/i)*Inv[P%i]%P;
    }
    rep(i,2,n) Inv[i]=Inv[i-1]*Inv[i]%P;
    dp[0]=1;
    ll x=1,y=1,z=1;
    rep(i,1,n) {
        dp[i]=x*Fac[i-1]%P; // 累上(i-1)!
        x=(x+2*y+z)%P; // x
        y=(y+z)%P;     // y
        ll t=dp[i]*Inv[i]%P; // 考虑i的贡献
        x=(x+t)%P;
        y=(y+t)%P;
        z=(z+t)%P;
    }
    while(~scanf("%d",&n)) printf("%lld\n",dp[n]);
}

原文地址:https://www.cnblogs.com/chasedeath/p/12101410.html

时间: 2024-09-29 09:50:40

HDU-5332(前缀和优化dp/CDQ+NTT)的相关文章

Print Article hdu 3507 一道斜率优化DP 表示是基础题,但对我来说很难

Print Article Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 131072/65536 K (Java/Others)Total Submission(s): 4990    Accepted Submission(s): 1509 Problem Description Zero has an old printer that doesn't work well sometimes. As it is antique

bzoj2431: [HAOI2009]逆序对数列(前缀和优化dp)

2431: [HAOI2009]逆序对数列 Time Limit: 5 Sec  Memory Limit: 128 MBSubmit: 2312  Solved: 1330[Submit][Status][Discuss] Description 对于一个数列{ai},如果有i<j且ai>aj,那么我们称ai与aj为一对逆序对数.若对于任意一个由1~n自然数组成的 数列,可以很容易求出有多少个逆序对数.那么逆序对数为k的这样自然数数列到底有多少个? Input 第一行为两个整数n,k. Ou

P5241 序列(滚动数组+前缀和优化dp)

P5241 序列 挺神仙的一题 看看除了dp好像没什么其他办法了 想着怎么构个具体的图出来,然鹅不太现实. 于是我们想办法用几个参数来表示dp数组 加了几条边肯定要的吧,于是加个参数$i$表示已加了$i$条边 这显然是不够的.于是我们又想:强连通分量.....连通块....... 于是加个$j$表示还有$j$个强连通分量 于是dp数组为$f[i][j]$ 这是我们发现一个问题,状态$f[i][j]$不一定是合法的. 那dp不就GG了吗 再次撕烤,我们发现每次加上的边无非就3种情况: 1.把2个强

HDU 2829 Lawrence (斜率优化DP或四边形不等式优化DP)

题意:给定 n 个数,要你将其分成m + 1组,要求每组数必须是连续的而且要求得到的价值最小.一组数的价值定义为该组内任意两个数乘积之和,如果某组中仅有一个数,那么该组数的价值为0. 析:DP状态方程很容易想出来,dp[i][j] 表示前 j 个数分成 i 组.但是复杂度是三次方的,肯定会超时,就要对其进行优化. 有两种方式,一种是斜率对其进行优化,是一个很简单的斜率优化 dp[i][j] = min{dp[i-1][k] - w[k] + sum[k]*sum[k] - sum[k]*sum[

hdu 3401(单调队列优化dp)

注意:这题题意是有操作的天数相隔要大于w 然后列出状态转移方程就可以发现,可以用优点队列优化啦. 构造状态dp[i][j]表示第i 天拥有 j只股票的时候,赚了多少钱 状态转移有: 1.从前一天不买不卖: dp[i][j]=max(dp[i-1][j],dp[i][j]) 2.从前i-W-1天买进一些股: dp[i][j]=max(dp[i-W-1][k]-(j-k)*AP[i],dp[i][j]) 3.从i-W-1天卖掉一些股: dp[i][j]=max(dp[i-W-1][k]+(k-j)*

bzoj1492--斜率优化DP+cdq分治

显然在某一天要么花完所有钱,要么不花钱. 所以首先想到O(n^2)DP: f[i]=max{f[i-1],(f[j]*r[j]*a[i]+f[j]*b[i])/(a[j]*r[j]+b[j])},j<i 其中f[j]*r[j]/(a[j]*r[j]+b[j])是第j天最多能买多少A券,B类似. 假如我们已经找到了最优的j,那么有f[i]=(f[j]*r[j]*a[i]+f[j]*b[i])/(a[j]*r[j]+b[j]) 令x[j]=f[j]*r[j]/(a[j]*r[j]+b[j]),y[j

BNU 13064 Dice (I) 前缀和优化DP

Dice (I) You have N dices; each of them has K faces numbered from 1 to K. Now you have arranged the N dices in a line. You can rotate/flip any dice if you want. How many ways you can set the top faces such that the summation of all the top faces equa

HDU 6619 Horse 斜率优化dp

http://acm.hdu.edu.cn/showproblem.php?pid=6619 #include<bits/stdc++.h> #define fi first #define se second #define INF 0x3f3f3f3f #define LNF 0x3f3f3f3f3f3f3f3f #define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0) #define pqueue priority_qu

hdu 3401 单调队列优化+dp

http://acm.hdu.edu.cn/showproblem.php?pid=3401 Trade Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 5188    Accepted Submission(s): 1776 Problem Description Recently, lxhgww is addicted to stoc