兽径管理

通过了一道之前怎么做都RE的题

题目(又是USA Cow Orgnization):

约翰农场的牛群希望能够在 N 个(1<=N<=200) 草地之间任意移动。草地的编号由 1到 N。草地之间有树林隔开。牛群希望能够选择草地间的路径,使牛群能够从任一 片草地移动到任一片其它草地。 牛群可在路径上双向通行。

牛群并不能创造路径,但是他们会保有及利用已经发现的野兽所走出来的路径(以 下简称兽径)。每星期他们会选择并管理一些或全部已知的兽径当作通路。

牛群每星期初会发现一条新的兽径。他们接着必须决定管理哪些兽径来组成该周牛 群移动的通路,使得牛群得以从任一草地移动到任一草地。牛群只能使用当周有被 管理的兽径做为通路。

牛群希望他们管理的兽径长度和为最小。牛群可以从所有他们知道的所有兽径中挑 选出一些来管理。牛群可以挑选的兽径与它之前是否曾被管理无关。

兽径决不会是直线,因此连接两片草地之间的不同兽径长度可以不同。 此外虽然 两条兽径或许会相交,但牛群非常的专注,除非交点是在草地内,否则不会在交点 换到另外一条兽径上。

在每周开始的时候,牛群会描述他们新发现的兽径。如果可能的话,请找出可从任 何一草地通达另一草地的一组需管理的兽径,使其兽径长度和最小。

将题目翻译一下,就是每次添加一条边,每次求下最小生成树。

因为个人比较喜欢Kruskal,所以先要对边进行排序,然而每添加一条边就要排序一次,实在费功夫... 所以我们要找一种更为高效的排序方法

比如堆排

大体思路就是用堆排代替快排(sort),每次取堆首,如果采用,记录下当前使用的边的编号,完成最小生成树建立后再依次入栈,代码如下:

#include<iostream>
#include<cstdio>
#include<queue>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;
int fa[205];
int find(int x){
    if(fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}
void unity(int x,int y){
    x=find(x);
    y=find(y);
    fa[y]=fa[x];
}
struct Ed{
    int from,to,dis,num;
}ed[6005];
int ed_num;
void add(int from,int to,int dis){
    ed_num++;
    ed[ed_num].dis=dis;
    ed[ed_num].from=from;
    ed[ed_num].to=to;
    ed[ed_num].num=ed_num;
}
int used[6005];
int n,w;
priority_queue<Ed,vector<Ed>,greater<Ed> > q;
bool operator<(const Ed &a,const Ed &b){
    return a.dis<b.dis;
}
bool operator>(const Ed &a,const Ed &b){
    return a.dis>b.dis;
}
int t;
int main(){
    scanf("%d%d",&n,&w);
    for(int i=1;i<=w;i++){
        for(int j=1;j<=n;j++) fa[j]=j;
        t=0;
        int tot=0;
        int sum=0;
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        q.push(ed[ed_num]);
        while(!q.empty()){
            Ed u=q.top();
            q.pop();
            if(find(u.from)!=find(u.to)){
                unity(u.from,u.to);
                used[++t]=u.num;
                tot++;
                sum+=u.dis;
            }
            if(tot==n-1)break;
        }
        if(tot!=n-1)
            printf("-1
");
        else
            printf("%d
",sum);
        sum=0;
        for(int j=1;j<=t;j++)
            q.push(ed[used[j]]);
        memset(used,0,sizeof(used));
    }
    return 0;
}

但代码中并没有对某些已经进行pop处理的边进行重入堆操作,原因如下

如果将每条曾经pop过的边再次入堆,十分浪费空间,如果只入堆本次使用过的边,也是可以的

因为既然某些边被弹出但没有用到,说明其起点终点已经被其他更短的边合并,也就是说,这条边并不是能够连通几个点的最优方案,并“不够优秀”,所以以后并不会再次用到这条边,不用再次入堆(不加不应该MLE吗,怎么全RE了...)

原文地址:https://www.cnblogs.com/648-233/p/10926158.html