城市网络
作者:互联网
城市网络
Problem:
有一个树状的城市网络(即 n 个城市由 n-1 条道路连接的连通图),首都为 1 号城市,每个城市售卖价值为 a_i 的珠宝。
你是一个珠宝商,现在安排有 q 次行程,每次行程为从 u 号城市前往 v 号城市(走最短路径),保证 v 在 u 前往首都的最短路径上。 在每次行程开始时,你手上有价值为 c 的珠宝(每次行程可能不同),并且每经过一个城市时(包括 u 和 v ),假如那个城市中售卖的珠宝比你现在手上的每一种珠宝都要优秀(价值更高,即严格大于),那么你就会选择购入。
现在你想要对每一次行程,求出会进行多少次购买事件。
Input:
第一行,两个正整数 \(n , q (2 ≤ n ≤ 10^5 , 1 ≤ q ≤ 10^5)\)。 第二行,n 个正整数 \(a_i (1 ≤ a_i ≤ 10^5)\) 描述每个城市售卖的珠宝的价值。 接下来 n-1 行,每行描述一条道路 \(x , y (1 ≤ x,y ≤ n)\),表示有一条连接 x 和 y 的道路。 接下来 q 行,每行描述一次行程 \(u , v , c (1 ≤ u,v ≤ n , 1 ≤ c ≤ 10^5)\)。
Output:
对于每次行程输出一行,为所购买次数。
Example:
Input
5 4
3 5 1 2 4
1 2
1 3
2 4
3 5
4 2 1
4 2 2
4 2 3
5 1 5
Output
2
1
1
0
题解:
树上倍增,ST表,由于每次经过一个城市我们都会选择是否购买珠宝,决策后手上的珠宝都是经过路径上价值最大的珠宝,所以我们需要知道在需要经过的路径上,珠宝价值不断递增的城市有多少,本题如果用暴力方法,最差情况下会是O(n*q)的时间复杂度,显然这是不行的。本体应该用st表去维护每个节点上需要经过几个不断递增的节点,st表的递推式:
\[st[i][j]=st[st[i][j-1]][j-1] \]
\(st[i][j]\)表示从i点开始能买\(2^j\)个珠宝的点是哪个,所以如果一直\(st[i][0]\)的值,那么i往上买\(2^j\)个珠宝的点,可以由i往上买\(2^j-1\)个珠宝的点再买\(2^{j-1}\)个珠宝的点,所以递推式如上。对于\(st[i][0]\)我们也可以用倍增的方式求,但是我们用从大到小枚举j值,这样就可以找到最远的可以买到1个珠宝的位置(如果枚举到的位置比当前权值小则需要继续向上倍增(倍增长度减少一半,因为原长度倍增的话则相当于已经倍增过的长度))。
经过ST表处理后的树,对于每次查询我们只需要查询从u开始,且深度大于等于v的点数之和(映射到ST表中就是满足条件的倍增长度的总和)。
关于ST表:https://www.cnblogs.com/LeafLove/p/12345136.html
Code:
/**********************************************************
* @Author: Kirito
* @Date: 2020-07-27 08:15:25
* @Last Modified by: Kirito
* @Last Modified time: 2020-07-27 09:27:55
* @Remark:
**********************************************************/
#include <bits/stdc++.h>
#define lowbit(x) (x&(-x))
#define CSE(x,y) memset(x,y,sizeof(x))
#define INF 0x3f3f3f3f
#define Abs(x) (x>=0?x:(-x))
#define FAST ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll , ll> pll;
const int maxn=211111;
vector<int> v[maxn];
int a[maxn],f[maxn][20],dep[maxn],to[maxn];
void dfs(int p,int fa)
{
int x=fa;
for(int k=19;k>=0;k--){
if(f[x][k]&&a[f[x][k]]<=a[p]) x=f[x][k];
}
if(a[x]>a[p]) f[p][0]=x;
else f[p][0]=f[x][0];
for(int i=1;i<20;i++)
f[p][i]=f[f[p][i-1]][i-1];
dep[p]=dep[fa]+1;
for(auto t:v[p]){
if(t==fa) continue;
dfs(t,p);
}
return;
}
int n,q;
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.in","r",stdin);
#endif
FAST;
cin>>n>>q;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=0;i<n-1;i++){
int x,y;
cin>>x>>y;
v[x].push_back(y);
v[y].push_back(x);
}
for(int i=n+1;i<=n+q;i++){
int x,y,c;
cin>>x>>y>>c;
v[i].push_back(x);
v[x].push_back(i);
a[i]=c;
to[i-n]=y;
}
dfs(1,0);
for(int i=1;i<=q;i++){
int ans=0;
int x=i+n;
for(int k=19;k>=0;k--){
if(dep[f[x][k]]>=dep[to[i]]){
ans+=(1<<k);
x=f[x][k];
}
}
cout<<ans<<endl;
}
return 0;
}
标签:int,城市,网络,st,maxn,倍增,珠宝 来源: https://www.cnblogs.com/LeafLove/p/13384294.html