1493 次小生成树

1493 次小生成树

练完一本通的题目后把洛谷的生成树的题目再整理一下，之前做过的生成树的洛谷题，把部分都是纯模板，没有什么想写的吧
这个题就是模板的次小生成树，这个知识点来自一次省队选拔
虽然如此，但是大纲里有

准确来说，次小生成树又叫做严格次小生成树
次小生成树的定义为：边权和严格大于最小生成树的边权和最小的生成树
也就是边权和第二小的生成树
次小生成树，是个菜鸡都知道，和最小生成树肯定有关系
所以，我们的思路就是，枚举每一条没有加入最小生成树的边，加入最小生成树，同时在最小生成树中删除边权最大的
当然这是错的
我们要记录的是严格小于最大值的最大值，虽然很别扭，但是很好理解
发生冲突的时候我们需要删除严格小于最大值的最大值
那么我们如何进行查询？用倍增在树上进行查询即可

所以整个算法的流程就是
1.维护出最小生成树
2.构造出倍增数组
3.然后进行树上倍增查询
时间复杂度：写完代码再说

#include<bits/stdc++.h>
#define Pair pair<int, int> 
#define MP(x, y) make_pair(x, y)
#define fi first
#define se second 
#define LL long long
using namespace std;
const int MAXN = 3e5 + 10;
const LL INF = 1e15 + 10;
struct Edge 
{
    int u, v, w;
    bool operator < (const Edge &rhs) const 
	{
    	return w < rhs.w;
    }
} E[MAXN];
vector<Pair> v[MAXN];
int n,m;
int fa[MAXN], vis[MAXN], dep[MAXN], f[MAXN][19], mx[MAXN][19], me[MAXN][19];;
LL sum;
int x,y,z;
int find(int x)//小蝌蚪找爸爸函数 
{
    return fa[x] == x ? fa[x] : fa[x] = find(fa[x]);
}
void Kruskal()//维护最小生成树函数 
{
    sort(E + 1, E + m + 1);
    int tot = 0;
    for (int i = 1; i <= m; i++) 
	{
        int x = E[i].u, y = E[i].v, fx = find(x), fy = find(y);
        if (fx != fy)
		 {
            fa[fx] = fy; 
            tot++, sum += E[i].w, vis[i] = 1;
            v[x].push_back(MP(y, E[i].w)); 
            v[y].push_back(MP(x, E[i].w));
        }
        if (tot == n - 1) break;
    }
}
void dfs(int x, int fa)//搜索，维护最大值次大值 
 {
    dep[x] = dep[fa] + 1; f[x][0] = fa; 
    for(int i = 0, to; i < v[x].size(); i++) 
	{
    	if((to = v[x][i].fi) == fa) continue;
    	mx[to][0] = v[x][i].se;
    	dfs(to, x);
    }
}
void pre() 
{
    for (int i = 1; i <= 18; i++) 
	{
        for (int j = 1; j <= n; j++)
		 {
            f[j][i] = f[ f[j][i - 1] ][i - 1];
            int topf = f[j][i - 1];
            mx[j][i] = max(mx[j][i - 1], mx[topf][i - 1]);
            me[j][i] = max(me[j][i - 1], me[topf][i - 1]);
            if (mx[j][i - 1] > mx[topf][i - 1]) me[j][i] = max(me[j][i], mx[topf][i - 1]);
            else if (mx[j][i - 1] < mx[topf][i - 1]) me[j][i] = max(me[j][i], mx[j][i - 1]);
        }
    }
}
int LCA(int x, int y) 
{
    if (dep[x] < dep[y]) swap(x, y);
    for (int i = 18; i >= 0; i--) if (dep[ f[x][i] ] >= dep[y] ) x = f[x][i];
    if (x == y) return x;
    for (int i = 18; i >= 0; i--) if (f[x][i] != f[y][i]) x = f[x][i], y = f[y][i];
    return f[x][0];
}
int findmax(int x, int lca, int val)//比较最大值 
{
    LL ans = 0;
    for (int i = 18; i >= 0; i--) 
	{
        if (dep[ f[x][i] ] >= dep[lca]) 
		{
            if (mx[x][i] == val) ans = max(ans, (LL) me[x][i]);
            else ans = max(ans, (LL) mx[x][i]);
            x = f[x][i];
        }
    }
    return ans;
}
void work() 
{
    LL ans = INF;
    for (int i = 1; i <= m; i++)
	{ 
        if (vis[i]) continue;
        int x = E[i].u, y = E[i].v, z = E[i].w;
        int lca = LCA(x, y);
        int lmx = findmax(x, lca, z), rmx = findmax(y, lca, z);
        if (max(lmx, rmx) != z)  ans = min(ans, sum + z - max(lmx, rmx));
    }
    printf("%lld", ans);
}
int main() {
    cin>>n>>m;
    for (int i = 1; i <= n; i++) fa[i] = i;
    for (int i = 1; i <= m; i++) 
	{
		cin>>x>>y>>z;
        E[i] = (Edge) {x, y, z};
    }
    Kruskal();
    dfs(1, 0);
    pre();
    work();
    return 0;
}