【题解】CF986E Prince's Problem(树上差分+数论性质)
【题解】CF986E Prince's Problem(树上差分+数论性质)
题目大意:
给定你一棵树,有点权\(val_i\le 10^7\)。现在有\(m\)组询问给定参数\(x,y,w\)问你对于\((x->y)\)的路径经过的点集\(P\),问你这个东西:
\[\prod_{u \in P} {\mathrm{gcd}(w,val_u)} \mod 1000000007 \]
考虑这样一种做法:
-
把询问差分成
\[\]\[\] -
考虑这类从根出发如何解决,直接遍历整棵树,对于每个节点先将自己的贡献加入某种数据结构,再递归子树,最后回溯时撤回贡献。问题就变为如何快速求\(\prod \mathrm{gcd}\)
-
考虑\(gcd\)的本质就是对于质因子取\(min\)且求和,而且我们知道\(1e7\)以内的质数是不多的(\(5e6\)左右),所以直接对于每个质数开一个\(vector\)。\(vector_i\)中的下标\(j\)表示唯一分解后\(\alpha(prime[i])=j\)的树上节点有多少个。(\(\alpha\)是唯一分解后的指数)。考虑到\(\sum \alpha\)很小是\(\log\)级别,这样复杂度就是\(O(1e7+(n+m)\log 1e7)\)空间复杂度\(O((n +q)\log 1e7)\)
主要就是树上差分,还有处理\(\sum \min\{...\}\)的方法。
//@winlere
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#define inv(x) (ksm((x)%mod,mod-2))
using namespace std; typedef long long ll;
inline int qr(){
register int ret=0,f=0;
register char c=getchar();
while(c<48||c>57)f|=c==45,c=getchar();
while(c>=48&&c<=57) ret=ret*10+c-48,c=getchar();
return f?-ret:ret;
}
const int mod=1e9+7;
int n,m;
inline int ksm(const int&ba,const int&p){
int ret=1;
for(int t=p,b=ba%mod;t;t>>=1,b=1ll*b*b%mod)
if(t&1) ret=1ll*ret*b%mod;
return ret;
}
vector< vector<int> > s;
namespace PRE{
const int maxn=1e7+5;
int rank[maxn],usd[maxn],arc[maxn],Min[maxn];
vector<int> pr;
inline void pre(const int&n){
usd[1]=1; Min[1]=10;
for(int t=2;t<=n;++t){
if(!usd[t]) Min[t]=t,pr.push_back(t),rank[t]=pr.size(),arc[rank[t]]=t;
for(auto f:pr){
if(1ll*f*t>n)break;
usd[f*t]=1;
Min[f*t]=f;
if(t%f==0)break;
}
}
s.resize(pr.size()+1);
}
}
namespace TREE{
const int maxn=1e5+5;
int d[maxn];
vector<int> e[maxn];
vector<int> q[maxn];
int W[maxn],ans[maxn],w[maxn];
inline void add(const int&fr,const int&to){
e[fr].push_back(to);
e[to].push_back(fr);
}
int son[maxn],top[maxn],r[maxn],siz[maxn];
void pre(const int&now,const int&last){
siz[now]=1;
r[now]=last;
d[now]=d[last]+1;
for(auto t:e[now])
if(t^last) {
pre(t,now); siz[now]+=siz[t];
if(siz[now]>siz[son[now]]) son[now]=t;
}
}
void qaq(const int&now,const int&last){
top[now]=last;
if(son[now]) qaq(son[now],last);
for(auto t:e[now])
if((t^r[now])&&(t^son[now]))
qaq(t,t);
}
inline void init(const int&n){pre(1,0);qaq(1,0);}
inline int lca(int u,int v){
if(d[u]<d[v])swap(u,v);
while(top[u]^top[v]){
if(d[top[u]]<d[top[v]]) swap(u,v);
u=r[top[u]];
}
return d[u]<d[v]?u:v;
}
inline void addq(const int&x,const int&y,const int&id,const int&w){
ans[id]=1;
q[x].push_back(id);
q[y].push_back(id);
q[lca(x,y)].push_back(-id);
W[id]=w;
}
}
namespace sol{
const int maxn=1e5+5;
using namespace TREE;
using PRE::rank;
using PRE::Min;
inline void add(const int&now,const int&tag){
//now 是节点编号 tag 是操作类型
int k=w[now],cnt=0;
static int a[23],p[23];
while(k>1){
int f=Min[k];
a[++cnt]=0;
p[cnt]=f;
while(k%f==0) k/=f,++a[cnt];
}
for(int t=1;t<=cnt;++t){
if(((int)s[rank[p[t]]].size())<=1+a[t])
s[rank[p[t]]].resize(a[t]+2);
s[rank[p[t]]][a[t]]+=tag;
}
}
inline void que(const int&now,const int&id){
//now 是询问编号
int k=W[now>0?now:-now],cnt=0,ret=1;
static int a[23],p[23];
while(k>1){
int f=Min[k];
a[++cnt]=0;
p[cnt]=f;
while(k%f==0) k/=f,++a[cnt];
}
for(int t=1;t<=cnt;++t)
for(int i=1,ed=s[rank[p[t]]].size();i<ed;++i)
ret=1ll*ret*ksm(p[t],1ll*s[rank[p[t]]][i]*min(a[t],i)%(mod-1))%mod;
if(now<0) ans[-now]=1ll*ans[-now]*inv(ret)%mod*inv(ret)%mod*__gcd(W[-now],w[id])%mod;
else ans[now]=1ll*ans[now]*ret%mod;
}
void dfs(const int&now,const int&last){
add(now,1);
for(auto t:q[now]) que(t,now);
for(auto t:e[now])
if(t^last) dfs(t,now);
add(now,-1);
}
}
int main(){
PRE::pre(1e7);
n=qr();
for(int t=2;t<=n;++t) TREE::add(qr(),qr());
for(int t=1;t<=n;++t) TREE::w[t]=qr();
TREE::init(n);
m=qr();
for(int t=1,t1,t2,t3;t<=m;++t) {
t1=qr(); t2=qr(); t3=qr();
TREE::addq(t1,t2,t,t3);
}
sol::dfs(1,0);
for(int t=1;t<=m;++t) printf("%d\n",TREE::ans[t]);
return 0;
}
博客保留所有权利,谢绝学步园、码迷等不在文首明显处显著标明转载来源的任何个人或组织进行转载!其他文明转载授权且欢迎!