LightOJ-1287 Where to Run

状压dp求期望，难点就在于怎么判断下一个点是不是EJ

题目说只会走EJ点，代表只有能走遍所有点的路径才会有概率走

如果图不连通，就直接不走，输出0

dfs搜索时，只有在搜到所有点都走过的情况返回一个非0值，代表可以走通

如果没有走遍所有点且没有其他EJ点，返回0，代表这个点不是EJ

中途假如接下来有cnt个EJ点，他们的期望加起来是sum，再加上原地待5分钟的情况

总期望为(sum+5.0)/cnt，因为待五分钟后可能会继续待

最后答案需要减去返回的那个非0值

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=20;
const int M=33000;
double d[M][N];
bool vis[M][N];
struct Edge
{
    int to,nxt;
    double w;
}edge[N*N];
int tot;
int first[N];
void addedge(int u,int v,double w)
{
    edge[tot].to=v;
    edge[tot].w=w;
    edge[tot].nxt=first[u];
    first[u]=tot++;
}
double dp(int u,int p,int md)
{
    if(u==md) return 5.0;
    if(vis[u][p]) return d[u][p];
    vis[u][p]=true;
    d[u][p]=0.0;
    bool num[N];
    memset(num,false,sizeof(num));
    int t=u,k=0;
    while(t)
    {
        if(t&1) num[k]=true;
        t>>=1;
        k++;
    }
    int cnt=0;
    double sum=0.0;
    for(int i=first[p];i!=-1;i=edge[i].nxt)
    {
        if(num[edge[i].to]) continue;
        double ret=dp(u+(1<<edge[i].to),edge[i].to,md);
        if(ret!=0.0)
        {
            sum+=edge[i].w+ret;
            cnt++;
        } 
    }
    if(cnt==0) return 0.0;
    d[u][p]=(sum+5.0)/cnt;
    return d[u][p];
}
int main()
{
    int T,kase=0;
    int n,m,a,b;
    double c;
    scanf("%d",&T);
    while(T--)
    {
        tot=0;
        memset(first,-1,sizeof(first));
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%lf",&a,&b,&c);
            addedge(a,b,c);
            addedge(b,a,c);
        }
        memset(vis,false,sizeof(vis));
        double ans=dp(1,0,(1<<n)-1);
        printf("Case %d: %.6f\n",++kase,max(0.0,ans-5.0));
    }
    return 0;
}