hdu4670 树分治

这题说的给了一颗树 然后树上有一些整数值，分别由他给的那30个素数组成，有多少条路径的路径上的点的乘积为立方数， 把每个数分解成相应的素数模3后的值，然后压缩为一个3进制的数

然后进行树的分支

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <string.h>
#include <map>
using namespace std;
const int maxn=;
typedef long long LL;
int H[maxn],nx[maxn*],to[maxn*],numofE;
LL prime[],V[maxn],ans;
LL bas[];
int N,K;
LL fenjie(LL a)
{
    LL an=;
    for(int i=; i<K; i++)
    {
         int d=;
         while((a%prime[i])==){
            d++;a/=prime[i];
         }
         an+=bas[i]*(d%);
    }
    return an;
}
void add(int u, int v)
{
    numofE++;
    to[numofE]=v;
    nx[numofE]=H[u];
    H[u]=numofE;
}
int Q[maxn],fa[maxn],subnum[maxn];
LL P[maxn];
bool center[maxn];
int searchroot(int cur)
{
    int rear=;
    fa[cur]=-,Q[rear++]=cur;
    for(int i=; i<rear; i++)
    {
        int x=Q[i];
        for(int j = H[x]; j; j=nx[j])
        {
            int tto = to[j];
            if( tto == fa[x] || center[tto] )continue;
            Q[rear++]=tto; fa[tto]=x;
        }
    }
    int MIN=maxn*,root=cur;
    for(int i=rear-; i>=; i--)
    {
        int x=Q[i];
        subnum[x]=;
        int MA=;
        for(int j=H[x]; j; j=nx[j])
        {
             int tto=to[j];
             if(tto == fa[x] || center[tto] )continue;
             MA=max(MA,subnum[tto]);
             subnum[x]+=subnum[tto];
        }
        MA=max(MA,rear-subnum[x]);
        if(MIN>MA){
            MIN=MA; root=x;
        }
    }
    return root;
}
LL requrenum(LL a)
{
    LL an=;
    for(int i=; i<K; i++)
    {
        LL d=-(a%);
        a/=;
        if(d>)d=;
        an+=d*bas[i];
    }
    return an;
}
LL temp[maxn];
LL jia(LL a, LL b)
{
    LL an=;
    for(int i=; i<K; i++)
    {
        int d=(a% + b%)%;
        a/=;b/=;
        an+=bas[i]*d;
    }
    return an;
}
void count_pair(map<LL,LL>&ds,map<LL,LL>tds,int root)
{
   map<LL,LL>::iterator it;
   it=tds.begin();
   while(it!=tds.end()){
     LL re=requrenum(it->first);
     if(ds.count(re)){
        ans+=ds[re]*(it->second);
     }
     ++it;
   }
   it=tds.begin();
   while(it!=tds.end()){
     LL vv=jia(it->first,V[root]);
     if(ds.count(vv)){
        ds[vv]+=it->second;
     }else
        ds[vv]=it->second;
     it++;
   }
}
void updateedg(int cur, map<LL,LL> &ds)
{
    int rear=;
    fa[cur]=;
    Q[rear++]=cur;
    P[]=V[cur];
    if(ds.count(P[]))ds[P[]]++;
    else ds[P[]]=;
    for(int i=; i<rear; i++)
    {
        int x=Q[i];
        for(int j=H[x]; j; j=nx[j])
        {
            int tto=to[j];
            if(tto==fa[x]||center[tto])continue;
            fa[tto]=x;
            P[rear]=jia(P[i],V[tto]);
            if(ds.count(P[rear]))ds[ P[rear] ]++;
            else ds[ P[rear] ]=;
            Q[rear++]=tto;
        }
    }
}
void dfs(int cur)
{
    int root;
    root=searchroot(cur);
    center[root]=true;
    map<LL,LL>ds,tds;
    ds[V[root]]=;
    for(int i=H[root]; i; i=nx[i])
    {
        int tto=to[i];
        if(center[tto])continue;
        dfs(tto);
        tds.clear();
        updateedg(tto,tds);
        count_pair(ds,tds,root);
    }
    center[root]=false;

    return ;
}
int main()
{
    bas[]=;
    for(int i=;i<=; i++)
        bas[i]=bas[i-]*;
    while(scanf("%d",&N)==)
        {
               scanf("%d",&K);
                numofE=ans=;
               for(int i=; i<K; i++)scanf("%I64d",&prime[i]);
               for(int i=; i<=N; i++){
                    H[i]=;
                    scanf("%I64d",&V[i]);
                    V[i]=fenjie(V[i]);
                    if(V[i]==)ans++;
               }
               for(int i=; i<N; i++)
               {
                   int a,b;
                   scanf("%d%d",&a,&b);
                   add(a,b);
                   add(b,a);
               }
               dfs();
               printf("%I64d\n",ans);
        }
    return ;
}