// rmq
void dfs(int u,int f,int d) {
int tmp=++cnt;
E[tmp]=u; L[++tot]=tmp; H[u]=tot; dis[u]=d;
for(int i=0;i<vec[u].size();i++) {
int v=vec[u][i].f;
if(v!=f) {
dfs(v,u,d+vec[u][i].s);
L[++tot]=tmp;
}
}
}
void init() {
for(int i=1;i<=tot;i++) dp[0][i]=L[i];
int k=(int) (log(tot*1.0+1)/log(2.0));
for(int i=1;i<=k;i++)
for(int j=1;j+(1<<i)-1<=tot;j++)
dp[i][j]=min(dp[i-1][j],dp[i-1][j+(1<<(i-1))]);
}
int rmq(int x,int y) {
if(x>y) swap(x,y);
int k=(int)(log(y-x+1.0)/log(2.0));
return min(dp[k][x],dp[k][y-(1<<k)+1]);
}
int lca(int x,int y) {
return E[rmq(H[x],H[y])];
}
// 倍增
void dfs(int x,int fa,int depth,int cost){
dep[x] = depth;
p[x][0] = fa;
dis[x] = cost;
rep1(i,LOG)
p[x][i] = p[ p[x][i-1] ][i-1]; // 倍增,可以画个树理解一下,存的是节点x第2^i个祖先
for(int i=po[x];i;i=edge[i].next){
int y = edge[i].y;
if(y==fa)continue;
dfs(y,x,depth+1,cost+edge[i].val);
}
}
int lca(int x,int y){ // 倍增求lca
if(dep[x]>dep[y])swap(x,y);
if(dep[x]<dep[y]){
int del = dep[y]-dep[x];
rep(i,LOG)
if( del>>i & 1 )
y = p[y][i];
}
if(x!=y){
for(int i=LOG-1;i>=0;i--)
if( p[x][i]!=p[y][i] ){
x = p[x][i];
y = p[y][i];
}
x = p[x][0];
y = p[y][0];
}
return x;
}
LCA
最新推荐文章于 2024-09-30 12:49:19 发布