在一些邊已建的情況下求最小產生樹hdu1102

hdu1102 ：題意是在一個無向圖中有一些邊已經建好求最小產生樹，思路很簡單就是將已建的道路權值變為0即可。

我用的是kruskal求MST

import java.util.Arrays;import java.util.Scanner;public class Main {static int max=10010;static int parent[]=new int[max];static line data[];static int Nodenum;static void init(){for(int i=0;i<max;i++){parent[i]=i;}}static int find(int x){if(parent[x]!=x)parent[x]=find(parent[x]);return parent[x];}static void union(int fa,int fb){parent[fa]=fb;}static int kruskal(){int ans=0;init();for(int i=1;i<=Nodenum;i++){int fa=find(data[i].start);int fb=find(data[i].end);if(fa!=fb){ans+=data[i].value;union(fa,fb);}}return ans;}public static void main(String[] args) {Scanner in=new Scanner(System.in);while(in.hasNext()){int num=in.nextInt();Nodenum=num*(num-1)/2;data=new line[Nodenum+1];int len=0;for(int i=1;i<=num;i++){for(int j=1;j<=num;j++){int temp=in.nextInt();if(j>i){len++;data[len]=new line(i,j);data[len].value=temp;}}}int qnum=in.nextInt();for(int i=1;i<=qnum;i++){int a=in.nextInt();int b=in.nextInt();for(int j=1;j<=Nodenum;j++){if(data[j].start==a&&data[j].end==b){data[j].value=0;break;}}}Arrays.sort(data,1,Nodenum+1);//for(int i=1;i<=Nodenum;i++)//{//System.out.println(data[i].value);//}System.out.println(kruskal());}}}class line implements Comparable<line>{int start,end;int value;line(int x,int y){this.start=x;this.end=y;}@Overridepublic int compareTo(line o) {if(this.value<o.value)return -1;else return 1;}}