hdu4812 D Tree（点分治）

题意：求点权树上路径上点权积modp等于K的两端点字典序最小路径
解法：
1.id[u]:经过重心长度为x的另一端最小值
num[]和length[]：分别保存子树端点的端点值和长度。
遍历num和length数组，更新id数组
2.在id数组中查找”新路径的逆元*K”，用该路径两端点更新答案。

#pragma comment(linker,"/STACK:102400000,102400000")
#include <stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<iostream>
#define INF 0x3f3f3f3f
#define ll long long
#define X first
#define Y second
#define MP(x,y) make_pair((x),(y))
using namespace std;

const ll maxn = 100000+10;
const ll mod= 1e6+3;
typedef pair<ll,ll> F;
ll n,K;
struct Node{
    ll to,next;
}edge[maxn<<1];
ll head[maxn],tot;

void add(ll u,ll v){
    edge[tot].to=v,edge[tot].next=head[u],head[u]=tot++;
}

ll core,sum,son[maxn],maxs[maxn];
bool vis[maxn];
ll q[maxn];
ll ni[mod+10];

void init(){//线性预处理每个数的逆元
    ni[1]=1;
    for(int i=2;i<mod;i++)
    {
        int a=mod/i,b=mod%i;
        ni[i]=(ni[b]*(-a)%mod+mod)%mod;
    }
}

void getcore(ll u,ll fa){
    son[u]=1,maxs[u]=0;
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(v==fa||vis[v]) continue;
        getcore(v,u);
        son[u]+=son[v]; maxs[u]=max(maxs[u],son[v]);
    }
    maxs[u]=max(maxs[u],sum-son[u]);
    if(maxs[u]<maxs[core]) core=u;
}

//记录子树路径信息变量
ll num[maxn],length[maxn],cnt;
ll length2[maxn],cnt2;
ll id[mod+10];
F ans;

void cal(){//利用id数组计算ans
    for(ll i=1;i<=cnt;i++){
        ll x=length[i],y=num[i];
        x=(x*q[core])%mod;
        if(x==K) ans=min(ans,MP(min(y,core),max(y,core)));
        x=(K*ni[x])%mod;
        ll tmp=id[x];
        if(tmp){
            if(tmp>y) swap(y,tmp);
            ans=min(ans,MP(tmp,y));
        }
    }
}

void dfs(ll u,ll fa,ll depth){//求出每个子树路径信息（长度、节点标号）并保存在数组中
    num[++cnt]=u,length[cnt]=depth;
    if(!id[depth]) length2[++cnt2]=depth;
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(v==fa||vis[v]) continue;
        dfs(v,u,(depth*q[v])%mod);
    }
}

void update(){//更新id数组
    for(ll i=1;i<=cnt;i++){
        ll x=length[i],y=num[i];
        if(!id[x]) id[x]=y;
        else id[x]=min(id[x],y);
    }
}

void work(ll u){
    vis[u]=1; cnt2=0;
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(vis[v]) continue;
        cnt=0; dfs(v,u,q[v]);
        cal(); update();
    }
    //清空id数组和记录所有子树信息的数组
    for(ll i=1;i<=cnt2;i++) id[length2[i]]=0;
    cnt2=0;
    for(ll i=head[u];i!=-1;i=edge[i].next){
        ll v=edge[i].to;
        if(vis[v]) continue;
        sum=son[v],core=0; getcore(v,u);
        work(core);
    }
}

int main(){
    init();
    //freopen("a.txt","r",stdin);
    while(scanf("%lld%lld",&n,&K)!=EOF){
        memset(head,-1,sizeof(head)); tot=0;
        for(ll i=1;i<=n;i++) scanf("%lld",&q[i]);
        for(ll i=0;i<n-1;i++){
            ll u,v;
            scanf("%lld%lld",&u,&v);
            add(u,v); add(v,u);
        }
        ans=MP(INF,INF);
        memset(id,0,sizeof(id));
        memset(vis,0,sizeof(vis));
        core=0,sum=n,maxs[core]=INF;
        getcore(1,0);
        work(core);
        if(ans.X==INF) printf("No solution\n");
        else printf("%lld %lld\n",ans.X,ans.Y);
    }
    return 0;
}