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-6;
const int MAXN = 1e5 + 10;
const int MAXM = 2e6 + 10;
const ll mod = 1e9 + 7;

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

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

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

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) >> 1;
    if(k <= mid) update(k, l, mid, ls[o]);
    else update(k, mid + 1, 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 = 0; 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 = 2; i <= n; i++) {
        int f;
        scanf("%d",&f);
        vec[f].push_back(i);
    }
    for(int i = 1; i <= n; i++) {
        int x;
        scanf("%d",&x);
        for(int j = 0; j < yz[x].size(); j++)
            update(yz[x][j], 1, 1e5, root[i]);
    }
    mst(ans, -1);
    dfs(1);
    for(int i = 1; i <= n; i++)
        printf("%d
",ans[i]);
    return 0;
}