花花的森林（倍增，LCA

---

花花的森林，嗯，这是一篇正经的题解。

模拟考的时候没有看出来要怎么求啊，暴力地树形DP、换根、合并、求直径。居然也险险地拿到了80分，不过我们要正经地想正解。

容易想到我们可以让时光倒流，让空间扭转，离线地从最后一次操作往前，这样删边就变成加边了（合并总比拆开好做吧）

首先，我们要知道，两棵树合并后，新直径的两个端点一定是原来两棵树的两个直径的四个不同端点中的某两个（为什么？），自己画一下图，思考证明一下，这个结论并不难得出。

所以说合并时的新直径就只会有六种即 C(4,2)，那么我们分别求出这六个直径的值（当然原本的两种不用求），用最大的作为新树的直径就好啦~，并且记下此时的两个端点。

那要怎么求直径啊~？

当然是倍增啦。

可是可是，我们都把树拆成一棵棵了，怎么倍增啊，合并后不是还要重新算一次倍增的数组吗？

我们可以在原树上倍增呀，没必要真的把树拆了（如果题目让你做什么你就做什么，一定会被带离正解的），在一棵树上，两点之间的简单路径是唯一的。

（那你真是个小机灵鬼~）

然后合并就用并查集维护。

哦对了，还有一个细节，从后往前做我们会遇到除法的问题（难不成你要循环一遍整个森林求一次ans吗）所以在模质数的意义下，使用费马小定理求逆元来进行除法。

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<algorithm>

 #define For(i,a,b) for(register int i=a;i<=b;++i)
 #define Dwn(i,a,b) for(register int i=a;i>=b;--i)
 #define Re register
 #define Pn putchar('\n')
 #define llg long long
 using namespace std;
 const int N=1e5+,md=1e9+;
 llg a[N],ans=,dt[N],pt[N][],fn[N];
 int head[N],nxt[N*],v[N*],cnt=;
 int pa[N],sz[N],n,m,x,y,Q[N];
 int f[N][];
 llg d[N][];
 struct EDG{
     int a,b;
 }ed[N];
 inline void read(int &v){
     v=;
     char c=getchar();
     while(c<''||c>'')c=getchar();
     while(c>=''&&c<='')v=v*+c-'',c=getchar();
 }
 inline void read(llg &v){
     v=;
     char c=getchar();
     while(c<''||c>'')c=getchar();
     while(c>=''&&c<='')v=v*+c-'',c=getchar();
 }
 void write(llg x){
     if(x>)write(x/);
     int xx=x%;
     putchar(xx+'');
 }
 int find(int x){
     int r=x;
     while(pa[r]!=r)r=pa[r];
     int i,j;
     i=x;
     while(pa[i]!=r){
         j=pa[i]; pa[i]=r; i=j;
     }
     return r;
 }
 void add(int ux,int vx){
     cnt++;
     nxt[cnt]=head[ux]; head[ux]=cnt; v[cnt]=vx;
     cnt++;
     nxt[cnt]=head[vx]; head[vx]=cnt; v[cnt]=ux;
 }
 llg Qsm(llg x,int k){
     llg ans=;
     while(k){
         if(k&)ans=(ans*x)%md;
         x=(x*x)%md;
         k>>=;
     }
     return ans;
 }

 int dep[N];
 void DFS(int x,int fa){

     for(Re int i=head[x];i;i=nxt[i]){
         int vv=v[i];

         if(vv==fa)continue;

         f[vv][]=x; d[vv][]=a[vv];
         dep[vv]=dep[x]+;

         DFS(vv,x);
     }
 }

 llg getDis(int x,int y){
     llg as=;
     if(dep[x]<dep[y])swap(x,y);

     Dwn(b,,)if(f[x][b]!=-){
         if(dep[f[x][b]]>=dep[y]){
             as+=d[x][b];  x=f[x][b];
         }
     }

     if(x==y)return as+a[x];
     Dwn(b,,)if(f[x][b]!=-&&f[y][b]!=-&&f[x][b]!=f[y][b]){
         as+=d[x][b]; as+=d[y][b];
         x=f[x][b]; y=f[y][b];
     }
     as+=d[x][]; as+=d[y][];

     return as+a[f[x][]];
 }

 llg d00,d10,d01,d11,kd,dx=,px1,px2;

 int main(){
     freopen("forest.in","r",stdin);
     freopen("forest.out","w",stdout);
     read(n);
     For(i,,n){
         read(a[i]);
         ans=(ans*a[i])%md;
         pt[i][]=pt[i][]=i; dt[i]=a[i];
     }
     For(i,,n-){
         read(ed[i].a); read(ed[i].b);
         add(ed[i].a,ed[i].b);
     }
     memset(f,-,sizeof(f));

     DFS(,);

     For(b,,) For(i,,n){
         if(f[i][b-]==-)continue;

         f[i][b]= f[ f[i][b-] ][b-];
         d[i][b]= d[ f[i][b-] ][b-]+d[i][b-];
     }

     For(i,,n)pa[i]=i;

     For(i,,n-)read(Q[i]);
     fn[n]=ans;
     Dwn(i,n-,){
         int qs=Q[i];
         x=ed[qs].a; y=ed[qs].b;
         int pax=find(x),pay=find(y);

         ans=(ans*Qsm(dt[pax],md-))%md;
         ans=(ans*Qsm(dt[pay],md-))%md;

         if(dt[pax]>dt[pay]){
             dx=dt[pax]; px1=pt[pax][]; px2=pt[pax][];
         }else{
             dx=dt[pay]; px1=pt[pay][]; px2=pt[pay][];
         }

         d00=getDis(pt[pax][],pt[pay][]);
         if(d00>dx)dx=d00,px1=pt[pax][],px2=pt[pay][];

         d10=getDis(pt[pax][],pt[pay][]);
         if(d10>dx)dx=d10,px1=pt[pax][],px2=pt[pay][];

         d01=getDis(pt[pax][],pt[pay][]);
         if(d01>dx)dx=d01,px1=pt[pax][],px2=pt[pay][];

         d11=getDis(pt[pax][],pt[pay][]);
         if(d11>dx)dx=d11,px1=pt[pax][],px2=pt[pay][];

         pt[pax][]=px1; pt[pax][]=px2;

         dt[pax]=dx;
         ans=(ans*dx)%md;
         pa[pay]=pax;
         fn[i]=ans;
     }
     For(i,,n){
         write(fn[i]); Pn;
     }
     return ;
 }