【USACO 2010FEB】 slowdown

【題目鏈接】

           點擊打開鏈接

【算法】

           dfs序 + 線段樹

           樹鏈剖分同樣可以解決這個問題

【代碼】

            

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010

int i,n,p,a,b,timer;
int tag[MAXN<<2],sum[MAXN<<2],size[MAXN],fa[MAXN],dfn[MAXN];

vector< int > e[MAXN];

inline void dfs(int u)
{
		int i,v;
		dfn[u] = ++timer;
		size[u]	= 1;
		for (i = 0; i < e[u].size(); i++)
		{
				v = e[u][i];
				if (fa[u] != v)
				{
						fa[v] = u;
						dfs(v);
						size[u] += size[v];		
				}	
		}
}
inline void pushdown(int index)
{
		if (tag[index])
		{
				sum[index<<1] += tag[index];
				sum[index<<1|1] += tag[index];
				tag[index<<1] += tag[index];
				tag[index<<1|1] += tag[index];
				tag[index] = 0;
		}
}
inline void update(int index)
{
		sum[index] = sum[index<<1] + sum[index<<1|1];
} 
inline void modify(int index,int l,int r,int ql,int qr,int val)
{
		int mid;
		if (l == ql && r == qr)
		{
				sum[index] += val;
				tag[index] += val;
				return;
		}
		pushdown(index);
		mid = (l + r) >> 1;
		if (mid >= qr) modify(index<<1,l,mid,ql,qr,val);
		else if (mid + 1 <= ql) modify(index<<1|1,mid+1,r,ql,qr,val);
		else
		{
				modify(index<<1,l,mid,ql,mid,val);
				modify(index<<1|1,mid+1,r,mid+1,qr,val);
		}
		update(index);
}
inline int query(int index,int l,int r,int pos)
{
		int mid;
		if (l == r) return sum[index];
		pushdown(index);
		mid = (l + r) >> 1;
		if (mid >= pos) return query(index<<1,l,mid,pos);
		else return query(index<<1|1,mid+1,r,pos);		
}

int main() {
		
		scanf("%d",&n);
		for (i = 1; i < n; i++)
		{
				scanf("%d%d",&a,&b);
				e[a].push_back(b);
				e[b].push_back(a);
		}
		dfs(1);
		for (i = 1; i <= n; i++)
		{
				scanf("%d",&p);
				printf("%d\n",query(1,1,n,dfn[p]));
				modify(1,1,n,dfn[p],dfn[p]+size[p]-1,1);	
		}
		
		return 0;
	
}