做了好几个容斥了,一直找不到feel,这个做完在现在有一点感觉了。虽然刚开始也不会。但就是发现感觉不一样了。
首先,不考虑树的关系,单纯给出一个m,还有一个集合(里面数字任意),求集合里面跟m互质的数的个数。这就是典型的容斥题。
那么现在要求得使顶点跟子节点的互质关系,那么原理还是有容斥喽,还是回到一开始那个问题,如何计算跟m互质的数的个数,做法是将m进行质因子分解,然后计算集合中有多少能被m的因子整除,然后容斥加加减减即可。那么对于树,可以采用dfs递归求解。预处理每个数的素因子,在递归过程中计算x顶点的所有子节点的含有因子Y的集合中元素的个数,然后容斥计算即可。但在递归过程中要记录下一步计算的过程中用到的因子在之前一共出现了多少次,最后算的时候计算差值即可。代码不是很难,直接看也能理解。就是那个pre数组用的有点巧妙。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int prime[maxn],cnt;
bool vis[maxn];
vector<int> f[maxn];
void get_prime_fac(){
for(int i=2;i<maxn;i++){
if(!vis[i]){
prime[cnt++]=i;
for(int j=i;j<maxn;j+=i){
f[j].push_back(i);
vis[j]=1;
}
}
}
}
vector<int> g[maxn];
int val[maxn];
int ans[maxn];
int st[maxn];
int en[maxn];
int a[maxn]; //a[x]代表保存该x顶点的子树的因子个数。
int dfs_clock;
int facnum;
vector<int> fac;
inline void get_fac(int x){
fac=f[x];
facnum=fac.size();
}
inline void add(int x){
get_fac(val[x]);
for(int i=1;i<(1<<facnum);i++){
int tmp=1;
for(int j=0;j<facnum;j++){
if(i&(1<<j))
tmp*=fac[j];
}
a[tmp]++;
}
}
void dfs(int x,int fa){
int pre[100];
st[x]=++dfs_clock;
get_fac(val[x]);
for(int i=1;i<(1<<facnum);i++){
int tmp=1;
for(int j=0;j<facnum;j++){
if(i&(1<<j)){
tmp*=fac[j];
}
}
pre[i]=a[tmp]; //保存tmp因子出现之前对应的数量,拿x顶点为例的话,左子树会对计算右子树会产生影响。
}
for(int i=0;i<g[x].size();i++){
int v=g[x][i];
if(v==fa) continue;
dfs(v,x);
}
en[x]=dfs_clock;
get_fac(val[x]);
ans[x]=en[x]-st[x];
for(int i=1;i<(1<<facnum);i++){
int tmp=1,cnt=0;
for(int j=0;j<facnum;j++){
if(i&(1<<j)){
tmp*=fac[j];
cnt^=1;
}
}
if(cnt) ans[x]-=(a[tmp]-pre[i]);
else ans[x]+=(a[tmp]-pre[i]);
}
add(x);
if(val[x]==1) ans[x]++; //这是一个特别注意的地方。
}
int main(){
get_prime_fac();
int n,cs=1;
while(~scanf("%d",&n)){
for(int i=1;i<=n;i++){
g[i].clear();
}
for(int i=1;i<n;i++){
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1;i<=n;i++){
scanf("%d",&val[i]);
}
memset(a,0,sizeof a);
dfs_clock=0;
dfs(1,-1);
printf("Case #%d:",cs++);
for(int i=1;i<=n;i++){
printf(" %d",ans[i]);
}
printf("\n");
}
}
使用莫比乌斯反演,其实写下容斥的计算式子,可以发现其容斥系数其实就是莫比乌斯函数
所以集合中跟m不互素的数的个数
count函数表示集合中出现因子d的元素的个数。
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int mu[maxn],cnt,prime[maxn];
bool vis[maxn];
vector<int> f[maxn];
void get_mu_fac(){
memset(vis,0,sizeof vis);
mu[1]=1;
int tot=0;
for(int i=2;i<maxn;i++){
if(!vis[i]){
prime[tot++]=i;
mu[i]=-1;
}
for(int j=0;j<tot;j++){
if(i*prime[j]>=maxn) break;
vis[i*prime[j]]=1;
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){
f[j].push_back(i);
}
}
}
}
vector<int> g[maxn];
int val[maxn];
int ans[maxn];
int st[maxn];
int en[maxn];
int a[maxn]; //a[x]代表保存该x顶点的子树的因子个数。
int dfs_clock;
int facnum;
vector<int> fac;
inline void get_fac(int x){
fac=f[x];
facnum=fac.size();
}
void dfs(int x,int fa){
int pre[100];
st[x]=++dfs_clock;
get_fac(val[x]);
for(int i=0;i<facnum;i++){
pre[i]=a[fac[i]]; //保存tmp因子出现之前对应的数量,拿x顶点为例的话,左子树会对计算右子树会产生影响。
}
for(int i=0;i<g[x].size();i++){
int v=g[x][i];
if(v==fa) continue;
dfs(v,x);
}
en[x]=dfs_clock;
get_fac(val[x]);
ans[x]=en[x]-st[x];
for(int i=0;i<facnum;i++){
int sum=a[fac[i]]-pre[i];
ans[x]+=mu[fac[i]]*sum;
}
for(int i=0;i<facnum;i++){
a[fac[i]]++;
}
if(val[x]==1) ans[x]++; //这是一个特别注意的地方。
}
int main(){
get_mu_fac();
int n,cs=1;
while(~scanf("%d",&n)){
for(int i=1;i<=n;i++){
g[i].clear();
}
for(int i=1;i<n;i++){
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1;i<=n;i++){
scanf("%d",&val[i]);
}
memset(a,0,sizeof a);
dfs_clock=0;
dfs(1,-1);
printf("Case #%d:",cs++);
for(int i=1;i<=n;i++){
printf(" %d",ans[i]);
}
printf("\n");
}
}