2018多校第十场 HDU 6430 （线段树合并）

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

题意：一棵树上每个节点权值为v[i]，每个节点的heard值是：以它为LCA的两个节点的GCD的最大值，要求输出每个节点的heard值。

题解：权值范围是[1, 1e5]，1e5内数因子最多不超过200个，对每个点建一颗线段树，维护每个点的因子，dfs过程中由下往上合并线段树并更新答案。

 #include <bits/stdc++.h>
 using namespace std;
 #define ll long long
 #define ull unsigned long long
 #define mst(a,b) memset((a),(b),sizeof(a))
 #define mp(a,b) make_pair(a,b)
 #define fi first
 #define se second
 #define pi acos(-1)
 #define pii pair<int,int>
 #define pb push_back
 const int INF = 0x3f3f3f3f;
 const double eps = 1e-;
 const int MAXN = 1e5 + ;
 const int MAXM = 2e6 + ;
 const ll mod = 1e9 + ;

 vector<int>yz[MAXN],vec[MAXN];

 void init() {
     for(int i = ; i < MAXN; i++) {
         for(int j = ; j <= sqrt(i); j++) {
             if(i % j == ) {
                 yz[i].pb(j);
                 if(j != i / j) yz[i].push_back(i / j);
             }
         }
     }
 }

 int root[MAXN];
 int ls[MAXN * ], rs[MAXN * ], st[MAXN * ];
 int cnt = ;

 void pushup(int o) {
     st[o] = max(st[ls[o]], st[rs[o]]);
 }

 void update(int k,int l,int r,int &o) {
     if(!o) o = ++cnt;
     if(l == r) {
         st[o] = k;
         return ;
     }
     int mid = (l + r) >> ;
     if(k <= mid) update(k, l, mid, ls[o]);
     else update(k, mid + , r, rs[o]);
     pushup(o);
 }

 int mergee(int fa,int u,int &ans) {
     if(!fa || !u) return fa | u;
     if(st[fa] == st[u]) ans = max(ans, st[fa]);
     if(ls[fa] || ls[u]) ls[fa] = mergee(ls[fa], ls[u], ans);
     if(rs[fa] || rs[u]) rs[fa] = mergee(rs[fa], rs[u], ans);
     return fa;
 }

 int ans[MAXN];

 void dfs(int u) {
     for(int i = ; i < vec[u].size(); i++) {
         int v = vec[u][i];
         dfs(v);
         mergee(root[u], root[v], ans[u]);
     }
 }

 int main() {
 #ifdef local
     freopen("data.txt", "r", stdin);
 //    freopen("data.txt", "w", stdout);
 #endif
     init();
     int n;
     scanf("%d",&n);
     for(int i = ; i <= n; i++) {
         int f;
         scanf("%d",&f);
         vec[f].push_back(i);
     }
     for(int i = ; i <= n; i++) {
         int x;
         scanf("%d",&x);
         for(int j = ; j < yz[x].size(); j++)
             update(yz[x][j], , 1e5, root[i]);
     }
     mst(ans, -);
     dfs();
     for(int i = ; i <= n; i++)
         printf("%d\n",ans[i]);
     return ;
 }