zoukankan      html  css  js  c++  java
  • ●codeforces 553E Kyoya and Train

    题链:

    http://codeforces.com/problemset/problem/623/E

    题解:

    FFT,DP

    题意:

    一个有向图,给出每条边的起点u,终点v,费用c,以及花费每种时间的概率P[e][j](表示走第e条边花费时间为j的概率)

    现在需要从1号点走到n号点,如果不能在T个单位时间内到达,则达到后还要另外支付X的费用。

    求出所需支付的最小期望费用。


    先是一个暴力的DP方法:

    (考虑到每条边的耗时至少为1,可以把状态设为类似分层图的形式)

    定义$F[i][t]$为$t$时刻在$i$点时,到达终点n所需要的最小期望费用。

    不难得到DP转移,枚举每条i点的出边e:

    $$F[i][t]=min(c[e]+sum_{j=1}^{T}P[e][j] imes F[v[e]][t+j])$$

    然后令$G[e][t]$表示$t$时刻从${u[e]}$出发到达n点所需要的最小期望费用。

    即$$G[e][t]=c[e]+sum_{j=1}^{T}P[e][j] imes F[v[e]][t+j]$$

    所以$$F[i][t]=min(G[e][t])quad (u[e]=i)$$

    这个复杂度为$O(mT^2)$


    考虑优化,注意到求G的式子有点像卷积的形式。

    的确,我们只需要把$P[e]$数组翻转,即可得到:

    $$G[e][t]=G'[e][T+t]=c[e]+sum_{j=1}^{T}P[e][T-j] imes F[v[e]][t+j]$$

    这个就可以用FFT做了。

    但是我们并不知道P数组所有的值,又怎么办呢?

    这时采用分治的方法,(类似CDQ那种)

    基于这样一个事实:

    $$D_r=sum_{i=0}^{n-1}f_ig_{r-i}=sum_{i=0}^{k}f_ig_{r-i}+sum_{i=k+1}^{n-1}f_ig_{r-i}$$

    用文字描述就是:计算卷积时,可以把贡献分成几段来分别计算,

    或者说用FFT计算D的值时,可以先计算$sum_{i=0}^{k}f_ig_{r-i}这个卷积,再加上sum_{i=k+1}^{n-1}f_ig_{r-i}这个卷积$

    所以我们对时间进行分治:

    1.对于当前的分治层$t∈[l,r]$,我们先递归处理右边[mid,r],得到右边所有的F[i][mid~r],然后用FFT求值去贡献到G[e][l~mid-1]。

    2.继续递归处理左边[l,mid-1]。

    3.当到达递归最底层即$l=r$时,我们用G去求出所有的$F[i][l]$,因为此时的转移来源$G[e][t](t>l)$都计算完毕。

    算法的大体流程就这样,复杂度$O(mTlog^2T)$

    具体的实现还需特别注意:

    1.FFT时的下标转化感觉有点绕,(我是膜的别人的做法)。

    2.当t>给出的T时,由于已经超时,所以就直接按费用最短路前往n点。

    3.由于每次求G[e][t]时,需要的来源状态的时间t'<=t+T,所以数组的时间那一维开为2*T即可。

    代码:

    #include<bits/stdc++.h>
    #define MAXN 55
    #define MAXM 105
    #define MAXT 40005
    using namespace std;
    const double Pi=acos(-1);
    struct Complex{
    	double real,image;
    	Complex(double _real=0,double _image=0):real(_real),image(_image){}
    	Complex operator - () const{return Complex(-real,-image);}
    	friend Complex operator + (const Complex &A,const Complex &B){return Complex(A.real+B.real,A.image+B.image);}
    	friend Complex operator - (const Complex &A,const Complex &B){return A+(-B);}
    	friend Complex operator * (const Complex &A,const Complex &B){return Complex(A.real*B.real-A.image*B.image,A.image*B.real+A.real*B.image);}
    }A[65538],B[65538],nul(0,0);
    double F[MAXN][MAXT],G[MAXM][MAXT],P[MAXM][MAXT];
    int U[MAXM],V[MAXM],C[MAXM],S[MAXN][MAXN];
    int order[65538];
    int N,M,T,X;
    void read(int &x){
    	static int sign; static char ch;
    	x=0; sign=1; ch=getchar();
    	while(ch<'0'||'9'<ch){if(ch=='-')sign=-1;ch=getchar();}
    	while('0'<=ch&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    	x=x*sign;
    }
    void Floyd(){
    	memset(S,0x3f,sizeof(S));
    	for(int i=1;i<=N;i++) S[i][i]=0;
    	for(int e=1;e<=M;e++) S[U[e]][V[e]]=C[e];
    	for(int k=1;k<=N;k++)
    		for(int i=1;i<=N;i++) if(i!=k)
    			for(int j=1;j<=N;j++) if(j!=k&&j!=i)
    				S[i][j]=min(S[i][j],S[i][k]+S[k][j]);
    
    }
    void FFT(Complex *Y,int n,int sign){
    	for(int i=0;i<n;i++) if(i<order[i]) swap(Y[i],Y[order[i]]);
    	for(int d=2;d<=n;d<<=1){
    		Complex dw(cos(2*Pi/d),sin(sign*2*Pi/d)),w,tmp;
    		for(int i=0;w=Complex(1,0),i<n;i+=d)
    			for(int k=i;k<i+d/2;w=w*dw,k++)
    				tmp=w*Y[k+d/2],Y[k+d/2]=Y[k]-tmp,Y[k]=Y[k]+tmp;
    	}
    	if(sign==-1) for(int i=0;i<n;i++) Y[i].real/=1.0*n;
    }
    void contribution(int l,int mid,int r){
    	static int m,n,len;  m=r-l+1;
    	for(n=1,len=0;n<=m+r-mid;n<<=1) len++;
    	for(int i=1;i<n;i++) order[i]=(order[i>>1]>>1)|((i&1)<<(len-1));
    	for(int e=1;e<=M;e++){
    		for(int i=0;i<n;i++) A[i]=B[i]=nul;
    		for(int i=1;i<m;i++) B[i]=Complex(P[e][i],0);
    		for(int i=mid;i<=r;i++) A[r-i]=Complex(F[V[e]][i],0);
    		FFT(A,n,1); FFT(B,n,1);
    		for(int i=0;i<n;i++) A[i]=A[i]*B[i];
    		FFT(A,n,-1);
    		for(int i=l;i<mid;i++) G[e][i]+=A[r-i].real;
    	}
    }
    void solve(int l,int r){
    	if(l==r){
    		for(int i=1;i<N;i++) F[i][l]=S[i][N]+X;
    		for(int e=1;e<=M;e++)
    			F[U[e]][l]=min(F[U[e]][l],C[e]+G[e][l]);
    		return;
    	}
    	int mid=(l+r+1)/2;//向上取整
    	solve(mid,r);
    	contribution(l,mid,r);
    	solve(l,mid-1);
    }
    int main(){
    	read(N); read(M); read(T); read(X);
    	for(int e=1,x;e<=M;e++){
    		read(U[e]); read(V[e]); read(C[e]);
    		for(int i=1;i<=T;i++)
    			read(x),P[e][i]=1.0*x/100000;
    	}
    	Floyd();
    	for(int i=1;i<=N;i++)
    		for(int j=T+1;j<=2*T;j++) 
    			F[i][j]=S[i][N]+X;
    	for(int j=1;j<=T;j++) F[N][j]=0;
    	contribution(0,T+1,2*T);
    	solve(0,T);
    	printf("%.8lf
    ",F[1][0]);
    	return 0;
    }
    

      

  • 相关阅读:
    2017 《Java》预备作业计科1502宋奇蕊
    在 Kubernetes 上调度 GPU 资源
    ceph
    网络设备的 38 个知识点
    CF1066 ABCD
    单调队列优化动态规划
    对拍
    【关于此博客】
    使用Morphia框架操作mongodb
    通过mybatis读取数据库数据并提供rest接口访问
  • 原文地址:https://www.cnblogs.com/zj75211/p/8341345.html
Copyright © 2011-2022 走看看