码迷,mamicode.com
首页 > 其他好文 > 详细

BZOJ 1103 大都市

时间:2016-08-31 22:08:02      阅读:120      评论:0      收藏:0      [点我收藏+]

标签:

dfs序+BIT。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxv 250050
#define maxe 500500
using namespace std;
int n,x,y,m,g[maxv],nume=0,w[maxv],mx[maxv],dis[maxv],val[maxv],fath[maxv],times=0;
char s[5];
struct edge
{
    int v,nxt;
}e[maxe];
int lowbit(int x)
{
    return (x&(-x));
}
void addedge(int u,int v)
{
    e[++nume].v=v;
    e[nume].nxt=g[u];
    g[u]=nume;
}
void dfs(int x)
{
    w[x]=mx[x]=++times;
    for (int i=g[x];i;i=e[i].nxt)
    {
        int v=e[i].v;
        if (v!=fath[x])
        {
            dis[v]=dis[x]+1;fath[v]=x;
            dfs(v);
            mx[x]=max(mx[x],mx[v]);
        }
    }
}
void add(int pos,int x)
{
    for (int i=pos;i<=n+1;i+=lowbit(i))
        val[i]+=x;
}
void modify()
{
    scanf("%d%d",&x,&y);
    if (dis[x]>dis[y]) swap(x,y);
    add(w[y],1);add(mx[y]+1,-1);
}
void ask()
{
    scanf("%d",&x);int regis=x;x=w[x];
    int ret=0;
    while (x)
    {
        ret+=val[x];
        x-=lowbit(x);
    }
    printf("%d\n",dis[regis]-ret);
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n-1;i++)
    {
        scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);    
    }    
    dfs(1);
    scanf("%d",&m);
    for (int i=1;i<=n+m-1;i++)
    {
        scanf("%s",s);
        if (s[0]==A) modify();
        else ask();
    }
    return 0;
}

 

BZOJ 1103 大都市

标签:

原文地址:http://www.cnblogs.com/ziliuziliu/p/5827623.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!