题意:
每个点的子树上有几个节点与该节点的权值互质
思路:
首先要明白怎么求一个集合中有多少个数与n互素
将集合中的数用算术基本定理分解,记录每个数能形成的所有因子的个数,然后将n分解进行容斥,可以求出与n不互素的个数,总数减掉不互素的个数就是互素的数的个数
这个题也一样,只是把集合改成了树上的节点。
当我们dfs遍历树时,一定先访问父节点再访问子树,我们分别在访问父节点时和遍历完子树后再回到父节点时查询已经遍历过的树中所有与父节点权值不互质的个数,两者相减就是子树中与权值不互素的个数。
预处理出了1e5内所有数的因子
#include <iostream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define max_ 100010
#define mod 1000000007
#define inf 0x3f3f3f3f
bool vis[max_];
int prime[max_];
int pl=0;
void getprime()
{
for(int i=2;i<max_;i++)
{
if(vis[i]==false)
prime[++pl]=i;
for(int j=1;j<=pl&&prime[j]*i<max_;j++)
{
vis[i*prime[j]]=true;
if(i%prime[j]==0)
break;
}
}
}
int casnum=1;
vector< vector<int> >v;
vector<int>num[100010];
int n;
int cnt[max_];
int val[max_];
int ans[max_];
void brk(int n)
{
int tmp=n;
for(int i=1;i<=pl&&prime[i]*prime[i]<=n;i++)
{
if(n%prime[i]==0)
{
num[tmp].push_back(prime[i]);
while(n%prime[i]==0)
n/=prime[i];
}
if(n==1)
return;
}
if(n>1)
num[tmp].push_back(n);
}
int cal(int w,int k)
{
int ans=0;
for(int i=1;i<(1<<num[w].size());i++)
{
int sum=1;
int cct=0;
for(int j=0;j<num[w].size();j++)
{
if(i&(1<<j))
{
cct++;
sum*=num[w][j];
}
}
if(cct&1)
ans+=cnt[sum];
else
ans-=cnt[sum];
cnt[sum]+=k;
}
return ans;
}
int dfs(int u,int fa)
{
int pre=cal(val[u],0);
int sum=0;
for(int i=0;i<v[u].size();i++)
{
if(v[u][i]!=fa)
{
sum+=dfs(v[u][i],u);
}
}
int eend=cal(val[u],1);
ans[u]=sum-(eend-pre);
if(val[u]==1)
ans[u]++;
return sum+1;
}
int main(int argc, char const *argv[]) {
getprime();
for(int i=1;i<=100000;i++)
brk(i);
while(scanf("%d",&n)!=EOF)
{
v.clear();
v.resize(n+1);
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
}
for(int i=1;i<=n;i++)
scanf("%d",&val[i]);
dfs(1,-1);
printf("Case #%d:",casnum++);
for(int i=1;i<=n;i++)
printf(" %d",ans[i]);
printf("\n");
}
return 0;
}