题目描述
对于刚上大学的牛牛来说, 他面临的第一个问题是如何根据实际情况中情合适的课程。
在可以选择的课程中,有2n节课程安排在n个时间段上。在第 i ( 1≤ i≤n)个时同段上, 两节内容相同的课程同时在不同的地点进行, 其中, 牛牛预先被安排在教室 ci上课, 而另一节课程在教室 di进行。
在不提交任何申请的情况下,学生们需要按时间段的顺序依次完成所有的n节安排好的课程。如果学生想更换第i节课程的教室,则需要提出中情。若申请通过,学生就可以在第 i个时间段去教室 di上课, 否则仍然在教室 ci上课。
由于更换教室的需求太多, 申请不一定能获得通过。 通过计算, 牛牛发现申请更换第 i节课程的教室时, 中情被通过的概率是一个已知的实数 ki, 并且对于不同课程的申请, 被通过的概率是互相独立的。
学校规定, 所有的申请只能在学期开始前一次性提交, 并且每个人只能选择至多m节课程进行申请。 这意味着牛牛必须一次性决定是否申请更换每节课的教室, 而不能根据某些课程的申请结果来决定其他课程是否申请; 牛牛可以申请白己最希望更换教室的 m门课程,也可以不用完这m个中情的机会,甚至可以一门课程都不申请。
因为不同的课程可能会被安排在不同的教室进行, 所以牛牛需要利用课问时间从一间教室赶到另一间教室。
牛牛所在的大学有 v个教室,有 e条道路。每条道路连接两间教室, 并且是可以双向通行的。 由于道路的长度和拥;i者程度不同, 通过不同的道路耗费的体力可能会有所不同。当第i ( 1≤i≤n-1 )节课结束后,牛牛就会从这节课的教室出发,选择一条耗费体力最少的路径前往下一节课的教室。
现在牛牛想知道,申请哪几门课程可以使他因在教室问移动耗费的体力值的总和的期望值最小,请你帮他求出这个最小值。
输入输出格式
输入格式:
第一行四个整数 n,m,v,e 。 n表示这个学期内的时间段的数量; m表示牛牛最多可以申请更换多少节课程的教室; v表示牛牛学校里教室的数量; e表示牛牛的学校里道路的数量。
第二行n个正整数,第 i ( 1≤ i≤ n)个正整数表示c,,即第 i个时间段牛牛被安排上课的教室;保证1≤ ci≤ v。
第三行n个正整数,第 i ( 1≤ i≤ n)个正整数表示di,即第 i个时间段另一间上同样课程的教室;保证1≤ di≤ v。
第四行n个实数,第 i ( 1≤ i≤ n)个实数表示ki,即牛牛申请在第 i个时间段更换教室获得通过的概率。保证0≤ ki≤1 。
接下来 e行,每行三个正整数aj,bj,wj,表示有一条双向道路连接教室 aj ,bj ,通过这条道路需要耗费的体力值是 Wj ;保证1≤ aj,bj≤ v, 1≤ wj≤100 。
保证1≤n≤2000, 0≤m≤2000, 1≤v≤300, 0≤ e≤90000。
保证通过学校里的道路,从任何一间教室出发,都能到达其他所有的教室。
保证输入的实数最多包含3位小数。
输出格式:
输出一行,包含一个实数,四舎五入精确到小数点后恰好2位,表示答案。你的
输出必须和标准输出完全一样才算正确。
测试数据保证四舎五入后的答案和准确答案的差的绝对值不大于4 *10^-3 。 (如果你不知道什么是浮点误差, 这段话可以理解为: 对于大多数的算法, 你可以正常地使用浮点数类型而不用对它进行特殊的处理)
输入输出样例
3 2 3 3 2 1 2 1 2 1 0.8 0.2 0.5 1 2 5 1 3 3 2 3 1
2.80
说明
【样例1说明】
所有可行的申请方案和期望收益如下表:
【提示】
- 道路中可能会有多条双向道路连接相同的两间教室。 也有可能有道路两端连接
的是同一间教室。
2.请注意区分n,m,v,e的意义, n不是教室的数量, m不是道路的数量。
特殊性质1:图上任意两点 ai, bi, ai≠ bi间,存在一条耗费体力最少的路径只包含一条道路。
特殊性质2:对于所有的1≤ i≤ n, ki= 1 。
483 055 310
这道题考场上准备骗30+分,结果没看样例,期望的概念混淆了,就得了4分。其实很简单的暴力(位向量法+Floyd)就能得80分,原以为这题很难的。看了下题解正解是DP+Floyd,但DP什么都不会,还是不看了,反正暴力80分。
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
//n个时间段,最多换m个时间段,v个教室,e条边,c是换前,d是换后,dist是两点最短路,ok是是否换,use_m是已用的机会
int n,m,v,e,c[2005],d[2005],dist[305][305],ok[2005],use_m;
double k[2005],minn=1000000.0,tot;
//k是概率,minn是最小期望,tot是计算的期望
void print_subset(int cur) //现在该决定第cur个时间段是否换
{
if(cur==n+1) //n个时间段都已决定
{
tot=0.0; //初始化
for(int i=1;i<n;i++) //n个时间段,走n-1条路
{ //算期望时决定不换一定是c,决定换的话分别算c和d的概率加权
if(ok[i])
{ //前换后换
if(ok[i+1])tot=tot+k[i]*k[i+1]*dist[d[i]][d[i+1]]+k[i]*(1-k[i+1])*dist[d[i]][c[i+1]]
+(1-k[i])*k[i+1]*dist[c[i]][d[i+1]]+(1-k[i])*(1-k[i+1])*dist[c[i]][c[i+1]];
else tot=tot+k[i]*dist[d[i]][c[i+1]]+(1-k[i])*dist[c[i]][c[i+1]];
} //前换后不换
else
{ //前不换后换
if(ok[i+1])tot=tot+k[i+1]*dist[c[i]][d[i+1]]+(1-k[i+1])*dist[c[i]][c[i+1]];
else tot+=dist[c[i]][c[i+1]];
} //前不换后不换
}
minn=min(minn,tot); //与最优值比较
return;
}
//递归遍历所以情况 //注意是<而不是<=,因为=时就不能再继续换了
if(use_m<m){ok[cur]=1; use_m++; print_subset(cur+1); use_m--;}
ok[cur]=0; print_subset(cur+1);
}
int main()
{
//InPut
int a,b,x;
cin>>n>>m>>v>>e;
//Init
for(int i=1;i<=v;i++)for(int j=1;j<=v;j++)dist[i][j]=1000000;
for(int i=1;i<=v;i++)dist[i][i]=0;
//InPut
for(int i=1;i<=n;i++)scanf("%d",&c[i]);
for(int i=1;i<=n;i++)scanf("%d",&d[i]);
for(int i=1;i<=n;i++)scanf("%lf",&k[i]); //注意不是%d
for(int i=1;i<=e;i++){scanf("%d%d%d",&a,&b,&x);dist[a][b]=dist[b][a]=min(dist[a][b],x);}
//Floyd
for(int k=1;k<=v;k++)for(int i=1;i<=v;i++)for(int j=1;j<=v;j++)dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j]);
//Solve
print_subset(1);
//OutPut
printf("%.2lf\n",minn);
return 0;
}
上述代码得了80分,后来又想到运用秦九韶算法的思想,可以优化,应该能AC,结果还是80分,细看,略快,但不够,如此看来,想AC只能用DP了。
下面是运用秦九韶算法的思想的代码
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,v,e,c[2005],d[2005],dist[305][305],ok[2005],use_m;
double k[2005], minn=1000000.0,tot;
void print_subset(int cur,double tot) //现在该决定第cur个时间段,以前期望和为tot
{
double num=0.0;//计算此点和上一点之间的期望
if(cur==n+1){minn=min(minn,tot); return;} //End
if(use_m<m)
{
ok[cur]=1;
use_m++;
if(cur!=1) //注意cur=1是没有上一点
{
if(ok[cur-1]) num=k[cur]*k[cur-1]*dist[d[cur]][d[cur-1]]+k[cur]*(1-k[cur-1])*dist[d[cur]][c[cur-1]]
+(1-k[cur])*k[cur-1]*dist[c[cur]][d[cur-1]]+(1-k[cur])*(1-k[cur-1])*dist[c[cur]][c[cur-1]];
else num=k[cur]*dist[d[cur]][c[cur-1]]+(1-k[cur])*dist[c[cur]][c[cur-1]];
}
print_subset(cur+1,tot+num); //递归到下一个时更新了下一个的tot
use_m--;
}
ok[cur]=0;
num=0.0;
if(cur!=1)
{
if(ok[cur-1])num=k[cur-1]*dist[c[cur]][d[cur-1]]+(1-k[cur-1])*dist[c[cur]][c[cur-1]];
else num=dist[c[cur]][c[cur-1]];
}
print_subset(cur+1,tot+num);
}
int main()
{
//InPut
int a,b,x;
cin>>n>>m>>v>>e;
//Init
for(int i=1;i<=v;i++)for(int j=1;j<=v;j++)dist[i][j]=1000000;
for(int i=1;i<=v;i++)dist[i][i]=0;
//InPut
for(int i=1;i<=n;i++)scanf("%d",&c[i]);
for(int i=1;i<=n;i++)scanf("%d",&d[i]);
for(int i=1;i<=n;i++)scanf("%lf",&k[i]);
for(int i=1;i<=e;i++){scanf("%d%d%d",&a,&b,&x);dist[a][b]=dist[b][a]=min(dist[a][b],x);}
//Floyd
for(int k=1;k<=v;k++)for(int i=1;i<=v;i++)for(int j=1;j<=v;j++)dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j]);
//Solve
print_subset(1,0.0);
//OutPut
printf("%.2lf\n",minn);
return 0;
}