hdu4812 D Tree 【点分治+逆元+hash】

链接：http://acm.hdu.edu.cn/showproblem.php?pid=4812

题意：给你一棵树，树的每个节点都有一个权值，现在给你一个k，问是否有一条简单路径（u，v）上的权值积%mod等于k。输出字典序最小的（u，v）。

分析：很明显，这是点分治的题，每次分治重心，我们只要算过重心的路径是否这样的答案。

判断的时候，我们遍历以重心为根的子树，的到重心的子树到子树每个节点的乘积，然后判断hash表里面是否存在k*inv[mul]，然后更新答案就行了。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<string>
#include<vector>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<map>
#define INF 0x3f3f3f3f
#define Mn 100010
#define Mm 2000005
#define mod 1000003
#define CLR(a,b) memset((a),(b),sizeof((a)))
#define CLRS(a,b,Size) memset((a),(b),sizeof((a[0]))*(Size+1))
#define CPY(a,b) memcpy ((a), (b), sizeof((a)))
#pragma comment(linker, "/STACK:102400000,102400000")
#define ul u<<1
#define ur (u<<1)|1
using namespace std;
typedef long long ll;
struct edge {
    int v,next;
}e[Mm];
int tot,head[Mn];
void addedge(int u,int v) {
    e[tot].v=v;
    e[tot].next=head[u];
    head[u]=tot++;
}
int siz[Mn],vis[Mn],fSize,focus,sum;
void getFocus(int u,int fa) {
    siz[u]=1;
    int maxSonSize=0;
    for(int i=head[u];~i;i=e[i].next) {
        if(e[i].v==fa||vis[e[i].v]) continue;
        getFocus(e[i].v,u);
        siz[u]+=siz[e[i].v];
        maxSonSize=max(maxSonSize,siz[e[i].v]);
    }
    maxSonSize=max(maxSonSize,sum-siz[u]);
    if(fSize>maxSonSize) {
        fSize=maxSonSize;
        focus=u;
    }
}
ll val[Mn],inv[mod+5];
int has[mod+5],ansu,ansv;
void update(int u,int v) {
    if(u>v) swap(u,v);
    if(u==ansu) {
        ansv=min(ansv,v);
    } else if(u<ansu){
        ansu=u;
        ansv=v;
    }
}
pair<int ,ll> pil[Mn];
int cnt;
void getSonMul(int u,int fa,ll mul) {
    pil[cnt++]=make_pair(u,mul);
    for(int i=head[u];~i;i=e[i].next) {
        if(e[i].v==fa||vis[e[i].v]) continue;
        getSonMul(e[i].v,u,val[e[i].v]*mul%mod);
    }
}
int k;
void solve(int u) {
    vis[u]=1;
    cnt=0;
    for(int i=head[u];~i;i=e[i].next) {
        if(vis[e[i].v]) continue;
        int len=cnt;
        getSonMul(e[i].v,u,val[e[i].v]);
        for(int j=len;j<cnt;j++) {
            if(k==val[u]*pil[j].second%mod) update(pil[j].first,u);
            int x=k*inv[val[u]*pil[j].second%mod]%mod;
            if(has[x]) update(pil[j].first,has[x]);
        }
        for(int j=len;j<cnt;j++) {
            int x=pil[j].first;
            ll mul=pil[j].second;
            if(!has[mul]) has[mul]=x;
            else has[mul]=min(x,has[mul]);
        }
    }
    for(int i=0;i<cnt;i++) has[pil[i].second]=0;
    for(int i=head[u];~i;i=e[i].next) {
        if(vis[e[i].v]) continue;
        sum=fSize=siz[e[i].v];
        getFocus(e[i].v,focus=0);
        solve(focus);
    }
}
void init() {
    tot=0;
    CLR(head,-1);
    CLR(vis,0);
    ansu=INF;
    ansv=INF;
}
int main() {
    int n,u,v;
    inv[1]=1;
    for(int i=2;i<mod;i++) {
        int a=mod/i,b=mod%i;
        inv[i]=(inv[b]*(-a)%mod+mod)%mod;
    }
    while(~scanf("%d%d",&n,&k)) {
        init();
        for(int i=1;i<=n;i++) {
            scanf("%d",&val[i]);
        }
        for(int i=1;i<n;i++) {
            scanf("%d%d",&u,&v);
            addedge(u,v);
            addedge(v,u);
        }
        sum=fSize=n;
        getFocus(u,focus=0);
        solve(focus);
        if(ansu==INF) printf("No solution\n");
        else printf("%d %d\n",ansu,ansv);
    }
    return 0;
}