F. Make It Connected 解析(思維、MST)

Codeforce 1095 F. Make It Connected 解析(思維、MST)

今天我們來看看CF1095F
題目連結

題目
給你(n)個點，每個點(u)還有一個值(a[u])，還有給你(m)條可能的邊。
任兩點((u,v))都可能可以用(a[u]+a[v])這個權值來連接。
一開始圖上沒有邊，求最小的權值使得圖連通。

前言

思考方向還是不夠正確阿...

想法

首先知道我們想要構造MST，並且注意到:如果不考慮多出來的(m)條邊，那麼從權值最小的點連到所有其他點就是一種最好的構造。
那麼我們只要把這(n-1)條邊納入考慮一起造MST就好。

程式碼:

const int _n=2e5+10;
int parent[_n], rank[_n];
inline void dsinit(int n) {for (int i = 0; i < n; i++)parent[i] = i;memset(rank, 0, sizeof rank);}
inline int dsfind(int e) {return parent[e] == e ? e : parent[e] = dsfind(parent[e]);}
inline void dsunion(int s1, int s2) {if (rank[s1] < rank[s2])swap(s1, s2);parent[s2] = s1;if (rank[s1] == rank[s2]) rank[s1]++;}
//以上是dsu模板
struct E{
  int f,t;ll w;
  bool operator<(const E& rhs)const{return w<rhs.w;}
};
int t,n,m,s1,s2,id=0,x,y;
ll a[_n],sum=0,w;
vector<E> e;
main(void) {ios_base::sync_with_stdio(0);cin.tie(0);
  cin>>n>>m;rep(i,0,n)cin>>a[i]; rep(i,0,m){
    cin>>x>>y>>w;x--,y--;e.pb({x,y,w});
  }id=0;rep(i,1,n)if(a[i]<a[id])id=i;
  rep(i,m,m+n)if(i-m!=id)e.pb({id,i-m,a[id]+a[i-m]});
  sort(all(e)); dsinit(n);
  sum=0;rep(i,0,m+n-1){
    s1=dsfind(e[i].f),s2=dsfind(e[i].t);
    if(s1!=s2)dsunion(s1,s2),sum+=e[i].w;
  }cout<<sum<<'
';
  return 0;
}

標頭、模板請點Submission看
Submission