1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
| #include<bits/stdc++.h> using namespace std; #define int long long inline int read(){int x=0,f=1;char ch=getchar(); while(ch<'0'|| ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){ x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;} #define Z(x) (x)*(x) #define pb push_back
#define N 300010
struct node { int x, y, w; }; int n, m, i, j, k, T; int x[N], f[N], mn[N][22], mx[N][22], p[N], ans; int Log2[N], u, v, g[N], E[N]; vector<node>G;
int fa(int x) { if(f[x]==x) return x; return f[x]=fa(f[x]); }
int Mx(int l, int r) { int k = Log2[r-l+1]; if(x[mx[l][k]]>x[mx[r-(1<<k)+1][k]]) return mx[l][k]; return mx[r-(1<<k)+1][k]; }
int Mn(int l, int r) { int k = Log2[r-l+1]; if(x[mn[l][k]]<x[mn[r-(1<<k)+1][k]]) return mn[l][k]; return mn[r-(1<<k)+1][k]; }
signed main() { freopen("mst.in", "r", stdin); freopen("mst.out", "w", stdout);
n=read(); for(i=1; i<=n; ++i) x[i]=read(), f[i]=i, mx[i][0]=mn[i][0]=i; for(i=2; i<=n; ++i) Log2[i]=Log2[i>>1]+1; for(k=1; k<=20; ++k) for(i=1, j=(1<<k-1)+1; i+(1<<k)-1<=n; ++i, ++j) { if(x[mx[i][k-1]]>x[mx[j][k-1]]) mx[i][k]=mx[i][k-1]; else mx[i][k]=mx[j][k-1]; if(x[mn[i][k-1]]<x[mn[j][k-1]]) mn[i][k]=mn[i][k-1]; else mn[i][k]=mn[j][k-1]; } x[0]=-1e9; x[n+1]=1e9; while(1) { G.clear(); for(i=1; i<=n; ++i) p[i]=g[i]=E[i]=0; for(i=1; i<=n; ++i) { j=p[fa(i)]; if(j+1<=i-1) g[i]=Mx(j+1, i-1); if(x[g[j]]>x[g[i]]) g[i]=g[j]; if(!g[i]) continue;
G.pb({fa(i), fa(g[i]), x[i]-x[g[i]]}); p[fa(i)]=i; } for(i=1; i<=n; ++i) p[i]=g[i]=n+1; g[n+1]=n+1; for(i=n; i>=1; --i) { j=p[fa(i)]; if(i+1<=j-1) g[i]=Mn(i+1, j-1);
if(x[g[j]]<x[g[i]]) g[i]=g[j];
if(g[i]==n+1) continue;
G.pb({fa(i), fa(g[i]), x[g[i]]-x[i]}); p[fa(i)]=i; } sort(G.begin(), G.end(), [] (node x, node y) { return x.w<y.w; }); for(auto t : G) {
u=t.x; v=t.y;
if(E[fa(u)]) continue; if(fa(u)==fa(v)) continue; ans+=t.w; f[fa(u)]=fa(v); E[fa(u)]=1;
} for(i=2; i<=n; ++i) if(fa(i)!=fa(i-1)) break; if(i>n) break;
} printf("%lld", ans); return 0; }
|