洛谷 1195 口袋的天空
并查集水题。
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 | //Twenty #include<algorithm> #include<iostream> #include<cstdlib> #include<cstring> #include<cstdio> #include<cmath> #include<ctime> #include<queue> const int maxn=1e5+5; using namespace std; typedef long long LL; int ans,n,m,k,u,v,w,cnt,fa[maxn]; void read( int &ret) { int f=1; ret=0; char ch=getchar(); while (ch!= '-' &&(ch< '0' ||ch> '9' )) ch=getchar(); if (ch== '-' ) f=-1,ch=getchar(); for (;ch>= '0' &&ch<= '9' ;ch=getchar()) ret=ret*10+ch- '0' ; ret*=f; } struct edge{ int u,v,w; edge(){} edge( int u, int v, int w):u(u),v(v),w(w){} friend bool operator <( const edge&A, const edge&B) { return A.w<B.w; } }e[maxn]; int find( int x) { return x==fa[x]?x:fa[x]=find(fa[x]); } void init() { read(n); read(m); read(k); for ( int i=1;i<=m;i++) { read(u); read(v); read(w); e[++cnt]=edge(u,v,w); } sort(e+1,e+cnt+1); for ( int i=1;i<=n;i++) fa[i]=i; for ( int i=1;i<=cnt;i++) { u=e[i].u; v=e[i].v; w=e[i].w; int x=find(u),y=find(v); if (x!=y) { ans+=w; fa[x]=y; n--; if (n==k) break ; } } printf( "%d\n" ,ans); } int main() { init(); return 0; } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步