cf280 C. Game on Tree

链接

点击跳转

题解

算贡献,每个点肯定会被删掉,但是可能是和祖先一起删掉,也可能是被自己删掉

如果和祖先一起删掉,那么对答案的贡献就是 0 0 0,否则就是 1 1 1

而后者的概率是 1 d e p t h [ u ] \frac{1}{depth[u]} depth[u]1

所以答案就成了

∑ i = 1 n 1 d e p t h [ i ] \sum_{i=1}^n \frac{1}{depth[i]} i=1ndepth[i]1

代码

#include 
#include 
#include 
#define iinf 0x3f3f3f3f
#define linf (1ll<<60)
#define eps 1e-8
#define maxn 1000010
#define maxe 1000010
#define cl(x) memset(x,0,sizeof(x))
#define rep(_,__) for(_=1;_<=(__);_++)
#define em(x) emplace(x)
#define emb(x) emplace_back(x)
#define emf(x) emplace_front(x)
#define fi first
#define se second
#define de(x) cerr<<#x<<" = "<
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ll read(ll x=0)
{
    ll c, f(1);
    for(c=getchar();!isdigit(c);c=getchar())if(c=='-')f=-f;
    for(;isdigit(c);c=getchar())x=x*10+c-0x30;
    return f*x;
}
int depth[maxn], n;
vector<int> to[maxn];
void dfs(int u, int fa)
{
    depth[u]=depth[fa]+1;
    for(auto v:to[u])
        if(v!=fa)
            dfs(v,u);
}
int main()
{
    int i, u, v;
    n=read();
    rep(i,n-1)
    {
        u=read(), v=read();
        to[u].emb(v), to[v].emb(u);
    }
    dfs(1,0);
    double ans=0;
    rep(i,n)ans+=double(1)/depth[i];
    printf("%.8lf",ans);
    return 0;
}

你可能感兴趣的:(#,数学杂题)