BZOJ 3573 米特运输

http://www.lydsy.com/JudgeOnline/problem.php?id=3573

思路:如果固定了一个位置的米特容量，那其他位置的米特容量也就是确定的，我们分别假设当前n个装置都是固定的，然后这个装置的所有祖先的度数之积乘上当前装置的容量，就等于确定当前米特容量，得到根节点的米特容量，这样我们只要分别固定n个装置，算出根节点的容量，比较有几个根节点容量是相等的，用n减去即是答案。

#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
const double eps=1e-5;
double s[500005];
int tot,go[1000005],next[1000005],first[1000005],du[500005];
int a[500005],n;
int read(){
    char ch=getchar();int t=0,f=1;
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while ('0'<=ch&&ch<='9'){t=t*10+ch-'0';ch=getchar();}
    return t*f;
}
void insert(int x,int y){
    tot++;
    go[tot]=y;
    next[tot]=first[x];
    first[x]=tot;
}
void add(int x,int y){
    insert(x,y);insert(y,x);
}
void dfs(int x,int fa){
    for (int i=first[x];i;i=next[i]){
        int pur=go[i];
        if (pur==fa) continue;
        s[pur]=s[x]+log(du[x]);
        dfs(pur,x);
    }
}
int main(){
    n=read();
    for (int i=1;i<=n;i++)
     a[i]=read();
    for (int i=1;i<n;i++){
        int x=read(),y=read();
        add(x,y);du[x]++;du[y]++;
    } 
    for (int i=2;i<=n;i++)
     du[i]--;
    s[1]=log(1);
    dfs(1,0);
    for (int i=1;i<=n;i++)
     s[i]+=log(a[i]);
    std::sort(s+1,s+n+1);
    int tmp=1,ans=0;
    for (int i=2;i<=n;i++)
     if (fabs(s[i]-s[i-1])<=eps) tmp++;
     else ans=std::max(ans,tmp),tmp=1;
    ans=std::max(ans,tmp);
    printf("%d\n",n-ans); 
}