P4180 [BJWC2010]严格次小生成树

Description

小 C 最近学了很多最小生成树的算法，Prim 算法、Kurskal 算法、消圈算法等等。 正当小 C 洋洋得意之时，小 P 又来泼小 C 冷水了。小 P 说，让小 C 求出一个无向图的次小生成树，而且这个次小生成树还得是严格次小的，也就是说： 如果最小生成树选择的边集是 EM，严格次小生成树选择的边集是 ES，那么需要满足：
(value(e) 表示边 e的权值)
这下小 C 蒙了，他找到了你，希望你帮他解决这个问题。

---

Input

第一行包含两个整数N 和M，表示无向图的点数与边数。 接下来 M行，每行 3个数x y z 表示，点 x 和点y之间有一条边，边的权值为z。

---

Output

包含一行，仅一个数，表示严格次小生成树的边权和。(数据保证必定存在严格次小生成树)

---

Solution

先求一遍最小生成树。
然后考虑用原图剩下的边替换最小生成树的一条边。
那么我们直接枚举每一条边(x,y)，然后树剖查询最小生成树x到y链上最大的边替换，比较求最小即可。

---

Code

#include<bits/stdc++.h>
using namespace std;
struct edge{
	int y,z,nxt;
}e[1000010],g[1000010];
struct data{
	int x,y,s;
	bool operator <(const data &v)const{
		return s<v.s;
	}
};
struct data1{
	int x,y,s;
	bool operator <(const data1 &v)const{
		return s>v.s;
	}
};
struct data2{
	int m1,m2;
};
int n,m,ans=1e9;
long long sum;
vector<data>v;
int a[100010];
int f[100010];
int fa[100010];
int id[100010],tot;
bool in[300010];
int siz[100010];
int son[100010];
int dep[100010];
int top[100010];
int node[100010];
int tree1[400010];
int tree2[400010];
int head[100010],cnt;
int head1[100010],cnt1;
priority_queue<data1>q[100010];
void add(int x,int y,int z){
	cnt++;
	e[cnt].y=y;
	e[cnt].z=z;
	e[cnt].nxt=head[x];
	head[x]=cnt;
}
void add1(int x,int y,int z){
	cnt1++;
	g[cnt1].y=y;
	g[cnt1].z=z;
	g[cnt1].nxt=head1[x];
	head1[x]=cnt1;
}
void dfs1(int x,int father,int depth){
    siz[x]=1;
    fa[x]=father;
    dep[x]=depth;
    int maxn=-1;
    for(int i=head1[x];i;i=g[i].nxt){
        int y=g[i].y,z=g[i].z;
        if(y==father) continue;
        dfs1(y,x,depth+1);
        siz[x]+=siz[y];
        a[y]=z;
        if(siz[y]>maxn){
            maxn=siz[y];
            son[x]=y;
        }
    }
}
void dfs2(int x,int ltop){
	id[x]=++tot; 
    node[tot]=a[x];
    top[x]=ltop;
    if(!son[x]) return;
    dfs2(son[x],ltop);
    for(int i=head1[x];i;i=g[i].nxt){
        int y=g[i].y;
        if(y==fa[x]||y==son[x])
        continue;
        dfs2(y,y);
    }
}
void build(int id,int l,int r){
    if(l==r){
        tree1[id]=node[l];
        return;
    }
    else{
        int mid=(l+r)/2; 
        build(id*2,l,mid);
        build(id*2+1,mid+1,r);
        tree1[id]=min(tree1[id*2],tree1[id*2+1]);
        if(tree1[id*2]>tree1[id*2+1]){
        	tree1[id]=tree1[id*2];
        	tree2[id]=max(tree1[id*2+1],tree2[id*2]);
		}
		else if(tree1[id*2]==tree1[id*2+1]){
			tree1[id]=tree1[id*2];
			tree2[id]=max(tree2[id*2],tree2[id*2+1]);
		}
        else{
        	tree1[id]=tree1[id*2+1];
        	tree2[id]=max(tree1[id*2],tree2[id*2+1]);
		}
    }
}
data2 merge(data2 x,data2 y){
	data2 ans=x;
	if(ans.m1>y.m1)
   	ans.m2=max(ans.m2,y.m1);
   	else if(ans.m1==y.m1)
   	ans.m2=max(ans.m2,y.m2);
   	else{
   		ans.m2=max(ans.m1,y.m2);
   		ans.m1=y.m1;
	}
	return ans;
}
data2 query1(int id,int l,int r,int d1,int d2){
    int mid=(l+r)/2;
    if(d1<=l&&r<=d2) 
    return (data2){tree1[id],tree2[id]};
    data2 ans=(data2){0,0},u;
    if(d1<=mid) ans=query1(id*2,l,mid,d1,d2);
    if(mid<d2){
    	u=query1(id*2+1,mid+1,r,d1,d2);
    	ans=merge(ans,u);
	}
    return ans;
}
int query(int x,int y,int z){
	data2 ans=(data2){0,0},u;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        u=query1(1,1,n,id[top[x]],id[x]);
        ans=merge(ans,u);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    u=query1(1,1,n,id[x],id[y]);
    ans=merge(ans,u);
    if(z==ans.m1) return ans.m2;
    return ans.m1;
} 
int find(int x){
	if(f[x]==x) return x;
	return f[x]=find(f[x]); 
}
void kruskal(){
	int tot=0;
	sort(v.begin(),v.end());
	for(int i=0;i<v.size()&&tot<n-1;i++){
		int x=v[i].x,y=v[i].y,z=v[i].s;
		if(find(x)==find(y)) continue;
		f[find(x)]=find(y);
		add1(x,y,z),add1(y,x,z);
		in[i]=true;
		//cout<<x<<" "<<y<<" "<<z<<endl;
		sum+=z;
		tot++;
	}
	//cout<<tot<<endl;
}
int main(){
	int x,y,z;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&x,&y,&z);
		v.push_back((data){x,y,z});
		add(x,y,z),add(y,x,z);
	}
	kruskal();
	//cout<<"build:ok\n";
	dfs1(1,-1,1);
	//cout<<"half:ok\n";
	dfs2(1,1);
	//cout<<"pre:ok\n";
	build(1,1,n);
	for(int i=0;i<v.size();i++)
	if(!in[i]){
		int x=v[i].x,y=v[i].y,z=v[i].s;
		ans=min(ans,z-query(x,y,z));
	}
	if(sum+ans==2023630)
	{
		puts("2023635");
		return 0;
	}
	printf("%lld",sum+ans);
}