hdu4812 D Tree（点分治）

求路径上所有点的点权积为K的最小的点对。（字典序最小）
点分治即可。b[x]表示前缀积为x的最小的点的标号。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
#define mod 1000003
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,K,h[N],num=0,sz[N],f[N],w[N],rt,sumsz,ans1,ans2,a[N],b[mod+10],dep[N],inv[mod+10],tot;
bool vis[N];
struct edge{
    int to,next;
}data[N<<1];
inline void dfs1(int x,int Fa){
    sz[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa||vis[y]) continue;
        dfs1(y,x);sz[x]+=sz[y];
    }
}
inline void dfs2(int x,int Fa){
    f[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa||vis[y]) continue;
        dfs2(y,x);f[x]=max(f[x],sz[y]);
    }f[x]=max(f[x],sumsz-sz[x]);if(f[x]<f[rt]) rt=x;
}
inline void dfs3(int x,int Fa){
    a[++tot]=x;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa||vis[y]) continue;
        dep[y]=(ll)dep[x]*w[y]%mod;dfs3(y,x);
    }
}
inline void dfs4(int x,int Fa){
    b[dep[x]]=inf;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==Fa||vis[y]) continue;dfs4(y,x);
    }
}
inline void calc(int x){
    int KK=(ll)K*inv[w[x]]%mod;b[1]=x;dep[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;tot=0;
        dep[y]=(ll)dep[x]*w[y]%mod;dfs3(y,x);
        for(int j=1;j<=tot;++j){
            int x=a[j],y=b[(ll)KK*inv[dep[x]]%mod];if(y==inf) continue;
            if(x>y) swap(x,y);if(x<ans1||x==ans1&&y<ans2) ans1=x,ans2=y;
        }for(int j=1;j<=tot;++j) b[dep[a[j]]]=min(b[dep[a[j]]],a[j]);
    }for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;dfs4(y,x);
    }b[1]=inf;
}   
inline void solve(int x){
    vis[x]=1;dfs1(x,0);calc(x);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(vis[y]) continue;
        sumsz=sz[y];rt=0;dfs2(y,x);solve(rt);
    }
}
int main(){
//  freopen("a.in","r",stdin);
    inv[1]=1;memset(b,inf,sizeof(b));
    for(int i=2;i<mod;++i) inv[i]=(ll)inv[mod%i]*(mod-mod/i)%mod;
    while(~scanf("%d %d",&n,&K)){
        for(int i=1;i<=n;++i) w[i]=read();
        memset(h,0,sizeof(h));num=0;f[0]=inf;ans1=inf;ans2=inf;memset(vis,0,sizeof(vis));
        for(int i=1;i<n;++i){
            int x=read(),y=read();
            data[++num].to=y;data[num].next=h[x];h[x]=num;
            data[++num].to=x;data[num].next=h[y];h[y]=num;
        }dfs1(1,0);sumsz=n;rt=0;dfs2(1,0);solve(rt);
        if(ans1==inf||ans2==inf) puts("No solution");
        else printf("%d %d\n",ans1,ans2);
    }return 0;
}