2018 Multi-University Training Contest 10 - Problem E. TeaTree (树上 dsu // 线段树合并)

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

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

题解：小于 100000 的数最多只有一百多个约数. 建出所有点的约数线段树, 然后线段树合并, 重复的单点就可以对答案产生贡献. 当然有省内存的做法, DSU on tree, 不过反正时间复杂度不变. 时间与空间都是 O(100×nlogn)。

树上 dsu：

/**
具体的做法是，我们先计算一个点的轻儿子，再把它的影响消除。再计算重儿子，此时不必消除影响。

为了完成统计，最后再统计一遍轻儿子。

可以这么考虑：只有dfs到轻边时，才会将轻边的子树中合并到上一级的重链，

树链剖分将一棵树分割成了不超过lognlogn条重链。
每一个节点最多向上合并lognlogn次，单次修改复杂度O(1)O(1)。
所以整体复杂度是O(nlogn)O(nlogn)的。
**/
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<map>
#include<set>
#include<vector>
using namespace std;
const int maxn = 101000;
int n,v[maxn+5],sz[maxn+5],son[maxn+5],ans[maxn+5],vis[maxn+5],mx;
vector<int>g[maxn+5],yu[maxn+5],ct;
int Map[maxn+5];
void dfsson(int x)
{
    int len = g[x].size();
    son[x] = -1;
    sz[x] = 1;
    for(int i=0,y; i<len; i++)
    {
        y = g[x][i];
        dfsson(y);
        sz[x]+=sz[y];
        if(son[x]==-1||sz[son[x]]<sz[y]) son[x] = y;
    }
}
void chang(int x,int add)
{
    int len0 = yu[v[x]].size();
    for(int i=0; i<len0; i++)
    {
        int w= yu[v[x]][i];
        if(add==1) ct.push_back(w);
        if(add==1&&Map[w]) mx = max(mx,w);
        if(add==-1) Map[w]--;
    }
    int len = g[x].size();
    for(int i=0,y; i<len; i++)
    {
        y = g[x][i];
        chang(y,add);
    }
}
void solve(int x)
{
    mx = -1;
    int len = g[x].size();
    for(int i=0,y; i<len; i++)
    {
        y = g[x][i];
        if(y==son[x]) continue;
        ct.clear();
        chang(y,1);//把轻儿子的影响加进来
        for(int j=0,l=ct.size(); j<l; j++) Map[ct[j]]++;
    }
    int len0 = yu[v[x]].size();
    for(int i=0; i<len0; i++)
    {
        int w= yu[v[x]][i];
        if(Map[w]) mx = max(mx,w);
        Map[w]++;
    }
    ans[x] = mx;
}
void dfs(int x)
{
    int len = g[x].size();
    for(int i=0,y; i<len; i++)
    {
        y = g[x][i];
        if(y==son[x]) continue;
        dfs(y);//先计算轻儿子
        chang(y,-1);//把轻儿子的影响消除
    }
    if(son[x]!=-1) dfs(son[x]);//再计算重儿子。
    solve(x);//计算x
}
int main()
{
    for(int i=1; i<=maxn; i++)
    {
        for(int j=1; j*j<=i; j++)
        {
            if(i%j) continue;
            int a = j, b = i/a;
            yu[i].push_back(a);
            if(a!=b) yu[i].push_back(b);
        }
    }
    scanf("%d",&n);
    for(int i=2,f; i<=n; i++)
    {
        scanf("%d",&f);
        g[f].push_back(i);
    }
    for(int i=1; i<=n; i++) scanf("%d",&v[i]);
    dfsson(1);
    dfs(1);
    for(int i=1; i<=n; i++) printf("%d\n",ans[i]);
    return 0;
}

线段树合并：

#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
#include<stdio.h>
#define mid ((l+r)>>1)
using namespace std;
typedef long long int ll;
const int maxn = 1e5;
int n,v[maxn+5];
int ls[maxn*400],rs[maxn*400],root[maxn+5],sz,ans[maxn+5];
vector<int>g[maxn+5],yu[maxn+5];
void updata(int &o,int l,int r,int p)
{
    if(o==0) o = ++sz;
    if(l==r) return;
    if(p<=mid) updata(ls[o],l,mid,p);
    if(mid<p) updata(rs[o],mid+1,r,p);
}
int Merge(int o,int pre,int l,int r,int &Mx)
{
    if(!o||!pre) return o|pre;//这里可以比用set写 快很多
    if(l==r) Mx = max(Mx,l);
    if(ls[o]|ls[pre]) ls[o] = Merge(ls[o],ls[pre],l,mid,Mx);
    if(rs[o]|rs[pre]) rs[o] = Merge(rs[o],rs[pre],mid+1,r,Mx);
    return o;
}
void dfs(int x)
{
    int len = g[x].size();
    ans[x] = -1;
    for(int i=0;i<len;i++)
    {
        int y = g[x][i];
        dfs(y);
        root[x] = Merge(root[x],root[y],1,maxn,ans[x]);
    }
}
int main()
{
    for(int i=1; i<=maxn; i++)
    {
        for(int j=1; j*j<=i; j++)
        {
            if(i%j) continue;
            int a = j, b = i/a;
            yu[i].push_back(a);
            if(a!=b) yu[i].push_back(b);
        }
    }
    scanf("%d",&n);
    for(int i=2,f; i<=n; i++)
    {
        scanf("%d",&f);
        g[f].push_back(i);
    }
    for(int i=1; i<=n; i++)
    {
         scanf("%d",&v[i]);
         for(int j=0,l=yu[v[i]].size();j<l;j++) updata(root[i],1,maxn,yu[v[i]][j]);
    }
    dfs(1);
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    return 0;
}