bzoj 1415: [Noi2005]聪聪和可可（概率与期望）

最新推荐文章于 2018-10-04 15:45:30 发布

clover_hxy

最新推荐文章于 2018-10-04 15:45:30 发布

阅读量371

点赞数

分类专栏：概率与期望

本文链接：https://blog.csdn.net/clover_hxy/article/details/54317472

版权

概率与期望专栏收录该内容

40 篇文章 0 订阅

订阅专栏

1415: [Noi2005]聪聪和可可

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 1469 Solved: 870
[ Submit][ Status][ Discuss]

Description

Input

数据的第1行为两个整数N和E，以空格分隔，分别表示森林中的景点数和连接相邻景点的路的条数。第2行包含两个整数C和M，以空格分隔，分别表示初始时聪聪和可可所在的景点的编号。接下来E行，每行两个整数，第i+2行的两个整数Ai和Bi表示景点Ai和景点Bi之间有一条路。所有的路都是无向的，即：如果能从A走到B，就可以从B走到A。输入保证任何两个景点之间不会有多于一条路直接相连，且聪聪和可可之间必有路直接或间接的相连。

Output

输出1个实数，四舍五入保留三位小数，表示平均多少个时间单位后聪聪会把可可吃掉。

Sample Input

【输入样例1】
4 3
1 4
1 2
2 3
3 4
【输入样例2】
9 9
9 3
1 2
2 3
3 4
4 5
3 6
4 6
4 7
7 8
8 9

Sample Output

【输出样例1】
1.500
【输出样例2】
2.167

HINT

【样例说明1】
开始时，聪聪和可可分别在景点1和景点4。
第一个时刻，聪聪先走，她向更靠近可可(景点4)的景点走动，走到景点2，然后走到景点3；假定忽略走路所花时间。
可可后走，有两种可能：
第一种是走到景点3，这样聪聪和可可到达同一个景点，可可被吃掉，步数为1，概率为。
第二种是停在景点4，不被吃掉。概率为。
到第二个时刻，聪聪向更靠近可可(景点4)的景点走动，只需要走一步即和可可在同一景点。因此这种情况下聪聪会在两步吃掉可可。
所以平均的步数是1* +2* =1.5步。

对于所有的数据，1≤N,E≤1000。
对于50%的数据，1≤N≤50。

Source

[ Submit][ Status][ Discuss]

题解：概率与期望

p[i][j]表示聪聪在i,可可在j，聪聪下一步会到达的节点（就是i,j最短路径中与i相邻的编号最小的节点），可以做n遍bfs，预处理出p数组

w[i][j]表示与i相邻编号第j大的节点

t[i]表示节点i的度

考虑怎么转移，设f[i][j]表示聪聪在i,可可在j的期望步数

首先如果p[i][j]=j,或者p[p[i,j],j]=j（注意题目描述中提到如果聪聪无法一步到达可可，那么他可以多走一步）的话，聪聪就可以再经过一步到达，所以f[i][j]=1

如果i=j，那么f[i][j]=0

因为正着推不好转移所以我们考虑逆着转移

f[i][j]={(1/(t[j]+1))[sigma(k=1..w[j][0])f[p[p[i][j]][j],w[j][k]] + f[p[p[i][j]][j],j]}+1

为什么是p[p[i][j]][j]，而不是p[p[i][j]][w[j][k]]呢？

因为我们要求的是聪聪吃到可可的期望步数，我们每次都是根据可可所在的位置来确定聪聪下一步的走向。

所以f[i][j]更准确的理解是可可到达j这个点后，聪聪进行移动的期望步数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define N 1003
using namespace std;
int p[N][N],w[N][N],t[N],n,m,S,T,dis[N],pre[N];
double f[N][N];
void bfs(int x)
{
	queue<int> q; 
	memset(dis,0,sizeof(dis));
    dis[x]=1;
	for (int i=1;i<=w[x][0];i++){
		dis[w[x][i]]=dis[x]+1;
		pre[w[x][i]]=w[x][i];
		q.push(w[x][i]);
	}
	bool pd=false;
	while (!q.empty()) {
		int now=q.front(); q.pop();
		for (int i=1;i<=w[now][0];i++) 
		if (!dis[w[now][i]]){
			dis[w[now][i]]=dis[now]+1;
			pre[w[now][i]]=pre[now];
            q.push(w[now][i]);
		}
	}
	for (int i=1;i<=n;i++)
	 if (dis[i]) p[x][i]=pre[i];
	 else p[x][i]=0;
	p[x][x]=0;
}
double solve(int i,int j)
{
	if (f[i][j]) return f[i][j]; 
	if (p[i][j]==j||p[p[i][j]][j]==j) {
		f[i][j]=1;
		return f[i][j];
	} 
	if (i==j) {
		f[i][j]=0;
		return f[i][j];
	}
	double tot=0;
	for (int k=1;k<=w[j][0];k++)
	  tot+=solve(p[p[i][j]][j],w[j][k]);
	tot+=solve(p[p[i][j]][j],j);
	tot/=(double)(1+t[j]);
	f[i][j]=tot+1;
	return f[i][j];
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d%d",&n,&m);
	scanf("%d%d",&S,&T);
	for (int i=1;i<=m;i++) {
		int x,y; scanf("%d%d",&x,&y);
		t[x]++; t[y]++;
		w[x][++w[x][0]]=y;
		w[y][++w[y][0]]=x;
	}
	for (int i=1;i<=n;i++) 
	 sort(w[i]+1,w[i]+w[i][0]+1);
	for (int i=1;i<=n;i++) bfs(i);
	/*for (int i=1;i<=n;i++) {
		for (int j=1;j<=n;j++)
		 cout<<i<<" "<<j<<" "<<p[i][j]<<endl;
	}*/
	solve(S,T);
	printf("%.3lf\n",f[S][T]);
}