F - Exactly K Steps(树的直径 + 倍增)

F - Exactly K Steps

树的直径 + 倍增

分析

首先树的直径没啥好说的…
离一个点u最远的点一定是树的直径的两个端点;

所以我们首先要找一下树的直径,这个简单;
关键是对于每个询问我们要怎么快速找到对应的点?

我们可以利用这个性质:离一个点u最远的点一定是树的直径的两个端点(root1,root2);
那么可以知道如果这个点存在,u到max_dis(root1, root2)的路径上则一定会有一个点满足;
因此我们可以分别以root1,root2作为树根形成两棵树,然后对于u,在两棵树上往上找距离为k的点;

这个其实是比较难搞的(如果没学过lca的话),说起来我倍增好像是在学lca的时候学的…
我们可以借助倍增,f[u][i]表示u向上跳2^i的距离所能到的点是哪个;

下面考虑怎么预处理出这个数组:
这里设x是u的子节点;
那么f[x][0] = u;
f[x][i] = f[f[x][i-1]][i-1];
即x跳2^i步到的点 = x跳2^(i-1)步所到的点 再跳 2^(i-1)步;
(2^i = 2^(i-1) * 2)(跳两次i-1);

那么对于每个询问:u,k;
我们直接从大到小看i,考虑是否k >= 2^i;
如果可以跳,则u = f[u][i],一直往上跳,然后k -= 2^i;

代码

#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<map>
#include<algorithm>
#include<deque>
#include<stack>
#include<set>
#include <random>
#include<bitset>
// #include <unordered_map>
#include<math.h>
#include<string.h>
#define IOS ios::sync_with_stdio(false),cin.tie(0);
using namespace std;
 
#define pb push_back
#define coutl cout<<"------------"<<endl;
#define fi first
#define se second

#define ire(x) scanf("%d",&x)
#define iire(a,b) scanf("%d %d",&a,&b)
#define lre(x) scanf("%lld",&x)
#define llre(a,b) scanf("%lld %lld",&a,&b)
#define dre(x) scanf("%lf",&x)
#define ddre(a,b) scanf("%lf %lf",&a,&b)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define endl "\n"
#define PI acos(-1.0)
//#define int long long
// #define double long double
// typedef __int128 ll;
typedef long long ll;
typedef unsigned long long ull;

typedef pair<int, int> PII;
typedef pair<double, int> PDI;
typedef pair<ll, ll> PLL;
typedef pair<double, double> PDD;
typedef pair<double, pair<int, double> > PDID;
typedef pair<char, char> PCC;
typedef pair<char, pair<int, int> > PCII;
typedef pair<int, pair<int, int> > PIII;
typedef pair<int, pair<int, pair<int, int> > > PIIII;
typedef pair<ll, pair<int, int> > PLII;

const int maxn = 3e5 + 7;
const int N = 5e5 + 7;
const int M = 5e5 + 7;
const int mod = 19980829;
const int inv = mod - mod/2;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double pi = acos(-1);
const double eps = 1e-8;
  
ll gcd(ll a,ll b) {return b==0 ? a : gcd(b,a%b);}
ll lcm(ll a,ll b) {return a*b / gcd(a,b);}
ll qmi(ll a,ll b,ll p) {ll ans = 1; while(b) { if(b & 1) ans = ans * a % p; a = a * a % p; b >>= 1; } return ans;}
int lowbit(int x) {return x & (-x);}

vector<int> g[maxn];
int dep[3][maxn];
int f[3][maxn][20];
int n;
int vis[maxn];
int root1,root2;

void dfs(int u,int fa,int p)
{
	for(auto x : g[u])
	{
		if(fa == x) continue;
		
		dep[p][x] = dep[p][u] + 1;
		
		dfs(x,u,p);
	}
}

void sol()	//求树的直径
{
	dfs(1,-1,0);
	int mx = 0;
	for(int i=1;i<=n;i++)
		if(dep[0][i] > mx)
		{
			mx = dep[0][i];
			root1 = i;
		}
	
	rep(i,1,n) dep[0][i] = 0;
	
	dfs(root1,-1,0);
	mx = 0;
	for(int i=1;i<=n;i++)
		if(dep[0][i] > mx)
		{
			mx = dep[0][i];
			root2 = i;
		}
}

void bfs(int root,int p)  //预处理depth[] 和 f[][]数组, 从根开始向下更新 
{
    memset(vis,0,sizeof vis);

    queue<int> q;
    q.push(root);

    while(q.size())
    {
        int no = q.front(); //父节点 
        q.pop();

		vis[no] = 1;
		
        for(int j : g[no])    //子节点 
        {
        	if(vis[j]) continue;
        	
        	f[p][j][0] = no;   //子节点跳一步到父节点
            for(int k=1; k<20; k++)    //枚举往上跳2^k步
               f[p][j][k] = f[p][f[p][j][k-1]][k-1];    //2^k = 2^(k-1) + 2^(k-1)
        	
        	q.push(j);
        }
    }
}

void solve()
{
	cin>>n;
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		g[a].push_back(b);
		g[b].push_back(a);
	}
	
	sol();
	
	dfs(root1,-1,1);
	dfs(root2,-1,2);
	
	bfs(root1,1);
	bfs(root2,2);
	
	int q;
	cin>>q;
	while(q--)
	{
		int u,k;
		cin>>u>>k;
		
		if(dep[1][u] < k && dep[2][u] < k)
		{
			cout<<-1<<'\n';
			continue;
		}
		
		int mx = dep[1][u] > dep[2][u] ? 1 : 2;
		
		int i = 19;
		while(i >= 0)
		{
		    int dis = pow(2,i);
		    while(k >= dis)
		    {
		        u = f[mx][u][i];
		        k -= pow(2,i);
		    }
		    i--;
		}
		
		cout<<u<<'\n';
	}
}

int main()
{
//	IOS;
	
	int t;
//	ire(t);
//	cin>>t;
	t = 1;
	while(t--)
	{
		solve();
	}
    	
 	return 0;
}
	
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值