补题系列 1 最小布线

北航数据结构Judge网站原题，抄袭被查重后果自负！
5个月之前，由于数据结构期末考试不考图的大题，所以图学得不扎实，这里特地补上！

【问题描述】
北航主要办公科研楼有新主楼、逸夫楼、如心楼、办公楼、图书馆、主楼、一号楼…。北航网络中心计划要给相关建筑物间铺设光缆进行网络连通，请给出用料最少的铺设方案。
编写程序输入一个办公区域分布图及建筑物之间的距离，计算出用料最少的铺设方案（只有一组最优解，不用考虑多组解)。要求采用Prim或Kruskal算法实现。

【输入形式】
办公区域分布图的顶点（即建筑物）按照自然数（0，1，2，n-1）进行编号，从标准输入中首先输入两个正整数，分别表示线路图的顶点的数目和边的数目，然后在接下的行中输入每条边的信息，每条边占一行，具体形式如下：
<n> <e>
<id> <vi> <vj> <weight>
...
即顶点vi和vj之间边的权重是weight，边的编号是id。
【输出形式】
输出铺设光缆的最小用料数，然后另起一行输出需要铺设的边的id，并且输出的id值按照升序输出。
【样例输入】
 6 10
1 0 1 600
2 0 2 100
3 0 3 500
4 1 2 500
5 2 3 500
6 1 4 300
7 2 4 600
8 2 5 400
9 3 5 200
10 4 5 600
【样例输出】
1500
2 4 6 8 9


#include<iostream>
#include<algorithm>
#define MAX 300
using namespace std;
struct routes{
    int num;
    int a,b;
    int len;
};
routes road[MAX];
int pre[MAX];
int ans[MAX];
int allusenum=0;
int find(int root)
{
    int son=root;
    int temp;
    while(root!=pre[root])
        root=pre[root];
    while(son!=root)
    {
        temp=pre[son];
        pre[son]=root;
        son=temp;
    }
    return root;
}
bool cmp(const routes a,const routes b)
{
    return a.len <b.len ;
}
int kruskal(int routenum)
{
    int i,j=0;
    sort(road,road+routenum,cmp);
    int sum=0;
    for(i=0;i<routenum;i++)
    {
        int roota=find(road[i].a );
        int rootb=find(road[i].b );
        if(roota!=rootb)
        {
            sum+=road[i].len ;
            ans[j++]=road[i].num ;
            allusenum++;
            pre[roota]=rootb;
        }
    }
    return sum;
}
int main()
{
    int n,routenum;
    cin>>n>>routenum;
    for(int i=0;i<MAX;i++)
        pre[i]=i;
    for(int i=0;i<routenum;i++)
        cin>>road[i].num >>road[i].a >>road[i].b >>road[i].len ;
    int answer=kruskal(routenum);
        cout<<answer<<endl;
    sort(ans,ans+allusenum);
    for(int i=0;i<allusenum;i++)
        cout<<ans[i]<<" ";
    cout<<endl;

}