POJ 3156 - Interconnect (概率DP+hash)

题意:给一个图,有些点之间已经连边,现在给每对点之间加边的概率是相同的,问使得整个图连通,加边条数的期望是多少。

此题可以用概率DP+并查集+hash来做。

用dp(i,j,k...)表示当前的每个联通分量的点数分别是i,j,k...(连通分量的个数不固定)时,加边的期望。

这样以dp(i,j,k)为例分析状态转移的过程,dp(i,j,k)=p1*dp(i,j,k)+p2*dp(i+j,k)+p3*dp(i,j+k)+p4*dp(j,i+k)+1。

终止条件是dp(n)=0,因为此时图一定联通,所以期望是0。

这个式子是怎么来的呢?

p1*dp(i,j,k)表示增加了一条边之后原来的联通分量没有改变时的情况,p1表示选中这样的无用边的概率;p2*dp(i+j,k)表示增加了一条连结i和j这两个联通分量的边,p2表示选中i和j之间的边的概率。最后的1表示增加了一条边。

这里我们要计算选边概率。

首先有n*(n-1)/2条边。

如果选中了不改变图的联通分量的边,此时说明每次加的边都属于一个连通分量,所以一共有sum{(cnt(i)*cnt(i)-1)/2}(cnt(i)表示该联通分量内有cnt(i)个点)种选择情况。

如果选中了某条边连结了某两个联通分量i、j,这个时候一共有cnt(i)*cnt(j)种情况。

计算概率只需要除以边的总数即可。

计算连通分量可以用并查集。

因为这个题只看图的连通情况不在乎具体是哪个点,因此可以对每个联通分量的点数排序,使得每个状态唯一,这样可以得到很多重叠的子问题,整个问题就可以用记忆化搜索完成了。

很明显问题的状态比较复杂,直接下手很难处理也容易超时,所以用hash处理状态压缩判重。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#define MOD 100007
using namespace std;
int n,m;
struct State
{
    double val;
    ];
    bool vis;
    State()
    {
        memset(x,,sizeof(x));
        vis=false;
    }
    void clear()
    {
        memset(x,,sizeof(x));
        vis=false;
    }
    void mysort()
    {
        sort(x,x+);
    }
    int myhash() const
    {
        ;
        ,b=; i>=&&x[i]; --i)
        {
            res+=x[i]*b;
            res%=MOD;
            b*=;
            b%=MOD;
        }
        return res;
    }
    bool operator ==(State &t)const
    {
        ; i<; ++i) if(x[i]!=t.x[i]) return false;
        return true;
    }
    bool operator !=(State &t) const
    {
        return !(*this==t);
    }
};
State has[MOD+],st;
void inserthash(State &p)
{
    int v=p.myhash();
    while(has[v].vis)
        ;
    has[v]=p;
    has[v].vis=true;
}
double gethash(State &p)
{
    int v=p.myhash();
    while(has[v].vis&&has[v]!=p)
        ;
    ;
}
];
int find(int p)
{
    return p==father[p]?p:(father[p]=find(father[p]));
}
void init()
{
    ; i<=n; ++i) father[i]=i;
    st.clear();
}
double DP(State &s)
{
    ;
    double v=gethash(s);
    if(v!=-1.0) return v;
    ,sn=n*(n-)/2.0;;
    ; i>=&&s.x[i]; --i)
        q+=s.x[i]*(s.x[i]-)/2.0;
    double &res=s.val;
    res=;
    ; i>=&&s.x[i]; --i)
        ; j>=&&s.x[j]; --j)
        {
            double p=s.x[i]*s.x[j]/sn;
            State t;
            ; k>=; --k)
                if(k!=i&&k!=j) t.x[k]=s.x[k];
                else if(k==i)  t.x[k]=s.x[i]+s.x[j];
                ;
            t.mysort();
            res+=p*DP(t);
        }
    res=res/(-q/sn);
    inserthash(s);
    return res;
}
];
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        ; i<m; ++i)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            int fx=find(x),fy=find(y);
            if(fx!=fy) father[fx]=fy;
        }
        memset(cnt,,sizeof(cnt));
        ; i<=n; ++i)
            cnt[find(i)]++;
        ; i<=n; ++i)
            st.x[i-]=cnt[i];
        st.mysort();
        printf("%.6f\n",DP(st));
    }
    ;
}

学习了http://www.cnblogs.com/swm8023/archive/2012/09/21/2696593.html

上一篇:基于PI+QT实现OpenCV图像处理操作(基本环境搭建)


下一篇:解决JSP页面获取的数据库数据乱码问题