2-2 畅通工程之局部最小花费问题 (30分)

某地区经过对城镇交通状况的调查，得到现有城镇间快速道路的统计数据，并提出“畅通工程”的目标：使整个地区任何两个城镇间都可以实现快速交通（但不一定有直接的快速道路相连，只要互相间接通过快速路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建快速路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全地区畅通需要的最低成本。

输入格式:

输入的第一行给出村庄数目N (1≤N≤100)；随后的N(N−1)/2行对应村庄间道路的成本及修建状态：每行给出4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态 — 1表示已建，0表示未建。

输出格式:

输出全省畅通需要的最低成本。

输入样例:

4
1 2 1 1
1 3 4 0
1 4 1 1
2 3 3 0
2 4 2 1
3 4 5 0

输出样例:

3

代码一（待优化）： 

最后一个测试点过不去的代码

#include <stdio.h>
#include <set>
#define MAX 100000
using namespace std;
int a[105][105]={100000};

int main()
{
    int n;
    scanf("%d",&n);
    set<int> village;
    set<int>::iterator iter_set;//两趟点点表示作用域


    int x,y,dist,is,sum=0,num1=0;
    for(int i=0;i<n*(n-1)/2;i++)
    {
        scanf("%d%d%d%d",&x,&y,&dist,&is);
        if(is)
        {
            if(village.find(x)==village.end())
            {
                num1++;
                village.insert(x);
            }
            if(village.find(y)==village.end())
            {
                num1++;
                village.insert(y);
            }
        }
        else
        {
            a[x][y]=a[y][x]=dist;
        }
    }
    int min,labal;
    if(!num1)
    {
        village.insert(1);
        num1++;
    }
    for(int i=0;i<n-num1;i++)//共需要再并入n-num1个村庄
    {
        min=MAX;
        for(iter_set=village.begin();iter_set!=village.end();iter_set++)
        {
            for(int k=1;k<=n;k++)
            {
                if(a[*iter_set][k]<min&&village.find(k)==village.end())
                {
                    labal=k;
                    min=a[*iter_set][k];
                }
            }
        }
        village.insert(labal);
        sum+=min;
    }
    printf("%d",sum);
    return 0;
}
/**遍历用代码，不使用可以删去
for(iter_set=village.begin();iter_set!=village.end();iter_set++)
        printf("%d ",*iter_set);
*/

稍微改动了一下，已经修好路的村庄不再直接加入集合，改成设置修路花费为0，改完最后一个测试点通过

代码二（C++，30分代码）：

#include <stdio.h>
#include <set>
#define MAX 100000
using namespace std;
int a[105][105]={100000};

int main()
{
    int n;
    scanf("%d",&n);
    set<int> village;
    set<int>::iterator iter_set;//两趟点点表示作用域


    int x,y,dist,is,sum=0,num1=0;
    for(int i=0;i<n*(n-1)/2;i++)
    {
        scanf("%d%d%d%d",&x,&y,&dist,&is);
        if(is)
        {
            a[x][y]=a[y][x]=0;
        }
        else
        {
            a[x][y]=a[y][x]=dist;
        }
    }

    int min,labal;

        village.insert(1);
        num1++;
    for(int i=0;i<n-num1;i++)//共需要再并入n-num1个村庄
    {
        min=MAX;
        for(iter_set=village.begin();iter_set!=village.end();iter_set++)
        {
            for(int k=1;k<=n;k++)
            {
                if(a[*iter_set][k]<min&&village.find(k)==village.end())
                {
                    labal=k;
                    min=a[*iter_set][k];
                }
            }
        }
        village.insert(labal);
        sum+=min;
    }
    printf("%d",sum);
    return 0;
}

代码三（C语言，30分代码）：

和C++版代码比起来就是自己手搓一个set出来，不知道为啥学校非要求用C语言，库函数着实有点少

#include <stdio.h>
#include <stdlib.h>

#define MAX 100000
int a[105][105]={100000};
int village[105]={0},num1=0;


int main()
{
    int n;
    scanf("%d",&n);

    int x,y,dist,is,sum=0;
    for(int i=0;i<n*(n-1)/2;i++)
    {
        scanf("%d%d%d%d",&x,&y,&dist,&is);
        if(is)
        {
            a[x][y]=a[y][x]=0;
        }
        else
        {
            a[x][y]=a[y][x]=dist;
        }
    }
    int min,labal;

    village[num1++]=1;
    for(int i=0;i<n-1;i++)
    {
        min=MAX;
        for(int j=0;j<num1;j++)
        {
            for(int k=1;k<=n;k++)
            {
                if(a[village[j]][k]<min&&!set_find(k))
                {
                    labal=k;
                    min=a[village[j]][k];
                }
            }
        }
        set_insert(labal);
        sum+=min;
    }
    printf("%d",sum);
    return 0;
}
int set_find(int k)
{
    for(int i=0;i<num1;i++)
        if(k==village[i])return 1;
    return 0;
}
void set_insert(int x)
{
    for(int i=0;i<num1;i++)
    {
        if(village[i]>x)
        {
            for(int j=num1;j>i;j--)
                village[j]=village[j-1];
            village[i]=x;
            num1++;
            return ;
        }
    }
    village[num1++]=x;
    return ;
}