1103: [POI2007]大都市meg

最新推荐文章于 2018-03-04 14:38:00 发布

原创最新推荐文章于 2018-03-04 14:38:00 发布 · 318 阅读

0 ·

CC 4.0 BY-SA版权

1000-1250 专栏收录该内容

91 篇文章

订阅专栏

本文介绍了一种使用DFS序和树状数组解决边权修改与距离查询问题的方法。通过预处理得到每个节点的进入时间和退出时间，利用树状数组进行区间修改与查询操作。

题目链接

题目大意：将边权修改为0（初始为1）和查询某个点到1的距离

题解：题意杀……
dfs序搞搞就行了

我的收获：……。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
 
#define M 250005
 
int n,m,t,tim,c[M],x,y;
int head[M],in[M],out[M],vis[M];
 
struct edge{int to,nex;}e[M*4];
 
void add(int u,int v){e[t]=(edge){v,head[u]};head[u]=t++;}
 
void updata(int x,int v){
    for(int i=x;i<=n;i+=i&(-i)) c[i]+=v;
}
 
int query(int x){
    int ret=0;
    for(int i=x;i>0;i-=i&(-i)) ret+=c[i];
    return ret;
}
 
void dfs(int x){
    in[x]=++tim;
    for(int i=head[x];i!=-1;i=e[i].nex) dfs(e[i].to);
    out[x]=tim;
//这里不用++
}
 
void work()
{
    char opt[5];
    for(int i=1;i<m+n;i++){
        scanf("%s%d",opt,&x);
        if(opt[0]=='W') printf("%d\n",query(in[x]));
        else {scanf("%d",&y);if(x>y) swap(x,y);updata(in[y],-1),updata(out[y]+1,1);}
//清零
    }
}
 
void init()
{
    cin>>n;t=0;memset(head,-1,sizeof(head));
    for(int i=1;i<n;i++) scanf("%d%d",&x,&y),add(x,y);
    dfs(1);cin>>m;
    for(int i=2;i<=n;i++) updata(in[i],1),updata(out[i]+1,-1);
// without 1!!!
}
 
int main()
{
    init();
    work();
    return 0;
}