BZOJ 2878([Noi2012]-失落的游乐园树DP+出站年轮加+后市展望DP+vector的erase)

Description

放假了，小Z认为呆在家里特别无聊，于是决定一个人去游乐园玩。进入游乐园后，小Z看了看游乐园的地图，发现能够将游乐园抽象成有n个景点、m条道路的无向连通图，且该图中至多有一个环（即m仅仅可能等于n或者n-1）。

小Z如今所在的大门也正好是一个景点。小Z不知道什么好玩，于是他决定，从当前位置出发，每次随机去一个和当前景点有道路相连的景点。而且同一个景点不去两次（包含起始景点）。贪玩的小Z会一直游玩，直到当前景点的相邻景点都已经訪问过为止。小Z全部经过的景点按顺序构成一条非反复路径，他想知道这条路径的期望长度是多少？小Z把游乐园的抽象地图画下来带回了家，但是忘了标哪个点是大门，他仅仅好如果每一个景点都可能是大门（即每一个景点作为起始点的概率是一样的）。同一时候，他每次在选择下一个景点时会等概率地随机选择一个还没去过的相邻景点。

Input

第一行是两个整数n和m，分别表示景点数和道路数。接下来行。每行三个整数Xi, Yi, Wi。分别表示第i条路径的两个景点为Xi, Yi，路径长Wi。全部景点的编号从1至n，两个景点之间至多仅仅有一条道路。

Output

共一行。包括一个实数。即路径的期望长度。

Sample Input

4 3
1 2 3
2 3 1
3 4 4

Sample Output

6.00000000

【例子解释】例子数据中共同拥有6条不同的路径: 路径长度概率
1-->4 8 1/4
2-->1 3 1/8
2-->4 5 1/8
3-->1 4 1/8
3-->4 4 1/8
4-->1 8 1/4
因此期望长度 = 8/4 + 3/8 + 5/8 + 4/8 + 4/8 + 8/4 = 6.00
【评分方法】本题没有部分分，你程序的输出仅仅有和标准答案的差距不超过0.01时，才干获得该測试点的满分，否则不得分。

【数据规模和约定】对于100%的数据，1 <= Wi <= 100。

測试点编号 n m 备注
1 n=10 m = n-1 保证图是链状
2 n=100 仅仅有节点1的度数大于2
3 n=1000 /
4 n=100000 /
5 n=100000 /
6 n=10 m = n /
7 n=100 环中节点个数<=5
8 n=1000 环中节点个数<=10
9 n=100000 环中节点个数<=15
10 n=100000 环中节点个数<=20

HINT

Source

对节点x。考虑向下走和向上走两种情况。期望分别为down[x]和up[x]

在本题中有vector的erase(q.begin(),q.begin()+k)函数使用方法(删除[0,k)区间)

#include<cstdio> #include<cstring> #include<cstdlib> #include<algorithm> #include<functional> #include<iostream> #include<cmath> #include<cctype> #include<ctime> #include<vector> using namespace std; #define For(i,n) for(int i=1;i<=n;i++) #define Fork(i,k,n) for(int i=k;i<=n;i++) #define Rep(i,n) for(int i=0;i<n;i++) #define ForD(i,n) for(int i=n;i;i--) #define RepD(i,n) for(int i=n;i>=0;i--) #define Forp(x) for(int p=pre[x];p;p=next[p]) #define Lson (x<<1) #define Rson ((x<<1)+1) #define MEM(a) memset(a,0,sizeof(a)); #define MEMI(a) memset(a,127,sizeof(a)); #define MEMi(a) memset(a,128,sizeof(a)); #define INF (2139062143) #define F (100000007) #define MAXN (100000+10) #define MAXM (300000+10) long long mul(long long a,long long b){return (a*b)%F;} long long add(long long a,long long b){return (a+b)%F;} long long sub(long long a,long long b){return (a-b+(a-b)/F*F+F)%F;} typedef long long ll; typedef long double ld; int n,m,edge[MAXM],pre[MAXN]={0},next[MAXM]={0},weight[MAXM],size=0; void addedge(int u,int v,int w){edge[++size]=v;weight[size]=w;next[size]=pre[u],pre[u]=size;} void addedge2(int u,int v,int w){addedge(u,v,w),addedge(v,u,w);} long double up[MAXN]={0.0},down[MAXN]={0.0}; int fa[MAXN]={0},son[MAXN]={0}; void dfs_down(int x) { Forp(x) { int v=edge[p]; if (v^fa[x]) { fa[v]=x;son[x]++; dfs_down(v); down[x]+=(ld)(weight[p]+down[v]); } } if (son[x]) down[x]/=(ld)son[x]; } void dfs_up(int x) { ld sum=0.0; Forp(x) { int v=edge[p]; if (v^fa[x]) { sum+=(ld)weight[p]+down[v]; } } if (fa[x]) sum+=up[x]; Forp(x) { int v=edge[p]; if (v^fa[x]) { up[v]=sum-(ld)weight[p]-down[v]; if ((son[x]-(!fa[x]))) up[v]/=(son[x]-(!fa[x])); up[v]+=(ld)weight[p]; dfs_up(v); } } } bool b[MAXN]={0},b2=0; vector<int> q,q2; void print() { /* cout<<q.size()<<endl; Rep(i,q.size()) cout<<q[i]<<' ';cout<<endl; cout<<q2.size()<<endl; Rep(i,q2.size()) cout<<q2[i]<<' ';cout<<endl; */ } void find_circle(int x) { b[x]=1;q.push_back(x); Forp(x) { int v=edge[p]; if (!b[v]) { q2.push_back(weight[p]);find_circle(v); if (b2) return; } else if (q[q.size()-2]^v) { int k=0; while(q[k]^v) k++; print(); q.erase(q.begin(),q.begin()+k); q2.erase(q2.begin(),q2.begin()+k); q2.push_back(weight[p]); b2=1;return; } } q.pop_back();q2.pop_back(); } void circle_down() { int size=q.size(); Rep(i,size) { int x=q[i]; ld sum=0.0; Forp(x) { int v=edge[p]; if ((v^q[(i-1+size)%size])&&v^(q[(i+1)%size])) { fa[v]=x;son[x]++; dfs_down(v); down[x]+=(ld)(weight[p]+down[v]); } } if (son[x]) down[x]/=(ld)son[x]; } } void circle_up() { int size=q.size(); Rep(i,size) { int x=q[i]; //处理环 ld p=1.0/(ld)(2),w_past=0.0; For(j,size-1) { w_past+=q2[(i+j-1)%size]; int v=q[(i+j)%size]; if (j!=size-1) up[x]+=p*son[v]/(ld)(son[v]+1)*(w_past+down[v]); else up[x]+=p*(w_past+down[v]); p/=(ld)son[v]+1; } p=1.0/(ld)(2),w_past=0.0; For(j,size-1) { w_past+=q2[(i-j+size)%size]; int v=q[(i-j+size)%size]; if (j!=size-1) up[x]+=p*son[v]/(ld)(son[v]+1)*(w_past+down[v]); else up[x]+=p*(w_past+down[v]); p/=(ld)son[v]+1; } //处理子节点 ld sum=0.0; Forp(x) { int v=edge[p]; if ((v^q[(i-1+size)%size])&&v^(q[(i+1)%size])) { sum+=(ld)weight[p]+down[v]; } } Forp(x) { int v=edge[p]; if ((v^q[(i-1+size)%size])&&v^(q[(i+1)%size])) { up[v]=sum-(ld)weight[p]-down[v]; up[v]/=(ld)(son[x]+1); up[v]+=(2.0/(ld)(son[x]+1))*up[x]; up[v]+=(ld)weight[p]; dfs_up(v); } } } } int main() { // freopen("bzoj2878.in","r",stdin); // freopen("bzoj2878.out","w",stdout); scanf("%d%d",&n,&m); For(i,m) { int u,v,w; scanf("%d%d%d",&u,&v,&w); addedge2(u,v,w); } // For(i,n) // Forp(i) cout<<i<<' '<<edge[p]<<' '<<weight[p]<<endl; cout.setf(ios::fixed); cout.precision(5); if (m==n-1) { dfs_down(1);dfs_up(1); // For(i,n) cout<<i<<":"<<down[i]<<' '<<up[i]<<endl; ld ans=0.0; For(i,n) if (fa[i]) ans+=((ld)son[i]*down[i]+up[i])/(son[i]+1); else ans+=down[i]; ans/=(ld)n; cout<<ans<<endl; } else { find_circle(1); print(); circle_down(); circle_up(); // For(i,n) cout<<i<<':'<<down[i]<<' '<<up[i]<<endl; MEM(b); Rep(i,q.size()) b[q[i]]=1; ld ans=0.0; For(i,n) if (b[i]) ans+=2.0/(ld)(son[i]+2)*up[i]+(ld)(son[i])/(ld)(son[i]+2)*down[i]; else ans+=1.0/(ld)(son[i]+1)*up[i]+(ld)(son[i])/(ld)(son[i]+1)*down[i]; ans/=(ld)n; cout<<ans<<endl; } return 0; }