UOJ #262. 【NOIP2016】换教室

对于刚上大学的牛牛来说，他面临的第一个问题是如何根据实际情况申请合适的课程。

在可以选择的课程中，有 2n 节课程安排在 n 个时间段上。在第 i（1≤i≤n）个时间段上，两节内容相同的课程同时在不同的地点进行，其中，牛牛预先被安排在教室 c_i 上课，而另一节课程在教室 d_i 进行。

在不提交任何申请的情况下，学生们需要按时间段的顺序依次完成所有的 n 节安排好的课程。如果学生想更换第 i 节课程的教室，则需要提出申请。若申请通过，学生就可以在第 i个时间段去教室 d_i 上课，否则仍然在教室 c_i 上课。

由于更换教室的需求太多，申请不一定能获得通过。通过计算，牛牛发现申请更换第 i 节课程的教室时，申请被通过的概率是一个已知的实数 k_i，并且对于不同课程的申请，被通过的概率是互相独立的。

学校规定，所有的申请只能在学期开始前一次性提交，并且每个人只能选择至多 m 节课程进行申请。这意味着牛牛必须一次性决定是否申请更换每节课的教室，而不能根据某些课程的申请结果来决定其他课程是否申请；牛牛可以申请自己最希望更换教室的 m 门课程，也可以不用完这 m 个申请的机会，甚至可以一门课程都不申请。

因为不同的课程可能会被安排在不同的教室进行，所以牛牛需要利用课间时间从一间教室赶到另一间教室。

牛牛所在的大学有 v 个教室，有 e 条道路。每条道路连接两间教室，并且是可以双向通行的。由于道路的长度和拥堵程度不同，通过不同的道路耗费的体力可能会有所不同。 当第 i（1≤i≤n−1）节课结束后，牛牛就会从这节课的教室出发，选择一条耗费体力最少的路径前往下一节课的教室。

现在牛牛想知道，申请哪几门课程可以使他因在教室间移动耗费的体力值的总和的期望值最小，请你帮他求出这个最小值。

输入格式

从标准输入读入数据。

第一行四个整数 n,m,v,e。n 表示这个学期内的时间段的数量；m 表示牛牛最多可以申请更换多少节课程的教室；v 表示牛牛学校里教室的数量；e表示牛牛的学校里道路的数量。

第二行 n 个正整数，第 i（1≤i≤n）个正整数表示 c_i，即第 i 个时间段牛牛被安排上课的教室；保证 1≤c_i≤v。

第三行 n 个正整数，第 i（1≤i≤n）个正整数表示 d_i，即第 i 个时间段另一间上同样课程的教室；保证 1≤d_i≤v。

第四行 n 个实数，第 i（1≤i≤n）个实数表示 k_i，即牛牛申请在第 i 个时间段更换教室获得通过的概率。保证 0≤k_i≤1。

接下来 e 行，每行三个正整数 a_j,b_j,w_j，表示有一条双向道路连接教室 a_j,b_j，通过这条道路需要耗费的体力值是 w_j；保证 1≤a_j,b_j≤v， 1≤w_j≤100。

保证 1≤n≤2000，0≤m≤2000，1≤v≤300，0≤e≤90000。

保证通过学校里的道路，从任何一间教室出发，都能到达其他所有的教室。

保证输入的实数最多包含 3 位小数。

输出格式

输出到标准输出。

输出一行，包含一个实数，四舍五入精确到小数点后恰好2位，表示答案。你的输出必须和标准输出完全一样才算正确。

测试数据保证四舍五入后的答案和准确答案的差的绝对值不大于 4×10−3。 （如果你不知道什么是浮点误差，这段话可以理解为：对于大多数的算法，你可以正常地使用浮点数类型而不用对它进行特殊的处理）

样例一

input

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

output

2.80

explanation

所有可行的申请方案和期望收益如下表：

申请更换教室的时间段	申请通过的时间段	出现的概率	耗费的体力值	耗费的体力值的期望
无	无	1.0	8	8.0
1	1	0.8	4	4.8
	无	0.2	8
2	2	0.2	0	6.4
	无	0.8	8
3	3	0.5	4	6.0
	无	0.5	8
1、2	1、2	0.16	4	4.48
	1	0.64	4
	2	0.04	0
	无	0.16	8
1、3	1、3	0.4	0	2.8
	1	0.4	4
	3	0.1	4
	无	0.1	8
2、3	2、3	0.1	4	5.2
	2	0.1	0
	3	0.4	4
	无	0.4	8

样例二

input

见下载。

output

见下载。

explanation

1. 道路中可能会有多条双向道路连接相同的两间教室。也有可能有道路两端连接的是同一间教室。
2. 请注意区分 n,m,v,e 的意义，n 不是教室的数量，m 不是道路的数量。

限制与约定

测试点	n	m	v	特殊性质1	特殊性质2
1	≤1	≤1	≤300	×	×
2	≤2	≤0	≤20≤
3		≤1	≤100
4		≤2	≤300
5	≤3	≤0	≤20	√	√
6		≤1	≤100		×
7		≤2	≤300	×
8	≤10	≤0		√	√
9		≤1	≤20		×
10		≤2	≤100	×
11		≤10	≤300		√
12	≤20	≤0	≤20	√	×
13		≤1	≤100	×
14		≤2	≤300	√
15		≤20		×	√
16	≤300	≤0	≤20		×
17		≤1	≤100
18		≤2	≤300	√	√
19		≤300		×
20	≤2000	≤0	≤20		×
21		≤1
22		≤2	≤100
23		≤2000
24			≤300
25

特殊性质1：图上任意两点 a_i,b_i （a_i≠b_i）间，存在一条耗费体力最少的路径只包含一条道路。

特殊性质2：对于所有的 1≤i≤n,k_i=1。

时间限制：1s

空间限制：512MB

下载

样例数据下载

题目链接:http://uoj.ac/problem/262

解题报告

首先,先用floyd处理两两点对之间的最短路,复杂度O(n³),

inline void floyd(){
	FOR(k,1,v)
		FOR(i,1,v)
			FOR(j,1,v)
				if(dis[i][j]>dis[i][k]+dis[k][j]&&i!=j&&i!=k&&j!=k)
					dis[i][j]=dis[i][k]+dis[k][j],dis[j][i]=dis[i][k]+dis[k][j];
	return;
}

再推一推概率/期望DP状态转移方程

定义:f[i][0/1][j]:

1.i表示在第i个时间段.

2.0/1表示当前是否申请了换教室.

3.j表示用了j次申请

所以状态转移为

(1).当前时间段不申请,上个时间段也不申请，

则最小代价为f[i-1][0][j]+dis[c[i-1]][c[i]],

(2).当前时间段不申请,但上个时间段申请了，

则最小期望代价为f[i-1][1][j]+(1.00-k[i-1])*dis[c[i-1]][c[i]]+k[i-1]*dis[d[i-1]][c[i]],

所以,f[i][0][j]=min(f[i-1][0][j]+dis[c[i-1]][c[i]],f[i-1][1][j]+(1.00-k[i-1])*(db)dis[c[i-1]][c[i]]+k[i-1]*(db)dis[d[i-1]][c[i]]).

(3).当前时间段申请了,但上个时间段不申请，

则最小期望代价为f[i-1][0][j-1]+k[i]*dis[c[i-1]][d[i]]+(1.00-k[i])*dis[c[i-1]][c[i]],

(4).当前时间段申请了,上个时间段也申请了，

则最小期望代价为f[i-1][1][j-1]+k[i-1]*k[i]*dis[d[i-1]][d[i]]+(1.00-k[i-1])*k[i]*dis[c[i-1]][d[i]]+k[i-1]*(1.00-k[i])*dis[d[i-1]][c[i]]+(1.00-k[i-1])*(1.00-k[i])*dis[c[i-1]][c[i]],

所以,f[i][1][j]=min(f[i-1][0][j-1]+k[i]*dis[c[i-1]][d[i]]+(1.00-k[i])*dis[c[i-1]][c[i]],f[i-1][1][j-1]+k[i-1]*k[i]*dis[d[i-1]][d[i]]+(1.00-k[i-1])*k[i]*dis[c[i-1]][d[i]]+k[i-1]*(1.00-k[i])*dis[d[i-1]][c[i]]+(1.00-k[i-1])*(1.00-k[i])*dis[c[i-1]][c[i]]).

所以,总复杂度为O(n*m).

ans为max{f[n][0/1][0~m]}

状态转移实现如下

FOR(i,0,n)FOR(j,0,m)f[i][1][j]=f[i][0][j]=(db)INF;
	f[1][0][0]=0.0;
	f[1][1][1]=0.0;
	
	FOR(i,2,n) 
        f[i][0][0]=f[i-1][0][0]+(db)dis[c[i-1]][c[i]];
	FOR(j,1,m)
		FOR(i,2,n){
			f[i][0][j]=f[i-1][0][j]+dis[c[i-1]][c[i]];
			cmp=(db)(f[i-1][1][j]+(1.00-k[i-1])*(db)dis[c[i-1]][c[i]]+k[i-1]*(db)dis[d[i-1]][c[i]]);
			if((db)cmp<(db)f[i][0][j])f[i][0][j]=(db)cmp;	
			f[i][1][j]=f[i-1][0][j-1]+k[i]*(db)dis[c[i-1]][d[i]]+(1.00-k[i])*(db)dis[c[i-1]][c[i]];
			cmp=(db)(f[i-1][1][j-1]+k[i-1]*k[i]*(db)dis[d[i-1]][d[i]]+(1.00-k[i-1])*k[i]*(db)dis[c[i-1]][d[i]]+k[i-1]*(1.00-k[i])*(db)dis[d[i-1]][c[i]]+(1.00-k[i-1])*(1.00-k[i])*(db)dis[c[i-1]][c[i]]);
			if((db)f[i][1][j]>(db)cmp)f[i][1][j]=(db)cmp;
		}

AC代码

#include<cstdio>
#include<iostream>
#define db double
#define FOR(i,s,t) for(register int i=s;i<=t;++i)
using namespace std;
db f[2011][2][2011];
int dis[311][311];
int c[2011];
int d[2011];
db k[2011];
db cmp,ans;
const db INF=1000000000.11;
int n,m,v,e;
int x,y,z,tot;
inline void floyd(){
	FOR(k,1,v)
		FOR(i,1,v)
			FOR(j,1,v)
				if(dis[i][j]>dis[i][k]+dis[k][j]&&i!=j&&i!=k&&j!=k)
					dis[i][j]=dis[i][k]+dis[k][j],dis[j][i]=dis[i][k]+dis[k][j];
	return;
}
int main(){
	scanf("%d%d%d%d",&n,&m,&v,&e);
	FOR(i,1,n)scanf("%d",&c[i]);
	FOR(i,1,n)scanf("%d",&d[i]);
	FOR(i,1,n)scanf("%lf",&k[i]);
	FOR(i,1,v)FOR(j,1,v)dis[i][j]=(db)INF;
	FOR(i,1,v)dis[0][i]=0,dis[i][0]=0,dis[i][i]=0;
	while(e--){
		scanf("%d%d%d",&x,&y,&z);
		dis[x][y]=min(dis[x][y],z),dis[y][x]=min(dis[x][y],z);
	}
	floyd();
	FOR(i,0,n)FOR(j,0,m)f[i][1][j]=f[i][0][j]=(db)INF;
	f[1][0][0]=0.0;
	f[1][1][1]=0.0;
	
	FOR(i,2,n) 
        f[i][0][0]=f[i-1][0][0]+(db)dis[c[i-1]][c[i]];
	FOR(j,1,m)
		FOR(i,2,n){
			f[i][0][j]=f[i-1][0][j]+dis[c[i-1]][c[i]];
			cmp=(db)(f[i-1][1][j]+(1.00-k[i-1])*(db)dis[c[i-1]][c[i]]+k[i-1]*(db)dis[d[i-1]][c[i]]);
			if((db)cmp<(db)f[i][0][j])f[i][0][j]=(db)cmp;	
			f[i][1][j]=f[i-1][0][j-1]+k[i]*(db)dis[c[i-1]][d[i]]+(1.00-k[i])*(db)dis[c[i-1]][c[i]];
			cmp=(db)(f[i-1][1][j-1]+k[i-1]*k[i]*(db)dis[d[i-1]][d[i]]+(1.00-k[i-1])*k[i]*(db)dis[c[i-1]][d[i]]+k[i-1]*(1.00-k[i])*(db)dis[d[i-1]][c[i]]+(1.00-k[i-1])*(1.00-k[i])*(db)dis[c[i-1]][c[i]]);
			if((db)f[i][1][j]>(db)cmp)f[i][1][j]=(db)cmp;
		}
	ans=(db)INF;
	FOR(i,0,m){
		if(f[n][0][i]<ans)ans=f[n][0][i];
		if(f[n][1][i]<ans)ans=f[n][1][i];
	}
	printf("%.2lf",ans);
	return 0;
}