bzoj1415 [Noi2005]聪聪和可可(期望概率DP+最短路)

来源:互联网 发布:少儿趣味编程课程 编辑:程序博客网 时间:2024/05/19 13:46

bzoj1415 [Noi2005]聪聪和可可

原题地址:http://www.lydsy.com/JudgeOnline/problem.php?id=1415

题意:
n点m边无向图,给定初始位置两点S,T,每个时间单位S会走最靠近T的点走两步(如果有多条最短路走编号最小的点),T会等概率向周围走或不动(都是1/(degree(T)+1)的概率),求期望多少时间单位后S和T相遇。

数据范围
对于 100% 的数据,1≤N,E≤1000。

题解:
n遍spfa预处理当S在i点,T在j点时,S下一步会走的点。
然后记忆化搜索,f[i][j]表示S在i点,T在j点是,还期望多少时间相遇。
边界: f[i][i]=0
定义u为走了两步之后的i,d为j的度数,v为与j相邻的点。
f[i][j]=(f[u][j]+sum(f[u][v]))/(d+1)+1

代码:

#include<cstdio>#include<iostream>#include<algorithm>#include<cstring>#include<queue>using namespace std;const int N=1005;int n,m,head[N],to[2*N],nxt[2*N],num=0,S,T;deque<int> q;int point[N][N],dis[N];bool inq[N];double f[N][N];void build(int u,int v){    num++;    to[num]=v;    nxt[num]=head[u];    head[u]=num;} void spfa(int s){    memset(inq,0,sizeof(inq));    memset(dis,60,sizeof(dis));    int inf=dis[0];    while(!q.empty()) q.pop_back();    inq[s]=1; dis[s]=0;    q.push_back(s);    while(!q.empty())    {        int u=q.front();        q.pop_front(); inq[u]=0;        for(int i=head[u];i;i=nxt[i])        {            int v=to[i];            if(dis[v]>dis[u]+1)            {                dis[v]=dis[u]+1;                if(!inq[v])                {                    if(q.empty()||dis[v]<dis[q.front()]) q.push_front(v);                    else q.push_back(v);                    inq[v]=1;                }            }        }    }    for(int i=1;i<=n;i++)    {        if(i==s)        {point[i][s]=i;continue;}        int mn=inf; int tmp=-1;        for(int j=head[i];j;j=nxt[j])        {            int v=to[j];            if(dis[v]<mn)            {mn=dis[v]; tmp=v;}                     else if(dis[v]==mn&&tmp>v)            tmp=v;        }        point[i][s]=tmp;    }}double dp(int a,int b){    if(f[a][b]>=0) return f[a][b];    f[a][b]=0.0;    int x=a; int y=b;    x=point[x][y]; x=point[x][y];    if(x==y) return f[a][b]=1.0;    int out=0;    for(int i=head[y];i;i=nxt[i])    {        out++;        int v=to[i];        f[a][b]+=dp(x,v);    }    f[a][b]+=dp(x,y);    f[a][b]=(double)f[a][b]/(out+1);    f[a][b]+=1.0;    return f[a][b];}int main(){    scanf("%d%d",&n,&m);    scanf("%d%d",&S,&T);    for(int i=1;i<=n;i++)    for(int j=1;j<=n;j++)    f[i][j]=-1.0;    for(int i=1;i<=m;i++)    {        int u,v;        scanf("%d%d",&u,&v);        build(u,v); build(v,u);    }    for(int i=1;i<=n;i++)    {spfa(i);f[i][i]=0.0;}    printf("%0.3lf\n",dp(S,T));    return 0;}
原创粉丝点击