标签:本质 mod ace == highlight min ons turn 分解
题意:给一棵带点权$w_i$的树,多次询问$(u,v,x)$,求出$\prod\limits_{i\in\text{path}(u,v)}(w_i,x)$
因为是乘法,所以可以把路径询问拆成到根询问,这样就可以离线做了
因为求$\gcd$的本质是质因数指数取$\min$,所以在离线dfs时每到一个点就把它的点权质因数分解打上标记然后统计答案即可
具体地,对于$w_x=\prod\limits_{i=1}^kp_i^{a_i}$,我们把每个$p_i$的$1\cdots a_i$次幂乘上$p_i$的标记,统计答案时对$x=\prod\limits_{i=1}^kp_i^{a_i}$,对每个$p_i$都往答案乘上$[1,a_i]$的标记即可,dfs退栈时除回去以撤销
#include<stdio.h> #include<vector> using namespace std; typedef long long ll; const int mod=1000000007,T=10000000; int mul(int a,int b){return a*(ll)b%mod;} int pr[10000010],d[10000010],s[10000010]; int pow(int a,int b){ int s=1; while(b){ if(b&1)s=mul(s,a); a=mul(a,a); b>>=1; } return s; } void sieve(){ int i,j,M; M=0; d[1]=1; s[1]=1; for(i=2;i<=T;i++){ s[i]=1; if(d[i]==0){ M++; pr[M]=d[i]=i; } for(j=1;j<=M;j++){ if(pr[j]*(ll)i>T)break; d[i*pr[j]]=pr[j]; if(i%pr[j]==0)break; } } } int h[100010],nex[200010],to[200010],dep[100010],fa[100010][17],M; void add(int a,int b){ M++; to[M]=b; nex[M]=h[a]; h[a]=M; } void dfs(int x){ dep[x]=dep[fa[x][0]]+1; for(int i=h[x];i;i=nex[i]){ if(to[i]!=fa[x][0]){ fa[to[i]][0]=x; dfs(to[i]); } } } int lca(int x,int y){ int i; if(dep[x]<dep[y])swap(x,y); for(i=16;i>=0;i--){ if(dep[fa[x][i]]>=dep[y])x=fa[x][i]; } if(x==y)return x; for(i=16;i>=0;i--){ if(fa[x][i]!=fa[y][i]){ x=fa[x][i]; y=fa[y][i]; } } return fa[x][0]; } struct par{ int x,f; par(int a=0,int b=0){x=a;f=b;} }; vector<par>v[100010]; int a[100010],val[100010],ans[100010]; void solve(int x){ int i,j,t; for(i=a[x];i>1;){ for(j=t=d[i];i%t==0;i/=t,j*=t)s[j]=mul(s[j],t); } for(par p:v[x]){ for(i=val[p.x];i>1;){ for(j=t=d[i];i%t==0;i/=t,j*=t)ans[p.x]=mul(ans[p.x],p.f?s[j]:pow(s[j],mod-2)); } } for(i=h[x];i;i=nex[i]){ if(to[i]!=fa[x][0])solve(to[i]); } for(i=a[x];i>1;){ for(j=t=d[i];i%t==0;i/=t,j*=t)s[j]=mul(s[j],pow(t,mod-2)); } } int main(){ sieve(); int n,q,i,j,x,y; scanf("%d",&n); for(i=1;i<n;i++){ scanf("%d%d",&x,&y); add(x,y); add(y,x); } for(i=1;i<=n;i++)scanf("%d",a+i); dfs(1); for(j=1;j<17;j++){ for(i=1;i<=n;i++)fa[i][j]=fa[fa[i][j-1]][j-1]; } scanf("%d",&q); for(i=1;i<=q;i++){ scanf("%d%d%d",&x,&y,val+i); j=lca(x,y); v[x].push_back(par(i,1)); v[y].push_back(par(i,1)); v[j].push_back(par(i,0)); v[fa[j][0]].push_back(par(i,0)); ans[i]=1; } solve(1); for(i=1;i<=q;i++)printf("%d\n",ans[i]); }
标签:本质 mod ace == highlight min ons turn 分解
原文地址:https://www.cnblogs.com/jefflyy/p/9112929.html