1103: [POI2007]大都市meg

来源:互联网 发布:淘宝宝贝改价格会影响排名吗 编辑:程序博客网 时间:2024/05/17 06:15

题目链接

题目大意:将边权修改为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;}