HYSBZ - 2243 [SDOI 2011] 염색 (트 리 체인 분할)
5208 단어 데이터 구조 - 트 리 체인 분할
제목: 중국어 문제 의 뜻 이 뚜렷 하 다.
분석: 트 리 체인 분할 + 선분 트 리 (구간 조회, 단점 조회, 구간 통합 + 구간 통합), 업데이트 시 하나의 결점 에 포 함 된 구간 이 아니라면 두 구간 의 인접 결점 색상 이 동일 한 지, Find 시 에 도 두 체인 의 인접 결점 색상 이 동일 한 지,이곳 의 인접 은 한 점 에서 한 변 만 가면 다른 결점 에 도착 할 수 있다 는 것 을 가리킨다.
참조 코드:
#include
#include
#include
#include
#include
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
#define lson rt<<1
#define rson rt<<1|1
const int maxn = 1e5+10;
int n,m;
int col[maxn];//
int head[maxn],tot;
struct Edge{
int to,next;
}edge[maxn<<1];
int siz[maxn],fa[maxn],son[maxn],dep[maxn];
int p[maxn],fp[maxn],top[maxn],poi;
// ,
int sum[maxn<<2],lc[maxn<<2],rc[maxn<<2],tag[maxn<<2];// ,
void Init()
{
tot = poi = 0;
mem(head,-1);
mem(son,-1);
}
void AddEdge( int u, int v)
{
edge[tot].to = v;
edge[tot].next = head[u];
head[u] = tot++;
}
void dfs1( int u, int pre, int d)
{
siz[u] = 1;
fa[u] = pre;
dep[u] = d;
for( int i = head[u]; ~i; i = edge[i].next)
{
int v = edge[i].to;
if( v != pre)
{
dfs1(v,u,d+1);
siz[u] += siz[v];
if( son[u] == -1 || siz[v] > siz[son[u]])
son[u] = v;
}
}
}
void dfs2( int u, int tp)
{
top[u] = tp;
p[u] = ++poi;
fp[p[u]] = u;
if( son[u] == -1)
return;
dfs2(son[u],tp);
for( int i = head[u]; ~i; i = edge[i].next)
{
int v = edge[i].to;
if( v != fa[u] && v!= son[u])
dfs2(v,v);
}
}
void PushUp( int rt)
{
sum[rt] = sum[lson]+sum[rson]-(rc[lson]==lc[rson]);
lc[rt] = lc[lson];
rc[rt] = rc[rson];
}
void Build( int l, int r, int rt)
{
tag[rt] = 0;
if( l == r)
{
sum[rt] = 1;
lc[rt] = rc[rt] = col[fp[l]];
return;
}
int mid = (l+r)>>1;
Build(l,mid,lson);
Build(mid+1,r,rson);
PushUp(rt);
}
void PushDown( int rt)
{
if( tag[rt])
{
sum[lson] = sum[rson] = 1;
lc[lson] = lc[rson] = rc[lson] = rc[rson] = lc[rt];
tag[lson] = tag[rson] = tag[rt];
tag[rt] = 0;
}
}
void Update( int l, int r, int rt, int L, int R, int color)
{
if( L == l && R == r)
{
sum[rt] = 1;
lc[rt] = rc[rt] = color;
tag[rt] = 1;
return;
}
PushDown(rt);
int mid = (l+r)>>1;
if( R <= mid)
Update(l,mid,lson,L,R,color);
else if( L > mid)
Update(mid+1,r,rson,L,R,color);
else
{
Update(l,mid,lson,L,mid,color);
Update(mid+1,r,rson,mid+1,R,color);
}
PushUp(rt);
}
int Query( int l, int r, int rt, int L, int R)
{
if( L == l && R == r)
return sum[rt];
PushDown(rt);
int mid = (l+r)>>1;
if( R <= mid)
return Query(l,mid,lson,L,R);
else if( L > mid)
return Query(mid+1,r,rson,L,R);
else
return Query(l,mid,lson,L,mid)+Query(mid+1,r,rson,mid+1,R)-(rc[lson]==lc[rson]);
}
void Change( int u, int v, int color)
{
int t1 = top[u];
int t2 = top[v];
while( t1 != t2)
{
if( dep[t1] < dep[t2])
{
swap(u,v);
swap(t1,t2);
}
Update(1,poi,1,p[t1],p[u],color);
u = fa[t1];
t1 = top[u];
}
if( dep[u] > dep[v])
swap(u,v);
Update(1,poi,1,p[u],p[v],color);
}
int Check( int l, int r, int rt, int pos)
{
if( l == pos && r == pos)
return lc[rt];
PushDown(rt);
int mid = (l+r)>>1;
if( pos <= mid)
return Check(l,mid,lson,pos);
else
return Check(mid+1,r,rson,pos);
}
int Find( int u, int v)
{
int ans = 0;
int t1 = top[u];
int t2 = top[v];
while( t1 != t2)
{
if( dep[t1] < dep[t2])
{
swap(u,v);
swap(t1,t2);
}
ans += Query(1,poi,1,p[t1],p[u]);
if( Check(1,poi,1,p[t1]) == Check(1,poi,1,p[fa[t1]]))
ans--;
u = fa[t1];
t1 = top[u];
}
if( dep[u] > dep[v])
swap(u,v);
ans += Query(1,poi,1,p[u],p[v]);
return ans;
}
int main()
{
while( ~scanf("%d%d",&n,&m))
{
Init();
for( int i = 1; i <= n; i++)
scanf("%d",&col[i]);
int u,v;
for( int i = 1; i < n; i++)
{
scanf("%d%d",&u,&v);
AddEdge(u,v);
AddEdge(v,u);
}
dfs1(1,0,0);
dfs2(1,1);
Build(1,poi,1);
while( m--)
{
char op[3];
int a,b,color;
scanf("%s",op);
if( op[0] == 'C')
{
scanf("%d%d%d",&a,&b,&color);
Change(a,b,color);
}
else if( op[0] == 'Q')
{
scanf("%d%d",&a,&b);
printf("%d
",Find(a,b));
}
}
}
return 0;
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
hdu 3966 Aragorn 's Story (나무 사슬 분할 + 나무 모양 배열)제목 링크: hdu 3966 Aragorn 's Story 제목: 나무 한 그루 를 정 하고 세 가지 조작 Q x: 노드 x 의 값 조회 I x y w: 노드 x 에서 y 까지 이 경로 의 모든 노드 의 값 증가 w...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.