hdu 4812 D Tree(分治)

题意：给出n个点的树，每个结点有权值，求一条路径，使得路径上点的乘积对1e6+3取模的结果为K。

思路：13年南京现场赛的题，当时太弱了……做不出来，这几天又拿出来看了看，有了些思路，但是这题时限卡得很紧，写完无限TLE，最后还是乱搞搞过去了。。。基本方法还是不难想的，就是树上的基于点的分治，如何计算乘积为K的路径呢？对于每个分支的点，计算从根节点到当前点的乘积x，然后利用逆元可以算出y=K*x^(-1)，然后通过hash去查找y的值是否出现过即可，逆元可以用扩展欧几里得预处理出来，对于相应的值要标记最小能得到它的点，边搞边去更新ans就行了。

#pragma comment(linker,"/STACK:102400000,102400000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
const int mod=1000000+3;
struct Edge
{
    int v,next;
    Edge(){}
    Edge(int v,int next):v(v),next(next){}
}edges[maxn<<1];
int head[maxn],sons[maxn],wroot,wval,nEdge;
int inv[mod+10],flag[mod+10],fff[mod+10],val[maxn];
int S[maxn],id[maxn],ans[2],cnt,tot,n,K;
bool vis[maxn];
void AddEdge(int u,int v)
{
    edges[++nEdge]=Edge(v,head[u]);
    head[u]=nEdge;
    edges[++nEdge]=Edge(u,head[v]);
    head[v]=nEdge;
}
ll ext_gcd(ll a,ll b,ll &x,ll &y)
{
    if(b==0)
    {
        x=1;y=0;
        return a;
    }
    ll d=ext_gcd(b,a%b,y,x);
    y-=a/b*x;
    return d;
}
void Init()
{
    memset(head,0xff,sizeof(head));
    memset(vis,0,sizeof(vis));
    memset(flag,0,sizeof(flag));
    nEdge=-1;cnt=0;tot=1;
    ans[0]=ans[1]=inf;
}
inline void Update(int x,int y)
{
    if(x>y) swap(x,y);
    if(x<ans[0]) ans[0]=x,ans[1]=y;
    else if(x==ans[0]&&y<ans[1]) ans[1]=y;
}
void getRoot(int u,int fa)
{
    sons[u]=1;
    int maxv=0;
    for(int k=head[u];k!=-1;k=edges[k].next)
    {
        int v=edges[k].v;
        if(v==fa||vis[v]) continue ;
        getRoot(v,u);
        sons[u]+=sons[v];
        maxv=max(maxv,sons[v]);
    }
    maxv=max(maxv,sons[0]-sons[u]);
    if(maxv<wval) {wval=maxv;wroot=u;}
}
void dfs(int u,int fa,int prev)
{
    int tmp=S[cnt]=(ll)val[u]*prev%mod;
    id[cnt++]=u; sons[u]=1;
    for(int k=head[u];k!=-1;k=edges[k].next)
    {
        int v=edges[k].v;
        if(v==fa||vis[v]) continue;
        dfs(v,u,tmp);
        sons[u]+=sons[v];
    }
}
void solve(int u,int cnode)
{
    if(cnode==1) return ;
    sons[0]=cnode;
    wval=inf;
    getRoot(u,-1);
    int x;
    for(int k=head[wroot];k!=-1;k=edges[k].next)
    {
        int v=edges[k].v;
        if(vis[v]||v==wroot) continue;
        cnt=0;
        dfs(v,wroot,val[wroot]);
        for(int i=0;i<cnt;++i)
        {
            if(S[i]==K) Update(wroot,id[i]);
            x=(ll)S[i]*inv[val[wroot]]%mod;
            x=(ll)K*inv[x]%mod;
            if(flag[x]==tot) Update(fff[x],id[i]);
        }
        for(int i=0;i<cnt;++i)
        {
            if(flag[S[i]]==tot)
                fff[S[i]]=min(id[i],fff[S[i]]);
            else
            {
                flag[S[i]]=tot;
                fff[S[i]]=id[i];
            }
        }
    }
    tot++;
    vis[wroot]=true;
    for(int k=head[wroot];k!=-1;k=edges[k].next)
    {
        int v=edges[k].v;
        if(v==wroot||vis[v]) continue;
        solve(v,sons[v]);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    ll x,y;
    for(int i=0;i<mod;++i)
    {
        ext_gcd(i,mod,x,y);
        x%=mod;
        if(x<0) x+=mod;
        inv[i]=x;
    }
    while(~scanf("%d%d",&n,&K))
    {
        Init();
        for(int i=1;i<=n;++i)
            scanf("%d",&val[i]);
        int u,v;
        for(int i=1;i<n;++i)
        {
            scanf("%d%d",&u,&v);
            AddEdge(u,v);
        }
        solve(1,n);
        if(ans[0]==inf) printf("No solution\n");
        else printf("%d %d\n",ans[0],ans[1]);
    }
    return 0;
}