牛客OI周赛14-普及组 C&D

C：

题解：
找树的重心。

#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int N=2e6+7;
int ne[N],head[N],e[N],cnt,ind[N],sum[N],sz[N],depth,n;
void add(int a,int b)
{
    e[cnt]=b,ne[cnt]=head[a],head[a]=cnt++;
}
void dfs(int fa,int pre)
{
    sz[fa]=1; int Max=0;
    for(int i=head[fa];~i;i=ne[i]){
        int j=e[i];
        if(j==pre) continue;
        dfs(j,fa);
        sz[fa]+=sz[j];
        Max=max(sz[j],Max);
    }
    sum[fa]=max(Max,n-sz[fa]);
}
void dfs(int u,int fa,int de)
{
    depth+=de;
    for(int i=head[u];~i;i=ne[i]){
        int j=e[i];
        if(j==fa) continue;
        dfs(j,u,de+1);
    }
}
signed main()
{
    memset(head,-1,sizeof head);
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int a,b; scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    dfs(1,-1);
    int Min=0x3f3f3f3f,idx;
    for(int i=1;i<=n;i++) if(sum[i]<Min) Min=sum[i],idx=i;
    dfs(idx,-1,0);
    cout<<depth<<endl;
}

D：

题解：
期望DP，我们定义$f(i)$代表从第一道题到第i题所需要的期望。
我们到第i+1题有两种情况，第一种是第i题直接到第i+1道题，或者掉下去并花费$f_i-f_{i-1}$天回来并上去。而第i题到第i+1题期望的天数是$1/a[i]$天。所以代价的期望乘以期望的天数就是答案。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+7;
double f[N],a[N];
int main()
{
    int n; cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%lf",&a[i]);
    }
    f[1]=0.0;
    for(int i=1;i<=n;i++){
        f[i+1]=f[i]+(1.0/a[i])*(1.0+(1.0-a[i])*(f[i]-f[i-1]));
    }
    printf("%.3lf",f[n+1]);
}