【九度OJ】1028【快排】【kruskal】【畅通工程系列2008】

稍微改一下1017题的代码，在读入边时就合并点，相当于为树的边，再用Kruskal的思想即可

代码：

package Test1;import java.io.BufferedReader;import java.io.IOException;import java.io.InputStreamReader;import java.io.StreamTokenizer;public class Test38_1028 {/** * by qr jobdu 1028 2014-8-21 *  * @throws IOException */public static void main(String[] args) throws IOException {StreamTokenizer st = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));while (true) {st.nextToken();int n = (int) st.nval;if (n == 0)break;int m=n*(n-1)/2;UnionFindSet ufs=new UnionFindSet(n+1);int num=0;//num of edgeint road[][]=new int[m][3];for(int i=0;i<m;i++){for(int j=0;j<3;j++){st.nextToken();road[i][j]=(int)st.nval;}int n1=road[i][0];int n2=road[i][1];st.nextToken();int flag=(int)st.nval;if(flag==1)if(ufs.find(n1)!=ufs.find(n2)){ufs.union(n1,n2);num++;}}quicksort(road,0,m-1);//quick sort increasing  from 0 to m(inclusive both)int len=0,node1,node2;  for(int i=0;i<m;i++){node1=road[i][0];node2=road[i][1];if(ufs.find(node1)!=ufs.find(node2)){ufs.union(node1,node2);len+=road[i][2];num++;}if(num==(n-1))break;}System.out.println(len);}}private static void quicksort(int[][] arr, int i, int j) {if(i==j)   //只有一个元素  结束递归return;int len=arr[0].length;int copyi=i;   //保存指针int copyj=j;int pivot[]=new int[len];   //获取支点for(int k=0;k<arr[0].length;k++)pivot[k]=arr[i][k];while(i<j){     //将比支点小的放到前面，比支点大的放到后面while(i<j){if(arr[j][2]<pivot[2]){for(int k=0;k<arr[0].length;k++)arr[i][k]=arr[j][k];break;}j--;}while(i<j){if(arr[i][2]>pivot[2]){for(int k=0;k<arr[0].length;k++)arr[j][k]=arr[i][k];break;}i++;}}for(int k=0;k<arr[0].length;k++)  //支点放到相应位置arr[i][k]=pivot[k];//左右分别快排if(copyi<i)  //判断下quicksort(arr,copyi,i-1);   //left  i??if(copyj>j)quicksort(arr,i+1,copyj); //right}}class UnionFindSet{  //并查集类int set[];int size[];UnionFindSet(){}UnionFindSet(int num){set=new int[num];size=new int[num];for(int i=0;i<num;i++){set[i]=i;size[i]=1;  //size数组为了平衡}}void union(int i,int j){int p=find(i);int q=find(j);if(p==q)return;if(size[p]<size[q]){  //平衡树大小set[p]=q;size[q]+=size[p];}else{set[q]=p;size[p]+=size[q];}}int find(int i){while(i!=set[i]){i=set[i];}return i;}}