【NOIP2016提高組】 Day1 T3 換教室
題目鏈接:https://www.luogu.org/problemnew/show/P1850
此題正解為dp。
我們先用floyd處理出任意兩個教室之間的距離,用dis[i][j]表示。
用f[i][j][0..1]表示在前i個課程中,用了j次換課的機會,第i節課選擇換還是不換。
f[i][j][0]可以選擇用f[i-1][j][0]更新(即第i-1節課選擇不換課),代價為dis[c[i-1]][c[i]]。
同時,第i-1節課也可以選擇換課,若換課成功,代價為dis[d[i-1]][c[i]],概率為k[i-1]。若換課不成功,則代價還是dis[c[i-1]][c[i]],概率為(1-k[i-1])。
則f[i][j][0]=min(f[i-1][j][0]+dis[c[i-1]][c[i]],(f[i-1][j-1][1]+dis[d[i-1]][c[i]])*k[i-1]+(f[i-1][j][0]+dis[c[i-1]][c[i]])*(1-k));
同理,f[i][j][1]可以選擇用f[i-1][j][0]更新,若換課成功,代價為dis[c[i-1]][d[i]],概率為k[i]。若換課不成功,則代價是dis[c[i-1]][c[i]],概率為(1-k[i])。
同時,f[i][j][1]也可以選擇用f[i-1][j][1]更新。
若兩次換課均成功,代價為dis[d[i-1]][d[i]],概率為k[i]*k[i-1]。
若兩次換課均不成功,代價為dis[c[i-1]][c[i]],概率為(1-k[i])*(1-k[i-1])。
若第i次成功而第i-1次不成功,代價為dis[c[i-1]][d[i]],概率為(1-k[i-1])*k[i]。
若第i次不成功而第i-1次成功,代價為dis[d[i-1]][c[i]],概率為k[i-1]*(1-k[i])。
則f[i][j][1]=min((f[i-1][j][0]+dis[c[i-1]][d[i]])*k[i]+(f[i-1][j][0]+dis[c[i-1]][c[i]])*(1-k[i]),f[i-1][j][1] + dis[d[i-1]][d[i]]*k[i]*k[i-1] + dis[c[i-1]][c[i]]*(1-k[i])*(1-k[i-1]) + dis[c[i-1]][d[i]]*(1-k[i-1])*k[i] + dis[d[i-1]][c[i]]*k[i-1]*(1-k[i]))。
可以看出每次轉移是O(1)的,則時間復雜度為O(n*m)+O(v^3)。
最終結果為min(f[n][i][k]) i∈[0,m],j∈[0,1]。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #define M 2002 5 using namespace std; 6 int n,m,v,e,c[M]={0},d[M]={0}; 7 double p[M]={0},dis[305][305]={0},f[M][M][2]={0},minn=123456789; 8 void pmin(double &x,double y){ 9 x=min(x,y); 10 } 11 12 int main(){ 13 scanf("%d%d%d%d",&n,&m,&v,&e); 14 for(int i=1;i<=n;i++) scanf("%d",c+i); 15 for(int i=1;i<=n;i++) scanf("%d",d+i); 16 for(int i=1;i<=n;i++) scanf("%lf",p+i); 17 for(int i=1;i<=v;i++) 18 for(int j=1;j<=v;j++) dis[i][j]=12345678; 19 for(int i=1;i<=e;i++){ 20 int x,y,z;scanf("%d%d%d",&x,&y,&z); 21 pmin(dis[x][y],z); pmin(dis[y][x],z); 22 } 23 for(int i=1;i<=v;i++) dis[i][i]=0; 24 for(int k=1;k<=v;k++) 25 for(int i=1;i<=v;i++) 26 for(int j=1;j<=v;j++) 27 dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]); 28 for(int i=0;i<=n;i++) 29 for(int j=0;j<=m;j++) f[i][j][0]=f[i][j][1]=12345678; 30 f[1][0][0]=0; f[1][1][1]=0; 31 for(int i=1;i<n;i++) 32 for(int j=0;j<=min(i,m);j++){ 33 pmin(f[i+1][j][0],f[i][j][0]+dis[c[i]][c[i+1]]); 34 double px;px=dis[c[i]][d[i+1]]*p[i+1]+dis[c[i]][c[i+1]]*(1-p[i+1]); 35 pmin(f[i+1][j+1][1],f[i][j][0]+px); 36 px=dis[d[i]][c[i+1]]*p[i]+dis[c[i]][c[i+1]]*(1-p[i]); 37 pmin(f[i+1][j][0],f[i][j][1]+px); 38 px=dis[d[i]][d[i+1]]*p[i]*p[i+1]+dis[d[i]][c[i+1]]*p[i]*(1-p[i+1])+dis[c[i]][d[i+1]]*(1-p[i])*p[i+1]+dis[c[i]][c[i+1]]*(1-p[i])*(1-p[i+1]); 39 pmin(f[i+1][j+1][1],f[i][j][1]+px); 40 } 41 42 for(int i=0;i<=m;i++){ 43 minn=min(minn,f[n][i][0]); 44 minn=min(minn,f[n][i][1]); 45 } 46 printf("%.2lf\n",minn); 47 }
【NOIP2016提高組】 Day1 T3 換教室