[BZOJ 2243][SDOI2011]染色(树链剖分+线段树)

Description

给定一棵有n个节点的无根树和m个操作,操作有2类:

1、将节点a到节点b路径上所有点都染成颜色c;

2、询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如“112221”由3段组成:“11”、“222”和“1”。

请你写一个程序依次完成这m个操作。

Solution

解决了一个困扰我很多年的问题=w=

(这题应该是刚学树剖的时候写的,然而莫名其妙的T掉了3个点,以为被卡常了,于是开始了卡常大作战…改了一下午之后就放弃了…今天翻出来发现是重儿子找错了,改了改就A了…另外感觉当时的代码风格真是十分不堪,当时的我真naive)

具体的一些细节我已经忘了QwQ

#include<iostream>
#include<cstdio>
#include<cstring>
#define MAXN 100005
#define RG register
using namespace std;
int n,m,head[MAXN],cnt=0,ans,col[MAXN];
int sz=0,pos[MAXN],top[MAXN],dep[MAXN],son[MAXN],father[MAXN],siz[MAXN],inv[MAXN];
inline int read()
{
    int f=1,x=0;char c=getchar();
    while(c<‘0‘||x>‘9‘){if(c==‘-‘)f=-1;c=getchar();}
    while(c>=‘0‘&&c<=‘9‘){x=x*10+c-‘0‘;c=getchar();}
    return f*x;
}
struct Node1{
    int next,to;
}Edges[MAXN*2];
inline void addedge(int u,int v)
{
    Edges[++cnt].next=head[u];
    head[u]=cnt;
    Edges[cnt].to=v;
}
inline void dfs1(int u)
{
    siz[u]=1;
    for(int i=head[u];~i;i=Edges[i].next)
    {
        int v=Edges[i].to;
        if(v==father[u])continue;
        father[v]=u;dep[v]=dep[u]+1;
        dfs1(v);siz[u]+=siz[v];
        if(siz[v]>siz[son[u]])son[u]=v;
    }
}
inline void dfs2(int u,int t)
{
    sz++,pos[u]=sz;
    inv[sz]=u,top[u]=t;
    if(son[u])dfs2(son[u],t);
    for(int i=head[u];~i;i=Edges[i].next)
    {
        int v=Edges[i].to;
        if(v==father[u]||v==son[u])continue;
        dfs2(v,v);
    }
}
struct Node2{
    int lc,rc,num,lazy;
}t[MAXN*3];
inline void pushdown(int idx,int l,int r)
{
    if(l!=r&&t[idx].lazy)
    {
        t[idx<<1].lazy=t[idx<<1|1].lazy=t[idx<<1].num=t[idx<<1|1].num=1;
        t[idx].lazy=0;
        t[idx<<1].lc=t[idx<<1|1].lc=t[idx<<1].rc=t[idx<<1|1].rc=t[idx].lc;
    }
}
inline void update(int idx)
{
    t[idx].lc=t[idx<<1].lc;
    t[idx].rc=t[idx<<1|1].rc;
    t[idx].num=t[idx<<1].num+t[idx<<1|1].num-(t[idx<<1].rc==t[idx<<1|1].lc);
}
inline void build(int idx,int l,int r)
{
    t[idx].lazy=0;
    if(l==r)
    {
        t[idx].lc=t[idx].rc=col[inv[l]];
        t[idx].num=1;
        return;
    }
    int mid=(l+r)>>1;
    build(idx<<1,l,mid),build(idx<<1|1,mid+1,r);
    update(idx);
}
inline void color(int idx,int l,int r,int a,int b,int c)
{
    if(l==a&&r==b)
    {
        t[idx].num=t[idx].lazy=1;
        t[idx].lc=t[idx].rc=c;
        return;
    }
    pushdown(idx,l,r);
    int mid=(l+r)>>1;
    if(a>mid)color(idx<<1|1,mid+1,r,a,b,c);
    else if(b<=mid)color(idx<<1,l,mid,a,b,c);
    else
    {
        color(idx<<1|1,mid+1,r,mid+1,b,c);
        color(idx<<1,l,mid,a,mid,c);
    }
    update(idx);
}
inline int query(int idx,int l,int r,int a,int b)
{
    if((l==a&&r==b)||t[idx].lazy)return t[idx].num;
    pushdown(idx,l,r);
    int mid=(l+r)>>1,ans=0;
    if(a>mid)return query(idx<<1|1,mid+1,r,a,b);
    else if(b<=mid)return query(idx<<1,l,mid,a,b);
    else
    {
        ans+=query(idx<<1,l,mid,a,mid);
        ans+=query(idx<<1|1,mid+1,r,mid+1,b);
        ans-=t[idx<<1].rc==t[idx<<1|1].lc;
    }
    return ans;
}
inline int getcolor(int idx,int l,int r,int a)
{
    if(l==r||t[idx].lazy)return t[idx].lc;
    int mid=(l+r)>>1;
    if(a<=mid)return getcolor(idx<<1,l,mid,a);
    else return getcolor(idx<<1|1,mid+1,r,a);
}
inline void solve1(int x,int y,int z)
{
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        color(1,1,sz,pos[top[x]],pos[x],z);
        x=father[top[x]];
    }
    if(pos[x]<pos[y])swap(x,y);
    color(1,1,sz,pos[y],pos[x],z);
}
inline void solve2(int x,int y)
{
    ans=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans+=query(1,1,sz,pos[top[x]],pos[x]);
        if(getcolor(1,1,sz,pos[top[x]])==getcolor(1,1,sz,pos[father[top[x]]]))ans--;
        x=father[top[x]];
    }
    if(pos[x]<pos[y])swap(x,y);
    ans+=query(1,1,sz,pos[y],pos[x]);
    printf("%d\n",ans);
}
int main()
{
    memset(head,-1,sizeof(head));
    n=read(),m=read();
    for(RG int i=1;i<=n;++i)col[i]=read();
    int x,y,z;
    for(RG int i=1;i<n;++i)
    {
        x=read(),y=read();
        addedge(x,y),addedge(y,x);
    }
    dfs1(1),dfs2(1,1);
    build(1,1,sz);
    char opt;
    for(RG int i=1;i<=m;++i)
    {
        opt=getchar();
        while(opt!=‘C‘&&opt!=‘Q‘)opt=getchar();
        if(opt==‘C‘)
        x=read(),y=read(),z=read(),solve1(x,y,z);
        else if(opt==‘Q‘)
        x=read(),y=read(),solve2(x,y);
    }
    return 0;
 }                  
时间: 2024-10-09 21:37:27

[BZOJ 2243][SDOI2011]染色(树链剖分+线段树)的相关文章

【BZOJ2243】[SDOI2011]染色 树链剖分+线段树

[BZOJ2243][SDOI2011]染色 Description 给定一棵有n个节点的无根树和m个操作,操作有2类: 1.将节点a到节点b路径上所有点都染成颜色c: 2.询问节点a到节点b路径上的颜色段数量(连续相同颜色被认为是同一段),如"112221"由3段组成:"11"."222"和"1". 请你写一个程序依次完成这m个操作. Input 第一行包含2个整数n和m,分别表示节点数和操作数: 第二行包含n个正整数表示n

BZOJ2243 [SDOI2011]染色(树链剖分+线段树合并)

题目链接 BZOJ2243 树链剖分+线段树合并 线段树合并的一些细节需要注意一下 #include <bits/stdc++.h> using namespace std; #define rep(i, a, b) for (int i(a); i <= (b); ++i) #define dec(i, a, b) for (int i(a); i >= (b); --i) typedef long long LL; const int N = 100010; int n, m,

【BZOJ】1146: [CTSC2008]网络管理Network(树链剖分+线段树套平衡树+二分 / dfs序+树状数组+主席树)

第一种做法(时间太感人): 这题我真的逗了,调了一下午,疯狂造数据,始终找不到错. 后来发现自己sb了,更新那里没有打id,直接套上u了.我.... 调了一下午啊!一下午的时光啊!本来说好中午A掉去学习第二种做法,噗 好吧,现在第一种做法是hld+seg+bst+二分,常数巨大,log^4级别,目前只会这种. 树剖后仍然用线段树维护dfs序区间,然后在每个区间建一颗平衡树,我用treap,(这题找最大啊,,,囧,并且要注意,这里的rank是比他大的数量,so,我们在二分时判断要判断一个范围,即要

bzoj 2157: 旅游【树链剖分+线段树】

裸的树链剖分+线段树 但是要注意一个地方--我WA了好几次才发现取完相反数之后max值和min值是要交换的-- #include<iostream> #include<cstdio> using namespace std; const int N=200005; int n,m,h[N],cnt,de[N],va[N],fa[N],si[N],hs[N],fr[N],id[N],tot,rl[N]; char c[10]; struct qwe { int ne,no,to,va

BZOJ2243 (树链剖分+线段树)

Problem 染色(BZOJ2243) 题目大意 给定一颗树,每个节点上有一种颜色. 要求支持两种操作: 操作1:将a->b上所有点染成一种颜色. 操作2:询问a->b上的颜色段数量. 解题分析 树链剖分+线段树. 开一个记录类型,记录某一段区间的信息.l 表示区间最左侧的颜色 , r 表示区间最右侧的颜色 , sum 表示区间中颜色段数量. 合并时判断一下左区间的右端点和有区间的左端点的颜色是否一样. 树上合并时需要用两个变量ans1,ans2来存储.ans1表示x往上走时形成的链的信息,

【bzoj1959】[Ahoi2005]LANE 航线规划 离线处理+树链剖分+线段树

题目描述 对Samuel星球的探险已经取得了非常巨大的成就,于是科学家们将目光投向了Samuel星球所在的星系——一个巨大的由千百万星球构成的Samuel星系. 星际空间站的Samuel II巨型计算机经过长期探测,已经锁定了Samuel星系中许多星球的空间坐标,并对这些星球从1开始编号1.2.3……. 一些先遣飞船已经出发,在星球之间开辟探险航线. 探险航线是双向的,例如从1号星球到3号星球开辟探险航线,那么从3号星球到1号星球也可以使用这条航线. 例如下图所示: 在5个星球之间,有5条探险航

Aizu 2450 Do use segment tree 树链剖分+线段树

Do use segment tree Time Limit: 1 Sec Memory Limit: 256 MB 题目连接 http://www.bnuoj.com/v3/problem_show.php?pid=39566 Description Given a tree with n (1 ≤ n ≤ 200,000) nodes and a list of q (1 ≤ q ≤ 100,000) queries, process the queries in order and out

Hdu 3966 Aragorn&#39;s Story (树链剖分 + 线段树区间更新)

题目链接: Hdu 3966 Aragorn's Story 题目描述: 给出一个树,每个节点都有一个权值,有三种操作: 1:( I, i, j, x ) 从i到j的路径上经过的节点全部都加上x: 2:( D, i, j, x ) 从i到j的路径上经过的节点全部都减去x: 3:(Q, x) 查询节点x的权值为多少? 解题思路: 可以用树链剖分对节点进行hash,然后用线段树维护(修改,查询),数据范围比较大,要对线段树进行区间更新 1 #include <cstdio> 2 #include

【bzoj3589】动态树 树链剖分+线段树

题目描述 别忘了这是一棵动态树, 每时每刻都是动态的. 小明要求你在这棵树上维护两种事件 事件0:这棵树长出了一些果子, 即某个子树中的每个节点都会长出K个果子. 事件1:小明希望你求出几条树枝上的果子数. 一条树枝其实就是一个从某个节点到根的路径的一段. 每次小明会选定一些树枝, 让你求出在这些树枝上的节点的果子数的和. 注意, 树枝之间可能会重合, 这时重合的部分的节点的果子只要算一次. 输入 第一行一个整数n(1<=n<=200,000), 即节点数. 接下来n-1行, 每行两个数字u,

bzoj4304 (树链剖分+线段树)

Problem T2 (bzoj4304 HAOI2015) 题目大意 给定一颗树,1为根节点,要求支持三种操作. 操作 1 :把某个节点 x 的点权增加 a . 操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a . 操作 3 :询问某个节点 x 到根的路径中所有点的点权和. 解题分析 练手题.树链剖分+线段树. 参考程序 1 #include <cstdio> 2 #include <cstring> 3 #include <cmath> 4 #incl