BZOJ1782: [Usaco2010 Feb]slowdown 慢慢游

1782: [Usaco2010 Feb]slowdown 慢慢游

Time Limit: 1 Sec  Memory Limit: 64 MB
Submit: 541  Solved: 326
[Submit][Status]

Description

每天Farmer John的N头奶牛(1 <= N <= 100000，编号1…N)从粮仓走向他的自己的牧场。牧场构成了一棵树，粮仓在1号牧场。恰好有N-1条道路直接连接着牧场，使得牧场之间都恰好有一条路径相连。第i条路连接着A_i，B_i，(1 <= A_i <= N; 1 <= B_i <= N)。奶牛们每人有一个私人牧场P_i (1 <= P_i <= N)。粮仓的门每次只能让一只奶牛离开。耐心的奶牛们会等到他们的前面的朋友们到达了自己的私人牧场后才离开。首先奶牛1离开，前往P_1；然后是奶牛2，以此类推。当奶牛i走向牧场P_i时候，他可能会经过正在吃草的同伴旁。当路过已经有奶牛的牧场时，奶牛i会放慢自己的速度，防止打扰他的朋友。 考虑如下的牧场结构（括号内的数字代表了牧场的所有者）。 

Input

* 第1行 : 一个正整数N * 第2…N行: 第i+1行包括一对正整数A_i,B_i * 第N+1..N+N行: 第 N+i行 包括一个正整数: P_i

Output

* 第一行到第N行：第i行表示第i只奶牛需要被放慢的次数

Sample Input

5
1 4
5 4
1 3
2 4
4
2
1
5
3

Sample Output

0
1
0
2
1

HINT

 

Source

Gold

题解：

树状数组+dfs序。。。大都市meg的简化版

代码：

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<string>
#define inf 1000000000
#define maxn 100000+1000
#define maxm 500+100
#define eps 1e-10
#define ll long long
#define pa pair<int,int>
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{int go,next;}e[2*maxn];
int n,tot,ti,s[2*maxn],l[maxn],r[maxn],head[maxn];
bool v[maxn];
void insert(int x,int y)
{
    e[++tot].go=y;e[tot].next=head[x];head[x]=tot;
    e[++tot].go=x;e[tot].next=head[y];head[y]=tot;
}
void dfs(int x)
{
    v[x]=1;
    l[x]=++ti;
    for(int i=head[x],y;i;i=e[i].next)
     if(!v[y=e[i].go])dfs(y);
    r[x]=++ti; 
}
void add(int x,int y)
{
    for(;x<=2*n;x+=x&(-x))s[x]+=y;
}
int sum(int x)
{
    int t=0;
    for(;x;x-=x&(-x))t+=s[x];
    return t;
}
int main()
{
    freopen("input.txt","r",stdin);
    freopen("output.txt","w",stdout);
    n=read();
    int x,y;
    for(int i=1;i<n;i++)x=read(),y=read(),insert(x,y);
    dfs(1);
    for(int i=1;i<=n;i++)
     {
         int x=read();
         printf("%d\n",sum(l[x]));
         add(l[x],1);add(r[x],-1);
     } 
    return 0;
}