这样阴沉的天气持续下去,我们不免担心起他的健康。
51nod魔法学校近日开展了主题为“天气晴朗”的魔法交流活动。
N名魔法师按阵法站好,之后选取N - 1条魔法链将所有魔法师的魔力连接起来,形成一个魔法阵。
魔法链是做法成功与否的关键。每一条魔法链都有一个魔力值V,魔法最终的效果取决于阵中所有魔法链的魔力值的和。
由于逆天改命的魔法过于暴力,所以我们要求阵中的魔法链的魔力值最大值尽可能的小,与此同时,魔力值之和要尽可能的大。
现在给定魔法师人数N,魔法链数目M。求此魔法阵的最大效果。
Input
两个正整数N,M。(1 <= N <= 10^5, N <= M <= 2 * 10^5)接下来M行,每一行有三个整数A, B, V。(1 <= A, B <= N, INT_MIN <= V <= INT_MAX)保证输入数据合法。
Output
输出一个正整数R,表示符合条件的魔法阵的魔力值之和。
Input示例
4 6 1 2 3 1 3 1 1 4 7 2 3 4 2 4 5 3 4 6
Output示例
12
题目的意思就是要求生成树中所有边的权值中,最大值最小,也就是说最大边权尽量小,组成一个生成树;
第二步就是求这个生成树的最大权值和,也就是最大生成树;
那先来说第一步,其实就是按照kruskal算法,将边权按照从小到大排序,然后从小到大枚举,看到哪个权值能够组成一棵树,其实也就是覆盖了所有的点;然后记录这个权值w_max,再次调用kruskal,这次从大到小排序,然后所有大于w_max的都过滤,只使用比这个小的边,然后组成的最大生成树,就是结果;
写了一下午 tle 不知道怎么回事 借鉴大牛的代码 思路基本一样 但还是 超时!!!操 贴下tle 代码 日后改把
#include<iostream>
#include<cstring>
#include<algorithm>
#include<stdio.h>
using namespace std;
#define maxn 10000+10
#define inf 9999999
#define mem(a,b) memset(a,b,sizeof(a))
struct road{int x,y,z;
}a[maxn];
int dis[maxn];
bool cmp(struct road q,struct road w){return q.z<w.z;
}
int find1(int x){if(x!=dis[x])dis[x]=find1(dis[x]);return dis[x];
}int mix(int x,int y){ //并查集int x1;int y1;x1=find1(x);y1=find1(y);if(x1!=y1){dis[y1]=x1;return 1;}return 0;
}
int main(){int n,m;cin>>m>>n;for(int j=1;j<=n;j++){scanf("%d%d%d",&a[j].x,&a[j].y,&a[j].z);}int sum=0,num=0;sort(a+1,a+n+1,cmp);int l=0;mem(dis,0);for(int j=1;j<=n;j++){if(l==m){l=j;break;}if(!dis[a[j].x]){dis[a[j].x]=1;l++;}if(!dis[a[j].y]){dis[a[j].y]=1;l++;}}while (l < n){if (a[l].z == a[l - 1].z){l ++;}elsebreak;}for(int j=1;j<=m;j++){dis[j]=j;}sum=0,num=0;for(int j=l-1;j>=1;j--){if(mix(a[j].x,a[j].y)){sum++;num+=a[j].z;}if(sum==m-1)break;}cout<<num<<endl;return 0;
}
大牛代码
#include<iostream>
#include<set>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
#define LL long long
typedef struct NODE
{int a,b;int v;
}Node;
Node nodes[200005];
bool cmp(const Node &s, const Node &t)
{return s.v < t.v;
}
int father[100005];
int rank[100005];
int hash[100005];
int search_x(int x)
{//return x == father[x] ? x : search_x(father[x]);if (x != father[x])father[x] = search_x(father[x]);return father[x];
}
bool union_xy(int x, int y) //我觉得 都是一样的路径压缩啊 不知道是不是这里出了问题
{int root1 = search_x(x);int root2 = search_x(y);if (root1 == root2)return false;else if (rank[root1] > rank[root2]) //下面都看不懂 好气! 自己好差劲的说!{father[root2] = root1;}else{father[root1] = root2;if (rank[root1] == rank[root2]){rank[root2]++;}}return true;
}
int main()
{int n,m;//cin >> n >> m;scanf("%d %d", &n, &m);for (int i = 0; i < m; ++ i){//cin >> nodes[i].a >> nodes[i].b >> nodes[i].v;scanf("%d %d %d", &nodes[i].a, &nodes[i].b, &nodes[i].v);}sort(nodes, nodes+m, cmp);set<int> st;int pos = 0;int sum = 0;memset (hash, 0, sizeof(hash));for (int i = 0; i < m; ++ i){if (sum == n){pos = i;break;}if (! hash[nodes[i].a]){sum ++;hash[nodes[i].a] = 1;}if (! hash[nodes[i].b]){sum ++;hash[nodes[i].b] = 1;}}//注意我这里是找到最大权值的上边界,这样我就可以直接从pos-1的位置求最大生成树了//不这样也行,那么在调用kruskal算法时,大于这个最大权值的过滤掉也是一样的while (pos < m){if (nodes[pos].v == nodes[pos - 1].v){pos ++;}elsebreak;}for (int i = 1; i <= n; ++ i){father[i] = i;rank[i] = 1;}LL weight = 0;int num = 0;for (int i = pos - 1; i >= 0; -- i){if (union_xy(nodes[i].a, nodes[i].b)){weight += nodes[i].v;num ++;}if (num == n - 1){//cout << weight <<endl;printf("%lld\n", weight);break;}}return 0;
}