[hdu5468]Puzzled Elena

考虑对一个集合求莫比乌斯反演的过程，最终答案即$\sum\limits_{d|ai}\mu(d)\cdot calc(d)$（calc(d)表示i子树中d倍数的个数）
dfs时维护当前遍历到的点中i倍数的个数，对i搜之前和i搜完后差分一下就可以得到i子树的信息，用上述式子计算即可

#include<bits/stdc++.h>
using namespace std;
#define N 100005
#define K 130
struct ji{
    int nex,to;
}edge[N<<1];
int E,t,n,m,x,y,a[N],head[N],mu[N],vis[N],p[N],sum[N],ans[N],v[N][K],s[N][K];
void add(int x,int y){
    edge[E].nex=head[x];
    edge[E].to=y;
    head[x]=E++;
}
void dfs(int k,int fa){
    for(int i=1;i<=v[k][0];i++)s[k][i]=sum[v[k][i]]++;
    for(int i=head[k];i!=-1;i=edge[i].nex)
        if (edge[i].to!=fa)dfs(edge[i].to,k);
    for(int i=1;i<=v[k][0];i++)ans[k]+=mu[v[k][i]]*(sum[v[k][i]]-s[k][i]);
}
int main(){
    mu[1]=1;
    for(int i=2;i<N-4;i++){
        if (!vis[i]){
            mu[i]=-1;
            p[++p[0]]=i;
        }
        for(int j=1;(j<=p[0])&&(i*p[j]<N-4);j++){
            vis[i*p[j]]=1;
            if (i%p[j]==0)break;
            mu[i*p[j]]=-mu[i];
        }
    }
    while (scanf("%d",&n)!=EOF){
        E=0;
        memset(head,-1,sizeof(head));
        memset(v,0,sizeof(v));
        memset(s,0,sizeof(s));
        memset(sum,0,sizeof(sum));
        memset(ans,0,sizeof(ans));
        for(int i=1;i<n;i++){
            scanf("%d%d",&x,&y);
            add(x,y);
            add(y,x);
        }
        for(int i=1;i<=n;i++)scanf("%d",&a[i]);
        for(int i=1;i<=n;i++){
            for(int j=1;j*j<=a[i];j++)
                if (a[i]%j==0)v[i][++v[i][0]]=j;
            for(int j=v[i][0];j;j--)
                if (v[i][j]*v[i][j]<a[i])v[i][++v[i][0]]=a[i]/v[i][j];
        }
        dfs(1,0);
        printf("Case #%d:",++t);
        for(int i=1;i<=n;i++)printf(" %d",ans[i]);
        printf("\n");
    }
}