BZOJ_3573_[Hnoi2014]米特运输_树形DP+hash

BZOJ_3573_[Hnoi2014]米特运输_树形DP+hash

题意：

给你一棵树每个点有一个权值，要求修改最少的权值，使得每个节点的权值等于其儿子的权值和且儿子的权值都相等。

分析：

首先我们发现在树中如果确定一个点的权值，那么整颗树的方案就能够确定

问题转化成求哪个方案包含的点最多

如何求包含这个点的是哪个方案？

可以给每个点分配一个新的权值

不妨假设1号点的权值不变

1号点的儿子的权值为原来的权值乘上1号点儿子的个数.......以此类推。

发现权值相同的点在一个方案里

由于权值可能很大，我们随缘取模哈希一下就行

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define LL long long
#define N 500050
int head[N],to[N<<1],nxt[N<<1],val[N],cnt;
int n,son[N],dep[N];
LL now[N];
int h[1930010],p=1910009;
int ans,key[1930010];
inline void add(int u,int v)
{
    to[++cnt]=v;nxt[cnt]=head[u];head[u]=cnt;
}
void insert(LL x)
{
    int k=(x%p+p)%p;
    while(h[k]&&key[k]!=x)
    {
        k++;
    }
    h[k]++;key[k]=x;
    ans=max(ans,h[k]);
}
void dfs(int x,int y)
{
    int i;
    for(i=head[x];i;i=nxt[i])
    {
        if(to[i]!=y)
        {
            son[x]++;
        }
    }
    for(i=head[x];i;i=nxt[i])
    {
        if(to[i]!=y)
        {
            now[to[i]]=now[x]*son[x]%p;
            insert(val[to[i]]*now[to[i]]%p);
            dfs(to[i],x);
        }
    }
}
int main()
{
    scanf("%d",&n);
    int i,x,y;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&val[i]);
    }
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    insert(val[1]);
    now[1]=1;
    dfs(1,0);
    printf("%d\n",n-ans);
}