题目大意:
输入n个点m条边的无向图G,每次随机添加一条非自环的边,求使G连通的期望加边次数。
添加每条边的概率是相等的,加完之后可能出现重边,即在两个点之间有多条边连接。n<=30,m<=1000。
既然提到连通,那么一定有连通块,求连通块的最快办法就是dfs。首先来一个dfs,可以求出有多少连通块,以及每个连通块的点个数。然后我们可以想办法来计算。
设dp(S)为图的连通状态为S时期望加边次数。S怎么表示我们暂且不管,先尝试得到dp方程。设
S
=
{
a
0
,
a
1
,
a
2
,
.
.
.
,
a
k
−
1
,
a
k
}
S=\{a_0,a_1,a_2,...,a_{k-1},a_k\}
S={a0,a1,a2,...,ak−1,ak}——我们知道了S应该表示为一个序列——,
s
i
=
{
a
0
,
a
1
,
.
.
.
,
a
x
+
a
y
,
.
.
.
,
a
k
−
1
}
s_i=\{a_0,a_1,...,a_x+a_y,...,a_{k-1}\}
si={a0,a1,...,ax+ay,...,ak−1},即si是S中删除两个元素,然后加起来再加进去后的序列。设P为加边后出现重边的概率,pi为转移到si的概率,则
d
p
(
S
)
=
(
(
∑
i
=
0
n
−
1
d
p
(
s
i
)
∗
p
i
)
+
1
)
/
(
1
−
p
)
dp(S)=((\displaystyle\sum_{i=0}^{n-1}dp(s_i)*p_i)+1)/(1-p)
dp(S)=((i=0∑n−1dp(si)∗pi)+1)/(1−p)(注:这个n不是上文中输入的n,而是状态S可以转移到的状态总数,状态编号从0记起)。经过计算,可以得知
P
=
(
∑
i
=
0
k
−
1
C
a
i
2
)
/
C
n
2
P=(\displaystyle\sum_{i=0}^{k-1}C_{a_i}^2)/C_n^2
P=(i=0∑k−1Cai2)/Cn2,
p
i
=
(
a
x
∗
a
y
)
/
C
n
2
p_i=(a_x*a_y)/C_n^2
pi=(ax∗ay)/Cn2。由于
C
x
2
C_x^2
Cx2类型的算式被多次使用,建议单独为C(x,2)编写一个函数,为C(x)。
当然,不可能用一个数组或vector作为记忆化数组的下标,直接用map<vector< int >,double>估计连map带vector就把堆溢了。于是就用紫书第五章的题目“集合栈计算机”的技巧,把vector用map映射到数字。
再开个记忆化数组double d[],就可以让速度飞快地往上涨。
网上有人说这道题一定要用哈希表,我本来打算用这个STL程序当“跳板”,确认主算法正确,再尝试换成哈希表的(这个方法详见紫书7.5节),结果意外一提就AC!哪里有用什么哈希表!
代码如下:
#include <iostream>
#include <cstring>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
int g[33][33],vis[33],cnt[33],n,m,k;
map<vector<int>,int>ids;
vector<vector<int> >vs;
double d[3333333];//懒得算有多少个状态了,搞个大数应该不会越界
int ID(vector<int>v){
if(!ids.count(v)){
ids[v]=vs.size();
vs.push_back(v);
}
return ids[v];
}
void dfs(int x,int id){
if(vis[x])return;
vis[x]=id;cnt[id]++;
for(int i=1;i<=n;i++)
if(g[x][i])dfs(i,id);
}
double C(int x){return double(x)*(x-1)/2;}
double dp(int id){
if(d[id])return d[id];
vector<int>v=vs[id];
if(v.size()==1)return 0;
double cn=C(n),ci=0;
for(int i=0;i<v.size();i++)ci+=C(v[i]);
double p=ci/cn,seg=0;
for(int i=0;i<v.size();i++)
for(int j=i+1;j<v.size();j++){
double pi=(v[i]*v[j])/cn;
vector<int> tmp=v;
tmp.erase(tmp.begin()+i);tmp.erase(tmp.begin()+j-1);
tmp.push_back(v[i]+v[j]);
sort(tmp.begin(),tmp.end());
int id=ID(tmp);
seg+=dp(id)*pi;
}
seg=(seg+1)/(1-p);
return d[id]=seg;
}
int main(){
cout<<fixed;cout.precision(6);
while(cin>>n>>m){k=1;
memset(g,0,sizeof(g));
memset(vis,0,sizeof(vis));
memset(cnt,0,sizeof(cnt));
memset(d,0,sizeof(d));
for(int i=0,x,y;i<m;i++)
{cin>>x>>y;g[x][y]=g[y][x]=1;}
for(int i=1;i<=n;i++)
if(!vis[i])dfs(i,k++);
sort(cnt+1,cnt+k);
vector<int>v(cnt+1,cnt+k);
int d=ID(v);
cout<<dp(d)<<endl;
}
return 0;
}