codevs 1231 最优布线问题 x(find函数要从娃娃抓起系列)
学校需要将n台计算机连接起来,不同的2台计算机之间的连接费用可能是不同的。为了节省费用,我们考虑采用间接数据传输结束,就是一台计算机可以间接地通过其他计算机实现和另外一台计算机连接。
为了使得任意两台计算机之间都是连通的(不管是直接还是间接的),需要在若干台计算机之间用网线直接连接,现在想使得总的连接费用最省,让你编程计算这个最小的费用。
输入第一行为两个整数n,m(2<=n<=100000,2<=m<=100000),表示计算机总数,和可以互相建立连接的连接个数。接下来m行,每行三个整数a,b,c 表示在机器a和机器b之间建立连接的话费是c。(题目保证一定存在可行的连通方案, 数据中可能存在权值不一样的重边,但是保证没有自环)
输出只有一行一个整数,表示最省的总连接费用。
3 3
1 2 1
1 3 2
2 3 1
2
最终答案需要用long long类型来保存
分类标签 Tags 点此展开
(把find(f[x]改为f[x]=find(f[x])))
这样一个处理!
不然数据越大,越难弄,需要从低向上找很久很久
运行结果
测试点#1.in 结果: 内存使用量: 256kB 时间使用量: 0ms测试点#2.in 结果: 内存使用量: 492kB 时间使用量: 15ms
测试点#3.in 结果: 内存使用量: 620kB 时间使用量: 26ms
测试点#4.in 结果: 内存使用量: 3176kB 时间使用量: 1000ms
测试点#5.in 结果: 内存使用量: 2924kB 时间使用量: 1000ms
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<string> 5 #include<algorithm> 6 #define Maxn 100010 7 8 using namespace std; 9 10 long long n,m,tot=0,f1,f2,k; 11 long long f[Maxn]; 12 bool u[Maxn]; 13 14 struct gx{ 15 long long x; 16 long long y; 17 long long w; 18 bool operator < (const gx &qwq)const 19 { 20 return w<qwq.w; 21 } 22 }gx[Maxn]; 23 24 int find(int x) 25 { 26 //if(f[x]!=x) f[x]=find(f[x]); 27 //return f[x]; 28 return x == f[x] ? x : find(f[x]); 29 } 30 31 int main() 32 { 33 //ios::sync_with_stdio(false); 34 cin>>n>>m; 35 for(int i=1;i<=m;i++) 36 { 37 cin>>gx[i].x>>gx[i].y>>gx[i].w; 38 } 39 for(int i=1;i<=n;i++) f[i]=i; 40 sort(gx+1,gx+1+m); 41 for(int i=1;i<=m;i++) 42 { 43 f1=find(gx[i].x); 44 f2=find(gx[i].y); 45 if(f1!=f2) 46 { 47 f[f2]=f1; 48 ++k; 49 tot+=gx[i].w; 50 } 51 if(k==n-1) break; 52 } 53 cout<<tot<<endl; 54 return 0; 55 }
AC的两个代码:
1)原因仅仅是因为把find函数改了
数据如下:
运行结果
测试点#1.in 结果: 内存使用量: 256kB 时间使用量: 0ms
测试点#2.in 结果: 内存使用量: 488kB 时间使用量: 12ms
测试点#3.in 结果: 内存使用量: 744kB 时间使用量: 25ms
测试点#4.in 结果: 内存使用量: 3052kB 时间使用量: 147ms
测试点#5.in 结果: 内存使用量: 2920kB 时间使用量: 142ms
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<string> 5 #include<algorithm> 6 #define Maxn 100010 7 8 using namespace std; 9 10 long long n,m,tot=0,f1,f2,k; 11 long long f[Maxn]; 12 bool u[Maxn]; 13 14 struct gx{ 15 long long x; 16 long long y; 17 long long w; 18 bool operator < (const gx &qwq)const 19 { 20 return w<qwq.w; 21 } 22 }gx[Maxn]; 23 24 int find(int x) 25 { 26 if(f[x]!=x) f[x]=find(f[x]); 27 return f[x]; 28 //return x == f[x] ? x : find(f[x]); 29 } 30 31 int main() 32 { 33 //ios::sync_with_stdio(false); 34 cin>>n>>m; 35 for(int i=1;i<=m;i++) 36 { 37 cin>>gx[i].x>>gx[i].y>>gx[i].w; 38 } 39 for(int i=1;i<=n;i++) f[i]=i; 40 sort(gx+1,gx+1+m); 41 for(int i=1;i<=m;i++) 42 { 43 f1=find(gx[i].x); 44 f2=find(gx[i].y); 45 if(f1!=f2) 46 { 47 f[f2]=f1; 48 ++k; 49 tot+=gx[i].w; 50 } 51 if(k==n-1) break; 52 } 53 cout<<tot<<endl; 54 return 0; 55 }
2)优化版(把cin,cout优化)
代码:
数据:
运行结果
测试点#1.in 结果: 内存使用量: 256kB 时间使用量: 0ms
测试点#2.in 结果: 内存使用量: 492kB 时间使用量: 4ms
测试点#3.in 结果: 内存使用量: 616kB 时间使用量: 10ms
测试点#4.in 结果: 内存使用量: 3180kB 时间使用量: 60ms
测试点#5.in 结果: 内存使用量: 3048kB 时间使用量: 54ms
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<string> 5 #include<algorithm> 6 #define Maxn 100010 7 8 using namespace std; 9 10 long long n,m,tot=0,f1,f2,k; 11 long long f[Maxn]; 12 bool u[Maxn]; 13 14 struct gx{ 15 long long x; 16 long long y; 17 long long w; 18 bool operator < (const gx &qwq)const 19 { 20 return w<qwq.w; 21 } 22 }gx[Maxn]; 23 24 int find(int x) 25 { 26 if(f[x]!=x) f[x]=find(f[x]); 27 return f[x]; 28 //return x == f[x] ? x : find(f[x]); 29 } 30 31 int main() 32 { 33 ios::sync_with_stdio(false); 34 cin>>n>>m; 35 for(int i=1;i<=m;i++) 36 { 37 cin>>gx[i].x>>gx[i].y>>gx[i].w; 38 } 39 for(int i=1;i<=n;i++) f[i]=i; 40 sort(gx+1,gx+1+m); 41 for(int i=1;i<=m;i++) 42 { 43 f1=find(gx[i].x); 44 f2=find(gx[i].y); 45 if(f1!=f2) 46 { 47 f[f2]=f1; 48 ++k; 49 tot+=gx[i].w; 50 } 51 if(k==n-1) break; 52 } 53 cout<<tot<<endl; 54 return 0; 55 }