hdu6430 Problem E. TeaTree 权值线段树 合并 暴力

新学到的权值线段树和它的合并,数组版的动态开点,以及如何估算线段树数组要开多大

敲开心~

应该还要学怎么分裂。。。

关于权值线段树的讲解看这里:https://www.cnblogs.com/Mychael/p/8665589.html

https://blog.csdn.net/Stupid_Turtle/article/details/80445998

不过这题也可以直接bitset存下所有的因子,每次合并的时候或一下两个二进制串,要求的就是公共的最右边的1对应的因子

从而不开权值线段树。。。反正都是暴力,比赛的时候没想到bitset,觉得暴力会MLE。。。准确来说也没想清楚要维护什么。。。orz

如果直接用vector存下每个点的子树的所有因子,然后比较,应该是会T的

用bitset和权值线段树就是为了优化

bitset用位运算优化,线段树相当于减少了比较的次数,从底层合并,两两比较O(nlogn)

树上启发式合并的方法。。留个坑来填

//权值线段树+线段树合并
//裸题&板子
#include 
using namespace std;
#define N 100005
int n,x,l;
int tot;
/*
int cnt[N];//记录i的银子个数
int a[N][400];//a[i][]存i的每个因子
不用vector MLE
*/
vector a[N];
int ans[N];
int head[N];
int root[N], ls[(N-5)*400], rs[(N-5)*400], sum[(N-5)*400];
//这里的上界N*400,是因为N个数要开线段树,每个线段树因为动态开点,大小为因子个数
struct node
{
    int y,next;
}edge[N];
void add(int x,int y)
{
    l++;
    edge[l].y = y;edge[l].next = head[x];head[x] = l;
}
void init()
{
    for (int i=1;i<=N-5;i++)
    {
        for (int j=1;j<=sqrt(i);j++)
            if (i % j == 0)
            {
                a[i].push_back(j);
                if (j * j != i) a[i].push_back(i/j);
            }
    }
}
inline void pushup(int rt)
{
    if (ls[rt] && rs[rt])
        sum[rt] = max(sum[ls[rt]], sum[rs[rt]]);
    else if (ls[rt]) sum[rt] = sum[ls[rt]];
    else if (rs[rt]) sum[rt] = sum[rs[rt]];
}
void updata(int &rt, int l, int r, int p)
{
    if (rt == 0) rt = ++tot;
    if (l == r)
    {
        sum[rt] = p;
        //这里的权值线段树维护的是这个子树内所有因子的最大值,所以长度只有400(也是这道题要想清楚的要维护的东西,其实就是暴力)
        return ;
    }
    int mid = (l + r) >> 1;
    if (p <= mid) updata(ls[rt], l, mid, p);
    if (p > mid) updata(rs[rt], mid+1, r, p);
    pushup(rt);
}
int Merge(int rt, int prt, int &ans)
{
    if (rt == 0 || prt == 0) return rt ^ prt;
    //维护最大的gcd
    if (sum[rt] == sum[prt]) ans = max(ans, sum[rt]);//合并的时候顺便求答案
    if (ls[rt] || ls[prt]) ls[rt] = Merge(ls[rt], ls[prt], ans);
    if (rs[rt] || rs[prt]) rs[rt] = Merge(rs[rt], rs[prt], ans);
    pushup(rt);
    return rt;
}
void dfs(int x)
{
    ans[x] = -1;
    for (int p=head[x];p;p=edge[p].next)
    {
        dfs(edge[p].y);
        root[x] = Merge(root[x], root[edge[p].y], ans[x]);
    }
}
int main()
{
    init();//预处理出1~1e5中每个数的所有因数,每个数的因数个数为sqrt(n)级别的
    scanf("%d", &n);
    //建边
    for (int i=2;i<=n;i++)
    {
        scanf("%d", &x);
        add(x,i);//因为这题保证了fa[i] < i
    }
    //对每个v[i]建权值线段树,根为root[i]
    tot = 0;
    for (int i=1;i<=n;i++)
    {
        scanf("%d", &x);
        root[i] = 0;
        for (int j = 0;j

 

你可能感兴趣的:(数据结构,线段树,权值线段树,启发式合并)