zoukankan      html  css  js  c++  java
  • bzoj3784 树上的路径

    传送门

    #include <bits/stdc++.h>
    const int N=500005;
    int to[N<<1],edge,Next[N<<1],last[N],siz[N],g,L[N*20],R[N*20],w[N<<1];
    int v[N*20],vis[N],cnt,st[N*20][21],lstl,lstr,bit[N*20],n,m,x,y,z;
    void add(int x,int y,int z){
    	to[++edge]=y;
    	Next[edge]=last[x];
    	last[x]=edge;
    	w[edge]=z;
    }
    void find(int x,int y,int &g,int nn){
    	siz[x]=1;
    	int mx=0;
    	for (int i=last[x];i;i=Next[i]){
    		int u=to[i];
    		if (u==y || vis[u]) continue;
    		find(u,x,g,nn);
    		siz[x]+=siz[u];
    		mx=std::max(mx,siz[u]);
    	}
    	if (mx<=nn/2 && nn-siz[x]<=nn/2) g=x;
    } 
    void dfs(int x,int y,int dis){
    	v[++cnt]=dis,L[cnt]=lstl,R[cnt]=lstr;
    	for (int i=last[x];i;i=Next[i]){
    		int u=to[i];
    		if (u==y || vis[u]) continue;
    		dfs(u,x,dis+w[i]);
    	}
    }
    void div(int x){
    	vis[x]=1;
    	v[++cnt]=0,lstl=cnt;
    	for (int i=last[x];i;i=Next[i]){
    		int u=to[i];
    		if (vis[u]) continue;
    		lstr=cnt;
    		dfs(u,x,w[i]);
    		siz[u]=cnt-lstr;
    	}
    	for (int i=last[x];i;i=Next[i]){
    		int u=to[i];
    		if (vis[u]) continue;
    		find(u,0,g,siz[u]);
    		div(g);
    	}
    }
    void init(){
    	for (int i=2;i<=cnt;i++) bit[i]=bit[i>>1]+1;
    	for (int i=1;i<=cnt;i++) st[i][0]=i;
    	for (int i=1;i<=20;i++)
    		for (int j=1;j+(1<<i)-1<=cnt;j++)
    			if (v[st[j][i-1]]>v[st[j+(1<<i-1)][i-1]]) st[j][i]=st[j][i-1];
    			else st[j][i]=st[j+(1<<i-1)][i-1];
    }
    int RMQ(int l,int r){
    	int len=bit[r-l+1];
    	if (v[st[l][len]]>v[st[r-(1<<len)+1][len]]) return st[l][len];
    	return st[r-(1<<len)+1][len]; 
    }
    struct note{
    	int x,l,r,mx;
    };
    inline bool operator <(note x,note y){
    	return v[x.x]+v[x.mx]<v[y.x]+v[y.mx];
    }
    std::priority_queue<note> q;
    int main(){
    	scanf("%d%d",&n,&m);
    	for (int i=1;i<n;i++){
    		scanf("%d%d%d",&x,&y,&z);
    		add(x,y,z),add(y,x,z);
    	}
    	find(1,0,g,n);
    	div(g);
    	init(); 
    	for (int i=1;i<=cnt;i++)
    		if (L[i]) q.push(note{i,L[i],R[i],RMQ(L[i],R[i])}); 
    	while (m--){
    		note t=q.top();
    		q.pop();
    		printf("%d
    ",v[t.x]+v[t.mx]);
    		if (t.mx>t.l) q.push(note{t.x,t.l,t.mx-1,RMQ(t.l,t.mx-1)});
    		if (t.r>t.mx) q.push(note{t.x,t.mx+1,t.r,RMQ(t.mx+1,t.r)});
    	}
    }
    
    * 生而自由 爱而无畏 *
  • 相关阅读:
    C# 实现 Snowflake算法生成唯一性Id
    kafka可视化客户端工具(Kafka Tool)的基本使用(转)
    docker 安装kafka
    Model类代码生成器
    使用docker 部署rabbitmq 镜像
    Vue 增删改查 demo
    git 提交代码到库
    Android ble蓝牙问题
    mac 配置 ssh 到git (Could not resolve hostname github.com, Failed to connect to github.com port 443 Operation timed out)
    okhttp
  • 原文地址:https://www.cnblogs.com/flyfeather6/p/14381011.html
Copyright © 2011-2022 走看看