牛客 城市网络 倍增

城市网络

题目链接

题意

给一棵树，有点权，每次从u->v刚开始自带了一个权值val，然后在路上只要遇见权值比他现有的大的就买下来，权值更新为那个点的权值，问能买几个点。 v是u的祖先。1是根节点
注意 是只要见到比他当前大的就必须买，并不是让找方法买最多的。

题解

买的肯定是递增的。
思路：倍增 dp[i][j]表示从i点往上开始买2^j个能到哪个点。
于是有个新的问题 怎么统计dp[i][0] 也就是i的祖先中离i最近且权值大于i的那个点是谁。
可以用单调栈
然后又有一个问题：因为刚开始给了一个权值，得从比那个权值大的最下面的点开始跳。这个怎么找？
题解告诉我不用找。。 只用在u下面再连一个点就可以。把这个点的权值设成给的这个权值。可是这对其他的答案没影响吗？因为设的这个点没有儿子。并且是往上跳，所以肯定没有影响。

代码

#include <algorithm>
#include <cstdio>
#include <iostream>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <cmath>
#include <set>
#include <cstring>
#include <string>
#include <bitset>
#include <stdlib.h>
#include <time.h> 
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
#define st first
#define sd second
#define mkp make_pair
#define pb push_back
void wenjian(){freopen("concatenation.in","r",stdin);freopen("concatenation.out","w",stdout);}
void tempwj(){freopen("hash.in","r",stdin);freopen("hash.out","w",stdout);}
ll gcd(ll a,ll b){return b == 0 ? a : gcd(b,a % b);}
ll qpow(ll a,ll b,ll mod){a %= mod;ll ans = 1;while(b){if(b & 1)ans = ans * a % mod;a = a * a % mod;b >>= 1;}return ans;}
struct cmp{bool operator()(const pii & a, const pii & b){return a.second > b.second;}};
int lb(int x){return  x & -x;}

const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 9901;

const int M = 10000007;
const int maxn = 2e5+5;
int dp[maxn][29];
int mdp[maxn][29];
int dep[maxn];
int a[maxn];
std::vector<int> vv[maxn];
stack<int> ss;
void dfs(int x,int fa,int de)
{
	dep[x] = de;
	stack<int> st;
	while(!ss.empty())
	{
		if(a[ss.top()] <= a[x])
		{
			st.push(ss.top());
			ss.pop();
		}
		else
			break;
	}
	if(!ss.empty())
		dp[x][0] = ss.top();
	ss.push(x);
	for (int i = 0; i < vv[x].size(); i ++ )
	{
		int v = vv[x][i];
		if(v == fa)
			continue;
		dfs(v,x,de + 1);
	}
	while(!ss.empty() && dep[ss.top()] >= dep[x])
		ss.pop();
	while(!st.empty())
	{
		ss.push(st.top());
		st.pop();
	}
}
int n;
int m;
void init()
{
	for (int i = 1; i <= n + m; i ++ )
	{
		for (int j = 1; j <= 24; j ++ )
		{
			dp[i][j] = dp[dp[i][j - 1]][j - 1];
		}
	}
}

int get_ans(int x,int y)
{
	int ans =0;
	for(int i = 24; i >= 0; i -- )
	{
		if(dep[dp[x][i]] >= dep[y])
		{
			ans += 1 << i;
			x = dp[x][i];
		}
	}
	return ans;
}



pii pp[maxn];
int main()
{
	
	scanf("%d%d",&n,&m);
	for (int i = 1; i <= n; i ++ )
		scanf("%d",&a[i]);
	for (int i = 1; i < n; i ++ )
	{
		int x,y;
		scanf("%d%d",&x,&y);
		vv[x].pb(y);
		vv[y].pb(x);
	}

	for (int i = 1; i <= m; i ++ )
	{
		int x,y,val;
		scanf("%d%d%d",&x,&y,&val);
		vv[x].pb(i + n);
		vv[i + n].pb(x);
		a[i + n] = val;
		pp[i].st = x;
		pp[i].sd = y;
	}
	dfs(1,0,1);init();
	// for (int  i= 1; i <= n + m; i ++ )
	// {
	// 	for (int j = 0; j <= 5; j ++ )
	// 		printf("%d ",dp[i][j]);
	// 	printf("\n");
	// }
	
	for (int i = 1; i <= m; i ++ )
	{
		int ans = get_ans(i + n,pp[i].sd);
		printf("%d\n",ans);
	}
}