洛谷P2982 [USACO10FEB]慢下来Slowing down(线段树 DFS序 区间增减 单点查询)

To 洛谷.2982 慢下来Slowing down

题目描述

Every day each of Farmer John's N (1 <= N <= 100,000) cows conveniently numbered 1..N move from the barn to her private pasture. The pastures are organized as a tree, with the barn being on pasture 1. Exactly N-1 cow unidirectional paths connect the pastures; directly connected pastures have exactly one path. Path i connects pastures A_i and B_i (1 <= A_i <= N; 1 <= B_i <= N).

Cow i has a private pasture P_i (1 <= P_i <= N). The barn's small door lets only one cow exit at a time; and the patient cows wait until their predecessor arrives at her private pasture. First cow 1 exits and moves to pasture P_1. Then cow 2 exits and goes to pasture P_2, and so on.

While cow i walks to P_i she might or might not pass through a pasture that already contains an eating cow. When a cow is present in a pasture, cow i walks slower than usual to prevent annoying her friend.

Consider the following pasture network, where the number between
parentheses indicates the pastures' owner.

        1 (3)        
       / 
  (1) 4   3 (5)
     /    
(2) 2   5 (4)

First, cow 1 walks to her pasture:

        1 (3)        
       / 
  [1] 4*  3 (5)
     /    
(2) 2   5 (4)

When cow 2 moves to her pasture, she first passes into the barn's
pasture, pasture 1. Then she sneaks around cow 1 in pasture 4 before
arriving at her own pasture.

        1 (3)
       / 
  [1] 4*  3 (5)
     /    
[2] 2*  5 (4)

Cow 3 doesn't get far at all -- she lounges in the barn's pasture, #1.

        1* [3]
       / 
  [1] 4*  3 (5)
     /    
[2] 2*  5 (4)

Cow 4 must slow for pasture 1 and 4 on her way to pasture 5:

        1* [3]
       / 
  [1] 4*  3 (5)
     /    
[2] 2*  5* [4]

Cow 5 slows for cow 3 in pasture 1 and then enters her own private pasture:

        1* [3]
       / 
  [1] 4*  3*[5]
     /    
[2] 2*  5* [4]

FJ would like to know how many times each cow has to slow down.

每天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会放慢自己的速度，防止打扰他的朋友。

FJ想要知道奶牛们总共要放慢多少次速度。

输入输出格式

输入格式：

• Line 1: Line 1 contains a single integer: N

• Lines 2..N: Line i+1 contains two space-separated integers: A_i and B_i

• Lines N+1..N+N: line N+i contains a single integer: P_i

输出格式：

• Lines 1..N: Line i contains the number of times cow i has to slow down.

输入输出样例

输入样例#1：

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

输出样例#1：

0 
1 
0 
2 
1 

代码：

#include<cstdio>
#include<cctype>
using namespace std;
const int N=100005;

int n,ENum,cnt,H[N<<1],Sum[N<<2],Dfn[N],Size[N];
struct Edge
{
    int to,nxt;
}e[N<<1];

void read(int &now)
{
    now=0;char c=getchar();
    for(;!isdigit(c);c=getchar());
    for(;isdigit(c);c=getchar())
      now=(now<<3)+(now<<1)+c-'0';
}

void AddEdge(int u,int v)
{
    e[++ENum].to = v;
    e[ENum].nxt = H[u];
    H[u] = ENum;
}

void DFS(int x)
{
    Size[x]=1;//根的子树的大小(包括自己) 
    Dfn[x]=++cnt;//Dfn[i]:i在dfs序中的下标 
    for(int i=H[x];i;i=e[i].nxt)
    {
        int to=e[i].to;
        if(Dfn[to]) continue;
        DFS(to);
        Size[x]+=Size[to];
    }
}

void PushUp(int rt)
{
    Sum[rt]=Sum[rt<<1]+Sum[rt<<1|1];
}
void PushDown(int rt)
{
    if(!Sum[rt]) return;
    Sum[rt<<1]+=Sum[rt];
    Sum[rt<<1|1]+=Sum[rt];
    Sum[rt]=0;
}
void ModifySum(int l,int r,int rt,int L,int R)
{
    if(L<=l && r<=R)
    {
        ++Sum[rt];//区间修改时 针对本题 懒标记+1 
        return;
    }
    PushDown(rt);
    int m=(l+r)>>1;
    if(L<=m) ModifySum(l,m,rt<<1,L,R);
    if(m<R) ModifySum(m+1,r,rt<<1|1,L,R);
    //PushUp(rt); 不需要进行PushUp!! 
}
int QuerySum(int l,int r,int rt,int p)
{
    if(l==r) return Sum[rt];
    PushDown(rt);
    int m=(l+r)>>1,res=0;
    if(p<=m) res+=QuerySum(l,m,rt<<1,p);
    else res+=QuerySum(m+1,r,rt<<1|1,p);
    return res;
}

int main()
{
    read(n);
    for(int i=1;i<n;i++)
    {
        int a,b;
        read(a);read(b);
        AddEdge(a,b);
        AddEdge(b,a);
    }
    DFS(1);
//    for(int i=1;i<=n;i++)
//      printf("%d:dfn:%d  size:%d
",i,Dfn[i],Size[i]);
    for(int i=1;i<=n;i++)
    {
        int a;
        read(a);
        printf("%d
",QuerySum(1,n,1,Dfn[a]));//先经过后更改 
        ModifySum(1,n,1,Dfn[a],Dfn[a]+Size[a]-1);//单点查询 区间修改 
    }
    return 0;
}