HDU6035（数学题）

很多博客说这是树形DP，菜菜的我并不懂。
这个题需要逆向思考，用总的可能数，减去对总的贡献不足的的同色连通块，就得到答案了，这个题难在树上连通块的搜索。
刚刚学会，还不是那么精通，直接上代码
数组的用处都写进去了，可以跟着程序跑跑，就能得到答案了，跑了一中午的我是个弱智。。。。
感谢鹏哥的代码
code：

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
using namespace std;
#define LL long long
const int maxn=2000000+10;
int c[maxn];
int sum[maxn];
int siz[maxn];
int vis[maxn];
int n;
LL ans;
int case_=1;
vector<int> e[maxn];
LL path(int a)
{
    return (LL)a*(LL)(a-1)/2;
}
void dfs(int u,int fa)
{
    siz[u]=1;
    int addsum=0;
    for(auto &v:e[u])//遍历e[u]容器的意思
    {
        if(v==fa)
            continue;
        int oldsum=sum[c[u]];//未递归之前和他颜色相同的点的次数
        dfs(v,u);
        int add=sum[c[u]]-oldsum;//递归之后产生的差值
        ans+=path((LL)(siz[v]-add));
        addsum+=add;
        siz[u]+=siz[v];//siz[u]现在放的是u的遍历过的子树所有节点的数目
    }
    sum[c[u]]+=siz[u]-addsum;//sum[]存放的是遍历到当前位置和他相同颜色且为根的子树大小和
}
void work()
{
    ans=0;
    memset(sum,0,sizeof(sum));
    memset(siz,0,sizeof(siz));
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
    {
        e[i].clear();
        scanf("%d",&c[i]);
        vis[c[i]]=1;
    }
    int num=0;
    for(int i=1;i<=n;i++)
    {
        num+=(vis[i]>0);
    }
    int u,v;
    for(int i=1;i<n;i++)
    {
        scanf("%d %d",&u,&v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    memset(vis,0,sizeof(vis));
    dfs(1,-1);
    vis[c[1]]=1;
    for(int i=2;i<=n;i++)
    {
        if(!vis[c[i]])
        {
            ans+=path((LL)n-sum[c[i]]);
            vis[c[i]]=1;
        }
    }
     LL myans=path((LL)n)*(LL)num-ans;
     printf("Case #%d: %lld\n",case_++,myans);
}
int main()
{

    while(~scanf("%d",&n))
    {
        work();
    }
    return 0;
}