zoukankan      html  css  js  c++  java
  • 树上乱搞(树上数据结构

    树上乱搞

    bzoj3319:黑白树

    允许离线的话,添加操作可以变为撤销操作

    做这题之前我们先体验一个水题, 一个离线就变得容易的题

    (WTM因为没有倒着输出而改了好些时间...傻子

    bzoj4551: [Tjoi2016&Heoi2016]树

    Description

    在2016年,佳媛姐姐刚刚学习了树,非常开心。现在他想解决这样一个问题:给定一颗有根树(根为1),有以下

    两种操作:1. 标记操作:对某个结点打上标记(在最开始,只有结点1有标记,其他结点均无标记,而且对于某个

    结点,可以打多次标记。)2. 询问操作:询问某个结点最近的一个打了标记的祖先(这个结点本身也算自己的祖

    先)你能帮帮他吗?

    分析

    显然,可以用线段树+树剖暴力在线做

    再想想,没必要这样,因为它维护的信息太简单了,所以,这时候就想到了离线..太牵强了吧

    (因为重点不在这,所以我们直接看黑白树吧....还是放上代码吧

    #include<cstdio>
    #include<algorithm>
    #include<iostream>
    using namespace std;
    const int MAX = 100000+99;
    
    int n,m;
    int stck[MAX];
    int father[MAX];
    
    int fa[MAX], cont[MAX];
    int find(int x) {
    	if(x == fa[x]) return x;
    	return fa[x] = find(fa[x]);
    }
    
    int cmd[MAX];
    char cmdtype[MAX];
    
    int main() {
    	scanf("%d%d",&n,&m);
    	int x,y;
    	for(int i = 1; i < n; i++) {
    		scanf("%d%d",&x,&y);
    		father[y] = x;
    	}
    	for(int i = 1; i <= m; i++) {
    		cin>>cmdtype[i];
    		scanf("%d",&cmd[i]);
    		if(cmdtype[i] == 'C') cont[cmd[i]]++;
    	}//离线
    	++cont[1];//第一个点打了 
    	for(int i = 1; i <= n; i++) {
    		if(cont[i]) fa[i] = i;
    		else fa[i] = father[i];
    	}//初始化: 标记过的点显然连自己;未标记过的点连向它爸,在之后的find里找答案。
    	//反着做,每到一个标记点的操作就使其标记数--(因为可能多次标记),标记数为0后就要连向它爸了。 
    	int ans[MAX], tot = 0;
    	for(int i = m; i >= 1; i--) {
    		if(cmdtype[i] == 'Q') {
    			ans[++tot] = find(cmd[i]);
    		} else {//相当于删边
    			if(--cont[cmd[i]] == 0) fa[cmd[i]] = father[cmd[i]];//也可以不记录它被染的次数,直接记录被染时间也行
    		}
    	}
    	for(int i = tot; i >= 1; i--)
    		printf("%d
    ",ans[i]);
    }
    

    黑白树Description

    给定一棵树,边的颜色为黑或白,初始时全部为白色。维护两个操作:

    1.查询u到根路径上的第一条黑色边的标号。

    2.将u到v 路径上的所有边的颜色设为黑色。

    Notice:这棵树的根节点为1

    分析

    实际上这题就是上题的翻版,就把单个节点修改改成了链修改,还有就是要把边的颜色下放到点(注意记录边的编号)

    所以我们还是可以线段树+树剖做,所以我们这样还是过不了emm...

    所以我们考虑用优美的并查集优化区间修改操作:在链修改的时候,用并查集维护x向上的第一个白节点是谁,这样,我们的修改操作就优化成O(n)的了(每个点就改了一次),注意同时记录第一次被染色的时间 (这题不像上面的那题可以记录被染次数,所以要记录被染时间)。查询操作同上一题,另开一个并查集记录向上的第一个白节点是谁,找到根节点即可。

    ps:不知道为啥自己的代码不是RE就是TLE,这里就粘上网上一位同学的解法

    说明一下,对于这个解法:它只用了一个并查集,因为它在重新赋值的时候,原来的那个已经没用了,它并没有开vector来记录每条边第一次被染色的时间,而是把之前的链式前向星清空,直接用单向链表遍历出边,起点是第一次染色时刻,终点即为编号

    #include <cstdio>
    #include <cstring>
    #include <iostream>
    using namespace std;
    const int maxn=1000010;
    int n,m,cnt;
    int to[maxn<<1],next[maxn<<1],head[maxn],vis[maxn],f[maxn],v[maxn];
    int dep[maxn],fa[maxn],son[maxn],top[maxn],siz[maxn],q[maxn],ans[maxn];
    inline int rd()
    {
        int ret=0,f=1;  char gc=getchar();
        while(gc<'0'||gc>'9') {if(gc=='-')f=-f;   gc=getchar();}
        while(gc>='0'&&gc<='9')   ret=ret*10+gc-'0',gc=getchar();
        return ret*f;
    }
    void add(int a,int b)
    {
        to[++cnt]=b,next[cnt]=head[a],head[a]=cnt;
    }
    void dfs1(int x)
    {
        siz[x]=1;
        for(int i=head[x];i;i=next[i])  if(to[i]!=fa[x])
        {
            fa[to[i]]=x,dep[to[i]]=dep[x]+1,v[to[i]]=(i+1)>>1,dfs1(to[i]),siz[x]+=siz[to[i]];
            if(siz[to[i]]>siz[son[x]])   son[x]=to[i];
        }
    }
    void dfs2(int x,int tp)
    {
        top[x]=tp;
        if(son[x])  dfs2(son[x],tp);
        for(int i=head[x];i;i=next[i])  if(to[i]!=fa[x]&&to[i]!=son[x]) dfs2(to[i],to[i]);
    }
    int lca(int x,int y)
    {
        while(top[x]!=top[y])
        {
            if(dep[top[x]]<dep[top[y]])  swap(x,y);
            x=fa[top[x]];
        }
        if(dep[x]<dep[y])    return x;
        return y;
    }
    int find(int x)
    {
        return (f[x]==x)?x:(f[x]=find(f[x]));
    }
    int main()
    {
        n=rd(),m=rd();
        int i,j,a,b,c;
        for(i=1;i<n;i++) a=rd(),b=rd(),add(a,b),add(b,a);
        for(i=1;i<=n;i++)    f[i]=i;
        dep[1]=1,dfs1(1),dfs2(1,1);
        memset(head,0,sizeof(head)),cnt=0;
        for(i=1;i<=m;i++)
        {
            if(rd()==1) q[i]=rd();
            else
            {
                a=rd(),b=rd(),c=lca(a,b);
                a=find(a),b=find(b);
                while(dep[a]>dep[c]) f[a]=find(fa[a]),add(i,a),vis[a]=1,a=f[a];
                while(dep[b]>dep[c]) f[b]=find(fa[b]),add(i,b),vis[b]=1,b=f[b];
            }
        }
        for(i=1;i<=n;i++)    f[i]=!vis[i]?fa[i]:i;
        for(i=m;i>=1;i--)
        {
            if(q[i])    ans[i]=v[find(q[i])];
            else    for(j=head[i];j;j=next[j])  f[to[j]]=find(fa[to[j]]);
        }
        for(i=1;i<=m;i++)    if(q[i])    printf("%d
    ",ans[i]);
        return 0;
    }
    

    暂时就这样吧

  • 相关阅读:
    python中的 upper() 、lower()、capitalize()、title()方法
    python中的strip()函数的用法
    python中的split()函数的用法
    python中的join()函数的用法
    appium 简单的测试例子(1)
    模拟器连接使用
    appuim连接时的一些出现的错误
    android获取包名的几种方法
    python+Appium(1)
    专注基础软件自研,巨杉入选“2020信创产业独角兽100强”
  • 原文地址:https://www.cnblogs.com/tyner/p/11405674.html
Copyright © 2011-2022 走看看