题目
题意: 给定n个点m条边的无向图,每个点具有一个点权,到达之后可以累加上这个权值的能力,而通过某一条边需要当前的能力值大于这条边的边权才可以(通过后能力值不会减少)。有q次查询,每次查询给定一个点和初始值,问能力值最多是多少。(n,m,q<=1e5)
思路: 一眼顶针,贪心就行了,对于单次查询起始是一个构建最小生成树的过程。但是有1e5次查询,就不能这么暴力咧。
考虑Kruskal重构树(都没学过这玩意),在构建最小生成树的时候额外加入n-1个点,每个点的点权是构建最小生成树时选择边的边权。
非常类似哈夫曼树的构建,每次选俩最小的然后合成咧。
这个玩意有什么用呢?叶子到某个节点这条路径可以表征构建树的过程,如果能从叶子到某个节点u,u的子树中所有叶子都能产生贡献(对应原题里的获得边权)。
所以建好树之后,我们就可以对于每个查询从叶子开始往上爬,如果父节点的点权(对应选择边的边权,也就是合成的代价) <= 初始值+累计获得的点权和,就可以往上爬。但是这样太慢了,如果是一条链的话还是1e10的复杂度.
可以打倍增,logn的时间求出节点i往上跳2^j的路径上的最长边。最长边是限制这个节点能否跳过去的关键,如果当前的点权和+初始值>=最长边,就可以跳了。具体细节看代码,铜牌题好难捏。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
const int M = N;
int a[N];
int h[N],e[M],ne[M],w[M],idx = 0;
int fa[N][18];
int mx[N][18]; //i往上跳2^j的边的最大限制
int n,m,k,T; int cnt,lg;
int f[N]; //并查集
int ww[N]; //边权限制
struct node{
int x,y,z;
bool operator<(const node&rhs)const
{
return z < rhs.z;
}
}q[N];
int find(int x)
{
return x==f[x]?x:f[x]=find(f[x]);
}
void add(int a,int b,int c)
{
e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;
}
void Kru()
{
for(int i=0;i<m;++i)
{
int x = q[i].x;
int y = q[i].y;
int z = q[i].z;
int u = find(x),v = find(y);
if(u!=v)
{
a[++cnt] = a[u] + a[v];
ww[cnt] = z;
f[u] = f[v] = cnt;
add(cnt,u,z),add(cnt,v,z);
}
}
}
void dfs(int cur,int father)
{
fa[cur][0] = father; mx[cur][0] = ww[father] - a[cur];
for(int i=1;i<=lg;++i)
{
fa[cur][i] = fa[fa[cur][i-1]][i-1];
mx[cur][i] = max(mx[cur][i-1],mx[fa[cur][i-1]][i-1]); //前半段最大、后半段最大
}
for(int i=h[cur];~i;i=ne[i])
{
int j = e[i];
// if(j==father) continue;
dfs(j,cur);
}
}
void solve()
{
memset(h,-1,sizeof(h));
scanf("%d%d%d",&n,&m,&T);
cnt = n; lg = log2(n)+1;
for(int i=1;i<=n;++i) scanf("%d",&a[i]),f[i] = i;
for(int i=n+1;i<2*n;++i) f[i] = i;
for(int i=0;i<m;++i)
{
int x,y,z; scanf("%d%d%d",&x,&y,&z);
q[i] = {x,y,z};
}
sort(q,q+m);
Kru();
dfs(cnt,0); //处理倍增父节点
// dfs2(cnt,0); //处理每个节点的限制
while(T--)
{
int x,k; scanf("%d%d",&x,&k);
for(int i=lg;i>=0;--i) if(k>=mx[x][i]&&fa[x][i]!=0) x = fa[x][i];
printf("%d\n",k+a[x]); //获得的social ability
}
}
signed main(void)
{
solve();
return 0;
}