rt,模板题
思路挺简单,每次找到不在最小生成树的一条边,断开from>tofrom>tofrom>to在树链上不等于该边边权的最大边
更新答案就好了,这里主要是注意一定要不等于该边边权,因为严格次小
是发方法还是倍增,我们的倍增数组维护最大值和次小值就好了
max2[j][i]=max(max2[fa[j][i-1]][i-1],max2[j][i-1]);
if(max1[j][i-1]>max1[fa[j][i-1]][i-1])max2[j][i]=max(max2[j][i],max1[fa[j][i-1]][i-1]);
else if(max1[j][i-1]<max1[fa[j][i-1]][i-1])max2[j][i]=max(max2[j][i],max1[j][i-1]);
更新的是两段中的较小段,相等不更新,保证次小
其他的话,我们主要用了一下算法
倍增+LCA
Kruskal+并查集
这里的倍增预处理我们学习了大佬的方法,分开写,更加便于调试虽然我是一遍过的
PS.数组能开多大开多大,30w条双向边我数组开了180w才过
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
const ll maxn = 300007;
const ll N = 18;
const ll INF = 21474836470000000;
struct node{
ll to,next,w;
}edge[maxn*6];
ll cnt,head[maxn],fa[maxn][N+2],dep[maxn],del[maxn];
void add(ll from,ll to,ll w){
edge[++cnt].to=to;
edge[cnt].next=head[from];
head[from]=cnt;
edge[cnt].w=w;
}
pair<ll, pair<ll,ll> >g[maxn*6];
ll n,m,f[maxn],max1[maxn][N+2],max2[maxn][N+2];
void dfs(ll u,ll fat){
for(ll i=head[u];i;i=edge[i].next){
ll to=edge[i].to;
if(to==fat)continue;
fa[to][0]=u;
max1[to][0]=edge[i].w;
max2[to][0]=-INF;
dep[to]=dep[u]+1ll;
dfs(to,u);
}
}
void prework(){
for(ll i=1;i<=N;i++){
for(ll j=1;j<=n;j++){
fa[j][i]=fa[fa[j][i-1]][i-1];
max1[j][i]=max(max1[fa[j][i-1]][i-1],max1[j][i-1]);
max2[j][i]=max(max2[fa[j][i-1]][i-1],max2[j][i-1]);
if(max1[j][i-1]>max1[fa[j][i-1]][i-1])max2[j][i]=max(max2[j][i],max1[fa[j][i-1]][i-1]);
else if(max1[j][i-1]<max1[fa[j][i-1]][i-1])max2[j][i]=max(max2[j][i],max1[j][i-1]);
}
}
}
ll find(ll x){ return f[x]==x?x:f[x]=find(f[x]); }
ll Kruskal()
{
sort(g+1,g+1+m);
ll sum=0;
for(ll i=1;i<=m;i++)
{
ll f1=find(g[i].second.first);
ll f2=find(g[i].second.second);
if(f1!=f2)
{
del[i]=1;
f[f1]=f2;
sum+=g[i].first;
add(g[i].second.first,g[i].second.second,g[i].first);
add(g[i].second.second,g[i].second.first,g[i].first);
}
}
return sum;
}
ll lca(ll a,ll b)
{
if(dep[a]>dep[b])swap(a,b);
for(ll i=N;i>=0;i--)
if(dep[fa[b][i]]>=dep[a])
b=fa[b][i];
if(a==b)return a;
for(ll i=N;i>=0;i--)
if(fa[a][i]^fa[b][i])
a=fa[a][i],
b=fa[b][i];
return fa[a][0];
}
ll split(ll x,ll y,ll vv){
ll ret=-INF;
if(dep[x]<dep[y])swap(x,y);
for(ll i=N;i>=0;i--){
if(dep[fa[x][i]]>=dep[y]){
if(max1[x][i]!=vv)ret=max(ret,max1[x][i]);
else ret=max(ret,max2[x][i]);
x=fa[x][i];
}
}
if(x==y)return ret;
for(ll i=N;i>=0;i--){
if(fa[x][i]^fa[y][i]){
if(max1[x][i]!=vv)ret=max(ret,max1[x][i]);
else ret=max(ret,max2[x][i]);
if(max1[y][i]!=vv)ret=max(ret,max1[y][i]);
else ret=max(ret,max2[y][i]);
x=fa[x][i];
y=fa[y][i];
}
}
if(max1[x][0]!=vv)ret=max(ret,max1[x][0]);
else ret=max(ret,max2[x][0]);
if(max1[y][0]!=vv)ret=max(ret,max1[y][0]);
else ret=max(ret,max2[y][0]);
return ret;
}
int main(){
scanf("%lld%lld",&n,&m);
for(ll i=1;i<=n;i++)f[i]=i;
for(ll i=1;i<=m;i++){
ll x,y,z;
scanf("%lld%lld%lld",&x,&y,&z);
if(x>y)swap(x,y);
g[i].first=z;
g[i].second.first=x;
g[i].second.second=y;
}
ll Val=Kruskal();
max2[1][0]=-INF;
dep[1]=1;
dfs(1,0);
prework();
ll Ans=INF;
for(ll i=1;i<=m;i++){
if(del[i])continue;
ll from=g[i].second.first;
ll to=g[i].second.second;
ll w=g[i].first;
ll Instead=split(from,to,w);
Ans=min(Ans,Val+w-Instead);
}
cout<<Ans;
}