洛谷2982 USACO10FEB 慢下来 Slowing down

题目：Slowing down

思路：

线段树。

如果有一头牛回到了自己的牧场，那么还没有回牧场的所有牧场在它的子树中的牛，建树的次数就会加一。

先对树dfs遍历，得到dfs序。设a[i]表示第i头牛需要减速的次数。

一头牛i回到牧场，就先查询a[i]，再把dfn[i]~low[i]的所有节点的值加一。

这样问题就可以转化为线段树的区间修改单点查询了。

代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <cstring>
#include <map>
using namespace std;

#define maxn 100000

int n;
vector<int> tr[maxn+5];
int pi[maxn+5];

int dfn[maxn+5],low[maxn+5];
int cnt=0;

int a[4*maxn+5]={0};
int lzy[4*maxn+5]={0};
int p,q;

void readin(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		tr[x].push_back(y),tr[y].push_back(x);
	}
	for(int i=1;i<=n;i++) scanf("%d",&pi[i]);
}

void dfs(int x,int fa){
	dfn[x]=++cnt;
	for(int i=0;i<tr[x].size();i++){
		if(tr[x][i]!=fa) dfs(tr[x][i],x);
	}
	low[x]=cnt;
}

void push_down(int o,int lc,int rc){
	lzy[lc]+=lzy[o],lzy[rc]+=lzy[o];
	a[lc]+=lzy[o],a[rc]+=lzy[o];
	lzy[o]=0;
}

int update(int o,int l,int r){
	if(p<=l&&r<=q){
		lzy[o]++,a[o]++;
		return a[o];
	}
	if(l>q||r<p) return 0;
	
	int lc=o*2,rc=o*2+1;
	int mid=l+(r-l)/2;
	push_down(o,lc,rc);
	a[o]=update(lc,l,mid)+update(rc,mid+1,r);
	return a[o];
}

int query(int o,int l,int r){
	if(l>p||r<p) return 0;
	if(l==r) return a[o];
	
	int lc=o*2,rc=o*2+1;
	int mid=l+(r-l)/2;
	push_down(o,lc,rc);
	return query(lc,l,mid)+query(rc,mid+1,r);
}

void slv(){
	for(int i=1;i<=n;i++){
		p=dfn[pi[i]],q=low[pi[i]];
		printf("%d\n",query(1,1,n));
		update(1,1,n);
	}
}

int main() {
	readin();
	dfs(1,0);
	slv();
	return 0;
}