当前位置: 代码迷 >> 综合 >> HDU - 3974 Assign the task(线段树+dfs序)
  详细解决方案

HDU - 3974 Assign the task(线段树+dfs序)

热度:20   发布时间:2023-12-17 02:48:31.0

题意:给你一个树形结构,两种操作,第一种是把一个结点染色,这个结点的所有子结点都要染成同样的颜色。第二种操作是查询某个结点的颜色。这让我们想到线段树,但是线段树是一维直线的,怎么解决树形结构呢?

先介绍一下dfs序,大家都知道dfs搜索顺序,从根结开始不断向叶子结点单向搜索,如果到叶子结点就向上回溯一层,继续向其他子结点搜索。这就造成了一种特性,就是搜索完所有子结点会再回到父结点,再搜完父结点的兄弟结点再回到祖先结点。这是一种非常舒服的搜索顺序。跟这道题什么关系呢?在看到这一题说道要给一个结点染色,就会给这个子树染色,那我们为什么不直接把这个子树当成区间呢。我们只需要用dfs序输出这个树,每个父结点都会连着一段为其子树的区间。直接用线段树进行维护就行了。

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=5e4+7;
int n,m;
vector<int> adj[maxn];
int cnt[maxn],pos[maxn],c[maxn<<2];
bool vis[maxn];
int dfs(int u)
{pos[u]=++m;int res=1;for(int i=0;i<adj[u].size();i++){int v=adj[u][i];if(vis[v])  continue;vis[v]=true;res+=dfs(v);}return cnt[u]=res;
}void push_down(int now)
{if(c[now]==-1)  return ;c[now<<1]=c[now<<1|1]=c[now];c[now]=-1;
}void update(int now,int l,int r,int ul,int ur,int t)
{if(ul<=l&&ur>=r){c[now]=t;return ;}push_down(now);int mid=(l+r)>>1;if(ul<=mid) update(now<<1,l,mid,ul,ur,t);if(ur>mid)  update(now<<1|1,mid+1,r,ul,ur,t);
}int query(int now,int l,int r,int p)
{if(l==r)    return c[now];int mid=(l+r)>>1;push_down(now);if(p<=mid)  return query(now<<1,l,mid,p);else return query(now<<1|1,mid+1,r,p);
}int main()
{int T,kase=1;scanf("%d",&T);while(T--){int u,v;scanf("%d",&n);for(int i=1;i<=n;i++)   adj[i].clear();memset(cnt,0,sizeof(cnt));for(int i=1;i<n;i++){scanf("%d%d",&u,&v);adj[v].push_back(u);cnt[u]++;}for(int i=1;i<=n;i++){if(!cnt[i]){m=0;memset(cnt,0,sizeof(cnt));memset(vis,false,sizeof(vis));dfs(i);break;}}memset(c,-1,sizeof(c));scanf("%d",&m);char option[3];printf("Case #%d:\n",kase++);for(int i=0;i<m;i++){scanf("%s",option);if(option[0]=='C'){scanf("%d",&u);printf("%d\n",query(1,1,n,pos[u]));}else{scanf("%d%d",&u,&v);update(1,1,n,pos[u],pos[u]+cnt[u]-1,v);}}}
}

 

  相关解决方案