LightOJ 1128 Greatest Parent

題面

題意

給出一棵樹,每個點都有一個權,每次查詢一個點的祖先中比某個數大的最大深度.

已知父節點的權絕對嚴格小于其子節點

方法

建樹之後,利用倍增進行與二分類似的思想進行查找,fa[i][j]記錄點i向上(1 << j)代的父親.

每次查找都找不超過目前範圍的最大祖宗.

代碼

#include<bits/stdc++.h>
#define N 500100
using namespace std;

int T,TT,n,m,dn[N],fa[N][25],dq,last,ds[N];
struct Bn
{
	int next,to;
}bn[200100];

inline int find(int l,int r)
{
	last--;
	if(l==r||dn[l]>=dq&&dn[fa[l][0]]<dq) return l;
	while(fa[l][last]==-1||dn[fa[l][last]]<dn[r])
		last--;
	if(dn[fa[l][last]]<dq)
	{
		return find(l,fa[l][last]);
	}
	else if(dn[fa[l][last]]>dq)
	{
		return find(fa[l][last],r);
	}
	return fa[l][last];
}

int main()
{
//	freopen("1.txt","r",stdin);
//	freopen("2.txt","w",stdout);
	int i,j,p,q;
	cin>>T;
	TT=T;
	while(T--)
	{
		memset(fa,-1,sizeof(fa));
		scanf("%d%d",&n,&m);
		for(i=1;i<=n-1;i++)
		{
			scanf("%d%d",&p,&q);
			dn[i]=q;
			fa[i][0]=p;
			ds[p]++;
		}
		dn[0]=1;
		
		printf("Case %d:\n",TT-T);
		for(i=1;i<=20;i++)
		{
			for(j=0;j<n;j++)
			{
				if(fa[j][i-1]==-1) continue;
				fa[j][i]=fa[fa[j][i-1]][i-1];
			}
		}
		for(i=1;i<=m;i++)
		{
			scanf("%d%d",&p,&dq);
			last=20;
			printf("%d\n",find(p,0));
		}
	}
}