3573: [Hnoi2014]米特运输

题目链接

题目大意：给定一棵树和每个点的权值，问最少改动多少个点的权值使得：1.每个点的所有儿子权值相等，2.每个点的权值等于所有儿子权值和

题解：确定一个点的权值就可以确定整棵树的权值，枚举每个点不变，计算所有点权与原先比较，但是这样是n^2的

仔细想想会发现如果A点权值不变时B点权值也不变，那么当A或B权值不变时根节点的权值是相同的

只需要算出个点不变时根节点的权值，取一下众数，ans=n-众数出现次数

但是乘起来会爆long long，取log，注意精度

我的收获：……

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int M=500005;
#define eps 1e-8

int n,t,maxn;
int a[M],head[M],sz[M];
double k[M],tmp[M];

struct edge{int to,nex;}e[M<<1];

void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}

void dfs(int x,int fa)
{
    for(int i=head[x];i!=-1;i=e[i].nex)
    if(e[i].to!=fa) sz[x]++;
    for(int i=head[x];i!=-1;i=e[i].nex)
    if(e[i].to!=fa) k[e[i].to]=log(sz[x])+k[x],dfs(e[i].to,x);
}

void work()
{
    for(int i=1;i<=n;i++) tmp[i]=log(a[i])+k[i];
    sort(tmp+1,tmp+n+1);
    double last=0;int num=0;
    for(int i=1;i<=n;i++)
    if(tmp[i]-last<eps) num++,maxn=max(maxn,num);
    else last=tmp[i],num=1;
    printf("%d\n",n-maxn);
}

void init()
{
    cin>>n;
    int x,y;memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    k[1]=log(1);dfs(1,0);
}

int main()
{
    init();
    work();
    return 0;
}