[bzoj1415][Noi2005]聪聪和可可 期望DP+记忆化搜索 & bzoj100题

1415: [Noi2005]聪聪和可可

Time Limit: 10 Sec   Memory Limit: 162 MB
[ 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

看Po神的题解，写出来WA了，发现bfs时定义变量有毒，为什么要写short才对，bfs里面memset跑的慢，重新设变量我竟然不知道要用static赋0。

我写了这么多次bfs，就是不知道为什么这里要声明short，还是没有弄明白这是什么意思，求解。

然后这道题就是先预处理出两点分别在一种位置上的下一个点走哪里，然后再处理分别在某一位置时的步数

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N = 1000 + 5;
int g[N][N],n,m,last[N*2],cnt,out[N],S,T;
double f[N][N];
struct Edge{
	int to,next;
}e[N*N];
void insert( int u, int v ){
	out[u]++; e[++cnt].to = v; e[cnt].next = last[u]; last[u] = cnt;
	out[v]++; e[++cnt].to = u; e[cnt].next = last[v]; last[v] = cnt;
}
void bfs( int u ){
	static int q[N],d[N],to[N],inq[N];
	short head=0,tail=0;//搞不懂，用int就死了
	memset(d,0x3f,sizeof(d));
	d[u] = 0; q[++tail] = u;
	while( head != tail ){
		int now = q[++head]; if( head >= N ) head = 0;
		for( int i = last[now]; i; i = e[i].next )
			if( d[e[i].to] > d[now] + 1 || d[e[i].to] == d[now] + 1 && to[e[i].to] > now ){
				d[e[i].to] = d[now] + 1;
				to[e[i].to] = now;
				if( !inq[e[i].to] ){ inq[e[i].to] = 1, q[++tail] = e[i].to; if( tail >= N ) tail = 0; }
			}
		inq[now] = 0;
	}
	for( int i = 1; i <= n; i++ ) g[i][u] = to[i];
}
double find( int x, int y ){
	if( x == y ) return f[x][y] = 0;
	if( g[x][y] == y || g[g[x][y]][y] == y ) return f[x][y] = 1;
	if( f[x][y] >= -1e-8 ) return f[x][y];
	int to = g[g[x][y]][y]; double res = 1;
	for( int i = last[y]; i; i = e[i].next ) res += find(to,e[i].to)/(out[y]+1);
	res += find(to,y)/(out[y]+1);
	return f[x][y] = res;
}
int main(){
	scanf("%d%d%d%d", &n, &m, &S, &T);
	for( int i = 1, u, v; i <= m; i++ ) scanf("%d%d", &u, &v ), insert( u, v );
	for( int i = 1; i <= n; i++ ) bfs(i);
	memset(f,0xc1,sizeof(f));
	printf("%.3lf",find(S,T));
	return 0;
}