希望
ADPC2-G
算法:Floyd,排列组合
- 因为给定的两棵树,所以任意两点之间的距离固定,且两点之间一定是一条链,所以可以用Floyd计算出两点之间的距离,并统计链长为x的个数num[x]。(对树进行Floyd有许多优良性质)
- 两棵树所形成的环一定是各取一条链,并将两条链的端点相连,因为m的值给定的很小,所以枚举a树中长度为l1的链,枚举b树中长度为l2的链,并满足l1+l2+2==m进行组合统计num[l1]*num[l2],(这里所求的值应该再乘二,可能会出现头头相连尾尾相连,和头尾相连两种情况)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#define N 400
#define ll long long
using namespace std;
ll numa[10],numb[10],a[N][N],b[N][N];
void init(int n)
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
a[i][j]=b[i][j]=2*N;
a[i][i]=b[i][i]=0;
}
}
int main()
{
//cout<<a[1][1];
int n,m,u,v;
cin>>n>>m;
init(n);
for(int i=1;i<n;i++)
{
cin>>u>>v;
a[u][v]=a[v][u]=1;
}
for(int i=1;i<n;i++)
{
cin>>u>>v;
b[u][v]=b[v][u]=1;
}
//对一棵树求解Floyd可以求解所有边的长度
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
if(a[i][j]>a[i][k]+a[k][j])
a[i][j]=a[i][k]+a[k][j];
if(b[i][j]>b[i][k]+b[k][j])
b[i][j]=b[i][k]+b[k][j];
}
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
if(a[i][j]<=5)numa[a[i][j]]++;
if(b[i][j]<=5)numb[b[i][j]]++;
}
}
double tot=0;
for(int i=1;i<m-2;i++)
{
int j=m-2-i;
tot+=(numa[i]*numb[j]*2);
}
for(int i=1;i<=n;i++)
tot/=i;
printf("%.4lf",tot);
return 0;
}