D Tree HDU - 4812

https://vjudge.net/problem/HDU-4812

点分就没一道不卡常的？

卡常记录：

1.求逆元忘开longlong

2.把solve中分离各个子树的方法，由“一开始全部加入，处理某个子树前先删除该子树”，变为“逐渐加入，每一次加入某个子树之前处理该子树，不用删除“（由于点对是无序的，只要一个方向处理过就行了，另一个方向不需要处理）

 //%:pragma GCC optimize(3)
 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 #include<set>
 using namespace std;
 #define fi first
 #define se second
 #define mp make_pair
 #define pb push_back
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pi;
 #define md 1000003
 struct E
 {
     int to,nxt;
 }e[];
 int f1[],ne;
 int sz[],fx[],a[],d[];
 bool vis[];
 int root,sum;pi ans;
 set<pi> s;
 int inv[];
 int n,K;
 void getroot(int x,int fa)
 {
     sz[x]=;fx[x]=;
     for(int k=f1[x];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             getroot(e[k].to,x);
             sz[x]+=sz[e[k].to];
             fx[x]=max(fx[x],sz[e[k].to]);
         }
     fx[x]=max(fx[x],sum-sz[x]);
     if(fx[x]<fx[root])    root=x;
 }
 void getsz(int x,int fa)
 {
     sz[x]=;
     for(int k=f1[x];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             getsz(e[k].to,x);
             sz[x]+=sz[e[k].to];
         }
 }
 void getd(int u,int fa)
 {
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
         {
             d[e[k].to]=ll(d[u])*a[e[k].to]%md;
             getd(e[k].to,u);
         }
 }
 void addd(int u,int fa)
 {
     s.insert(mp(d[u],u));
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
             addd(e[k].to,u);
 }
 void deld(int u,int fa)
 {
     s.erase(mp(d[u],u));
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
             deld(e[k].to,u);
 }
 void calc(int u,int fa)
 {
     int t=ll(K)*inv[ll(d[u])*a[root]%md]%md;
     auto it=s.lower_bound(mp(t,));
     if(it!=s.end()&&it->fi==t)
     {
         int a=u,b=it->se;
         if(a>b)    swap(a,b);
         ans=min(ans,mp(a,b));
     }
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to]&&e[k].to!=fa)
             calc(e[k].to,u);
 }
 void solve(int u)
 {
     s.clear();d[u]=;s.insert(mp(,u));vis[u]=;
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             d[e[k].to]=a[e[k].to];getd(e[k].to,u);
             calc(e[k].to,u);
             addd(e[k].to,u);
         }
     for(int k=f1[u];k;k=e[k].nxt)
         if(!vis[e[k].to])
         {
             root=;
             getsz(e[k].to,);sum=sz[e[k].to];
             getroot(e[k].to,);
             solve(root);
         }
 }
 int main()
 {
     int i,x,y;
     inv[]=;
     for(i=;i<=;i++)    inv[i]=ll(md-md/i)*inv[md%i]%md;
     fx[]=0x3f3f3f3f;
     while(scanf("%d%d",&n,&K)==)
     {
         for(i=;i<=n;i++)    f1[i]=vis[i]=;
         ne=;ans=mp(n+,n+);
         for(i=;i<=n;i++)    scanf("%d",&a[i]);
         for(i=;i<n;i++)
         {
             scanf("%d%d",&x,&y);
             e[++ne].to=y;e[ne].nxt=f1[x];f1[x]=ne;
             e[++ne].to=x;e[ne].nxt=f1[y];f1[y]=ne;
         }
         root=;sum=n;getroot(,);
         solve(root);
         if(ans==mp(n+,n+))    puts("No solution");
         else    printf("%d %d\n",ans.fi,ans.se);
     }
     return ;
 }