HDU - 5468 Puzzled Elena （dfs序+容斥原理/莫比乌斯反演）

题目：给出一棵树,每个点上有权值.然后求每棵子树中与根节点互质 即gcd(a,b)=1 的节点个数.

思路：该题涉及到一个典型问题.问x与S中有多少个数不互素。解决办法是将S中所有元素依次进行两个步骤：①将元素进行质因数分解。②将质因数可能产生的乘积的出现次数加1。记录一下遍历子树之前的cnt值和遍历子树之后的cnt值,作差就是这棵子树的cnt值。将x进行质因数分解，利用容斥原理求解。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
struct node{
    int v,nxt;
}edge[maxn<<1];
int first[maxn],tot;
void add(int u,int v)
{
    edge[++tot].v=v;
    edge[tot].nxt=first[u];
    first[u]=tot;
}
vector<int>pre[maxn];
void init()
{
    for(int n=1;n<maxn;n++)
    {
        int nn=n;
        for(int i=2;i*i<=nn;i++)
        if(nn%i==0)
        {
            pre[n].push_back(i);
            while(nn%i==0)
                nn/=i;
        }
        if(nn>1)
            pre[n].push_back(nn);
    }
}
int n;
int a[maxn],ans[maxn],cnt[maxn],siz[maxn];
int sum;
void DFS(int x,int k,int num,int g)
{
    if(num)
    {
        if(num&1)
            sum+=cnt[g];
        else sum-=cnt[g];
    }
    for(int i=k+1;i<pre[x].size();i++)
        DFS(x,i,num+1,g*pre[x][i]);
}
void updatecnt(int x,int k,int g)
{
    cnt[g]++;
    for(int i=k+1;i<pre[x].size();i++)
        updatecnt(x,i,g*pre[x][i]);
}
void dfs(int u,int fa)
{
    sum=0;DFS(a[u],-1,0,1);
    int sum1=sum;
    for(int i=first[u];i!=-1;i=edge[i].nxt)
    {
        int v=edge[i].v;
        if(v==fa) continue;
        dfs(v,u);
        siz[u]+=siz[v];
    }
    sum=0;DFS(a[u],-1,0,1);
    int sum2=sum;
    ans[u]=siz[u]-(sum2-sum1);
    siz[u]++;
    if(a[u]==1)
        ans[u]++;
    updatecnt(a[u],-1,1);
}
int main()
{
    int cas=0;
    init();
    while(~scanf("%d",&n))
    {
        tot=0;
        for(int i=1;i<=n;i++)
            first[i]=-1;
        for(int i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            add(x,y);
            add(y,x);
        }
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(cnt,0,sizeof cnt);
        memset(ans,0,sizeof ans);
        memset(siz,0,sizeof siz);
        dfs(1,-1);
        printf("Case #%d:",++cas);
        for(int i=1;i<=n;i++)
            printf(" %d",ans[i]);
        puts("");
    }
	return 0;
}

莫比乌斯反演代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
struct node{
    int v,nxt;
}edge[maxn<<1];
int first[maxn],tot;
void add(int u,int v)
{
    edge[++tot].v=v;
    edge[tot].nxt=first[u];
    first[u]=tot;
}
vector<int>pre[maxn];
const int MAXN = 100000;
bool check[MAXN+10];
int prime[MAXN+10];
int mu[MAXN+10];
void Mobius()
{
    memset(check,false,sizeof(check));
    mu[1] = 1;
    int tot = 0;
    for(int i = 2; i <= MAXN; i++)
    {
        if( !check[i] )
        {
            prime[tot++] = i;
            mu[i] = -1;
        }
        for(int j = 0; j < tot; j++)
        {
            if(i * prime[j] > MAXN) break;
            check[i * prime[j]] = true;
            if( i % prime[j] == 0)
            {
                mu[i * prime[j]] = 0;
                break;
            }
            else
            {
                mu[i * prime[j]] = -mu[i];
            }
        }
    }

    for(int i=2;i<=MAXN;i++)
    if(mu[i]!=0)
    {
        for(int j=i;j<=MAXN;j+=i)
        pre[j].push_back(i);
    }
}
int n;
int a[maxn],ans[maxn],cnt[maxn],siz[maxn];

void dfs(int u,int fa)
{
    vector<int>tmp;
    for(int i=0;i<pre[a[u]].size();i++)
        tmp.push_back(cnt[pre[a[u]][i]]);
    for(int i=first[u];i!=-1;i=edge[i].nxt)
    {
        int v=edge[i].v;
        if(v==fa) continue;
        dfs(v,u);
        siz[u]+=siz[v];
    }
    ans[u]=siz[u];
    for(int i=0;i<pre[a[u]].size();i++)
    {
        int d=pre[a[u]][i];
        int sum=cnt[d]-tmp[i];
        ans[u]+=mu[d]*sum;
    }
    if(a[u]==1) ans[u]++;
    siz[u]++;
    for(int i=0;i<pre[a[u]].size();i++)
        cnt[pre[a[u]][i]]++;
}
int main()
{
    int cas=0;
    Mobius();
    while(~scanf("%d",&n))
    {
        tot=0;
        for(int i=1;i<=n;i++)
            first[i]=-1;
        for(int i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            add(x,y);
            add(y,x);
        }
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(cnt,0,sizeof cnt);
        memset(ans,0,sizeof ans);
        memset(siz,0,sizeof siz);
        dfs(1,-1);
        printf("Case #%d:",++cas);
        for(int i=1;i<=n;i++)
            printf(" %d",ans[i]);
        puts("");
    }
	return 0;
}