BestWiring——Kruskal算法&并查集

        第一次认真地看着书写算法并且发第一篇blog，结果全部TLE~看来还是不够认真，没有针对题目对算法进行优化。今天没课，就好好改代码吧。晚上就是deadline了

//2014/12/10
//潘名扬
//BestWiring 最优布线
//Using Kruskal,quickSort,UFset



#include<malloc.h>
#include<iostream>
using namespace std;

#define maxE 10000
#define maxN 101

typedef int WItem;
typedef struct graph *Graph;
typedef struct edge{ int u; int v; WItem w; }Edge;
typedef struct ufset *UFset;
#define Malloc(type,n)  (type *)malloc((n)*sizeof(type)) 

//“Graph”结构
struct graph{
	WItem NoEdge;	//无边标记
	int n;			//顶点数
	int e;			//边数
	WItem a[maxN][maxN];
};

//并查集结构
struct ufset{		
	int *components;	//并查集中元素与所属子集关系
	int n;				//元素个数
};

//EDGE构造一条边（u，v，w），返回该边
Edge EDGE(int u, int v, WItem w)
{
	Edge e;
	e.u = u; e.v = v; e.w = w;
	return e;
}

//Edges抽取“Graph”图中所有的边到数组a[]，返回边数
int Edges(Edge a[], Graph G)
{
	int i, j, k(0);							//k为边数
	for (i = 1; i <= G->n; i++)
		for (j = i + 1; j <= G->n; j++)
			if (G->a[i][j]>0)
				a[k++] = EDGE(i, j, G->a[i][j]);
	return k;
}

//UFinit将components初始化为size大小的单元素集合，返回初始化的 UFset
UFset UFinit(int size)
{
	int e;
	UFset U = (UFset)malloc(sizeof*U);
	U->components = (int*)malloc((size + 1)*sizeof(int));
	for (e = 1; e <= size; e++)
		U->components[e] = e;
	U->n = size;
	return U;
}

//查找点e所属的集合
int UFfind(int e, UFset U)
{
	return U->components[e];
}

//合并AB两个集合，并将新集合命名为A，返回新集合名
int UFunion(int i, int j, UFset U)
{
	int k;
	for (k = 1; k <= U->n; k++)
		if (U->components[k] == j)
			U->components[k] = i;
	return i;
}

//快速排序部分

#define less(A,B) (A.w<B.w)
#define swap(A,B) {Edge t=A;A=B;B=t;}

//元素划分
int partition(Edge a[], int l, int r)
{
	int i = l - 1, j = r;
	Edge v = a[r];
	//将大于等于x的元素交换到右边区域
	//将小于等于x的元素交换到左边区域
	for (;;)
	{
		while (less(a[++i], v));
		while (less(v, a[--j]))if (j == l)break;
		if (i >= j)break;
		swap(a[i], a[j]);
	}
	swap(a[i], a[r]);
	return i;
}

//快速排序算法
void quicksort(Edge a[], int l, int r)
{
	int i;
	if (r <= l)return;
	i = partition(a, l, r);
	quicksort(a, l, i-1);
	quicksort(a, i + 1, r);
}

//mst为路径详情，返回最小代价
int Kruskal(/*Edge mst[],*/ Graph G)
{
	int e, i, k, s, t,ans(0),sign=1;
	Edge a[maxE];
	UFset U;
	e = Edges(a, G);
	quicksort(a, 0, e - 1);
	U = UFinit(G->n);
	for (i = 0, k = 0; (i < e)&&(k < G->n - 1); i++)
	{
		s = UFfind(a[i].u, U);
		t = UFfind(a[i].v, U);
		if (a[i].w == a[i + 1].w&&UFfind(a[i+1].u, U) ==s&& UFfind(a[i+1].v, U)==t)
			sign=-1;
		if (s != t)
		{
			//选取边a[i]
			/*mst[k++] = a[i];*/
			k++;
			ans+=a[i].w;//本题要求计算最小代价
			UFunion(s, t, U);
		}
	}
	return sign*ans;
}

int main()
{
	int t,x,y,m,ans;
	cin >> t;
	for (int i = 0; i < t; i++)
	{
		ans = 0;
		Graph G = (Graph)malloc(sizeof*G);
		cin >> G->n >>G->e;
		//读取边
		for (int j = 1; j <= G->e; j++)
		{
			cin >> x >> y>> m;
			G->a[x][y] = G->a[y][x] = m;
		}
		ans = Kruskal(G);
		if (ans > 0)
			cout << ans << endl << "Unique!" << endl;
		else
			cout << -1*ans << endl << "Not Unique!" << endl;

	}
	return 0;
} 

TT修改了半天的代码，还是TLE，突然想把cin、cout改成scanf、printf试试，结果……居然没超时了！但是！全部！WA了

惨了惨了，这下连错在哪里都不知道了，自己的几个数据都正确。。。

附上目前全部WA的代码：

//2014/12/10
//潘名扬
//BestWiring 最优布线
//Using Kruskal,quickSort,UFset


#include<stdio.h>
#include<malloc.h>
#include<iostream>
using namespace std;

#define maxE 10000
#define maxN 101

typedef struct edge{ int u; int v;int w; }Edge;
typedef struct ufset *UFset;

//并查集结构
struct ufset{		
	int *components;	//并查集中元素与所属子集关系
	int n;				//元素个数
};


//UFinit将components初始化为size大小的单元素集合，返回初始化的 UFset
UFset UFinit(int size)
{
	int e;
	UFset U = (UFset)malloc(sizeof*U);
	U->components = (int*)malloc((size + 1)*sizeof(int));
	for (e = 1; e <= size; e++)
		U->components[e] = e;
	U->n = size;
	return U;
}

//查找点e所属的集合
int UFfind(int e, UFset U)
{
	return U->components[e];
}

//合并AB两个集合，并将新集合命名为A，返回新集合名
int UFunion(int i, int j, UFset U)
{
	int k;
	for (k = 1; k <= U->n; k++)
		if (U->components[k] == j)
			U->components[k] = i;
	return i;
}

//快速排序部分

#define less(A,B) (A.w<B.w)
#define swap(A,B) {Edge t=A;A=B;B=t;}

//元素划分
int partition(Edge a[], int l, int r)
{
	int i = l - 1, j = r;
	Edge v = a[r];
	//将大于等于x的元素交换到右边区域
	//将小于等于x的元素交换到左边区域
	for (;;)
	{
		while (less(a[++i], v));
		while (less(v, a[--j]))if (j == l)break;
		if (i >= j)break;
		swap(a[i], a[j]);
	}
	swap(a[i], a[r]);
	return i;
}

//快速排序算法
void quicksort(Edge a[], int l, int r)
{
	int i;
	if (r <= l)return;
	i = partition(a, l, r);
	quicksort(a, l, i-1);
	quicksort(a, i + 1, r);
}

//mst为路径详情，返回最小代价
int Kruskal(/*Edge mst[], Graph G*/)
{
	int n, e, s, t, ans(0), isUnique(1), i, k;
	Edge a[maxE];
	UFset U;
	cin >>n >>e;

	//读取边
	for (int j = 0; j < e; j++)
		scanf_s("%d%d%d", &a[j].u,&a[j].v,&a[j].w);
	quicksort(a, 0, e - 1);
	U = UFinit(n);
	for (i = 0, k = 0; (k < n - 1); i++)
	{
		s = UFfind(a[i].u, U);
		t = UFfind(a[i].v, U);
		if (s != t)
		{
			k++;
			ans+=a[i].w;//本题要求计算最小代价
			UFunion(s, t, U);
		}
		if (a[i].w == a[i + 1].w&&UFfind(a[i + 1].u, U) == UFfind(a[i + 1].v, U))
			isUnique = -1;
	}
	if (isUnique > 0)
		printf("%d\n%s\n", ans, "Unique!");
	else
		printf("%d\n%s\n", ans, "Not Unique!");
	return 0;
}

int main()
{
	int testnum;
	scanf_s("%d", &testnum);
	for (int i = 0; i < testnum; i++)
		Kruskal();
	return 0;
}