poj2485 kruskal与prim
Kruskal:
#include<iostream> #include<cstdio> #include<algorithm> using namespace std; //#define debug #if defined debug #define CDBG(format,...) printf("File: "__FILE__", Line: %05d: "format"\n", __LINE__, ##__VA_ARGS__) int open_file(){ FILE *fp ; if((fp = freopen("input3.txt","r",stdin)) == NULL){ CDBG("error in freopen"); return -1; } CDBG("success in freopen"); return 0; } #else #define CDBG(format,...) do{}while(0) #endif #define N 510 int n,k,pre[N]; struct point { int x; int y; int len; }p[N*N]; /*int cmp(const void * x,const void * y) { return ((point*)x)->len>((point*)y)->len?1:-1; }*/ void swap(point &x,point &y) { point temp; temp = x; x = y; y = temp; // 从指针改成数组就行了 } int choose_pivot(int i,int j ) { return((i+j) /2); } void quicksort(point list[],int m,int n) { int key,i,j,k; if( m < n) { k = choose_pivot(m,n); if(m != k)
swap(list[m],list[k]); key = list[m].len; i = m+1; j = n; while(i <= j) { while((i <= n) && (list[i].len <= key)) i++; while((j >= m) && (list[j].len > key)) j--; if( i < j) swap(list[i],list[j]); } // 交换两个元素的位置 if(m != j)
swap(list[m],list[j]); // 递归地对较小的数据序列进行排序 quicksort(list,m,j-1); quicksort(list,j+1,n); } } int cmp(point x,point y) { return x.len<y.len; } int find(int x) { while(x!=pre[x]) x=pre[x]; return x; } void kruskal() { int i,mix,a,b; mix=0; //sort(p,p+k,cmp); for(i = 0; i < k ;i++) CDBG("%d ",p[i].len); quicksort(p,0,k-1); //跟 sort 结果完全不一样 for(i = 0; i < k ;i++) CDBG("%d ",p[i].len); for(i=1;i<=n;i++) pre[i]=i;// 可以初始化为 -1 等值 for(i=0;i<k;i++) { a=find(p[i].x); b=find(p[i].y); if(a!=b) { if(p[i].len>mix) mix=p[i].len; pre[b]=a; } } cout<<mix<<endl; } void input() { scanf("%d",&n); k=0; for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) { scanf("%d",&p[k].len); if(i != j && p[k].len != 0){ p[k].x=i; p[k].y=j; k++; //过滤掉ii 点 } } } int main() { int T; #ifdef debug if(open_file() != 0) return -1; #endif scanf("%d",&T); while(T--) { input(); kruskal(); } return 0; }
void quicksort(int left, int right){ if (left > right) return; int l = left; int r = right+1; path v, temp; v = ipath[left]; while (1){ while (l < r && ipath[--r].len >= v.len); while (l < r && ipath[++l].len <= v.len); if (l >= r) break; temp = ipath[l]; ipath[l] = ipath[r]; ipath[r] = temp; } ipath[left] = ipath[l]; ipath[l] = v; quicksort(left, l - 1); quicksort(l + 1, right); }
Prime:
参考: http://www.cnblogs.com/biyeymyhjob/archive/2012/07/30/2615542.html
/*
题目意思就是求最小生成树，并返回最长edge
求最小生成树一般有两种方法，prim算法 O[N^2]和kruskal算法 O[ElogE]，在这里我推荐使用prim算法，因为此题不是稀疏图，kruskal算法耗时较多（但也不会超时）
prim算法的基本思想
1.清空生成树，任取一个顶点加入生成树
2.在那些一个端点在生成树里，另一个端点不在生成树里的边中，选取一条权最小的边，将它和另一个端点加进生成树
3.重复步骤2，直到所有的顶点都进入了生成树为止，此时的生成树就是最小生成树
Prime()
{
//开始默认把1加入点集里
for(2~m)
dist[i]=map[1][i];
for(2~m)
{
//找跟相连的最短边
for(2~m)
{
if(!visit[j] && dist[j]<inf)
index="j";
}=""
visit[index]="1";
根据知道的点更新cost，这里方法有点像dijkstra，注意！！！=""
for(2~m)="" {="" if(!visit[j]="" &&="" cost[j]="">map[index][j])
cost[j]=map[index][j];
}
}
}
*/
#include<stdio.h> #include<string.h> #define debug #if defined debug #define CDBG(format,...) printf("File:"__FILE__", Line:%05d: "format"\n", __LINE__, ##__VA_ARGS__) #else #define CDBG(format,...) do{}while(0) #endif #define N 511 // n -> [3,500] #define inf 0x3f3f3f3f //distance -> [1,65536] int villages[N][N]; int visited[N]; int lowcost[N]; int open_file(){ FILE *fp ; if((fp = freopen("input3.txt","r",stdin)) == NULL){ CDBG("error in freopen"); return -1; } CDBG("success in freopen"); return 0; } void reset_visited(int n){ int i = 1; for( i = 1; i<= n; i++){ visited[i] = 0; lowcost[i] = 0;} } //下面2个方法都可以 int prim(int start_p,int n){ int i,j,pos,min = 0,result = 0; reset_visited(n); //start at point 1 visited[start_p] = 1; pos = start_p; //set low array from point 1 to other point for(i = 1; i <= n; i++) if(i != pos) { lowcost[i] = villages[pos][i]; CDBG("set lowcost[%d] = %d",i,lowcost[i]); } //check n-1 times again ， 每次添加一个点到 最小树 for(i = 1; i < n; i++){ // 使用 while循环代替 for 容易有一些问题 //find min distance 寻找下一个最短距离 min = inf; for(j = 1; j <= n; j++){ if(visited[j] == 0 && lowcost[j] < min) { min = lowcost[j];//从未访问的点找到 最短距离的 点
pos = j; CDBG("min = %d pos = %d j = %d",min,pos,j); } } CDBG("min = %d pos = %d ",min,pos); //add min distance; if(result < min) result = min; // 找最小树的最长距离 visited[pos] = 1; //update 更新当前最小的点 到 所有未访问过的点 的距离 for(j = 1; j <= n; j++){ if(visited[j] == 0 && lowcost[j] > villages[pos][j]) { lowcost[j] = villages[pos][j]; CDBG("update lowcost[%d] = %d",j ,lowcost[j]); } } } return result; } int prim(int n) { int u = 1; int i,j,k,start,min,max; //memset(vis,0,sizeof(vis)); reset_visited(n); for(i=1;i<=n;i++) if(i!=u) lowcost[i]=villages[1][i]; visited[1]=true; k=0; min=0;max=0; for(i=1;i<n;i++) //n-1条边 { min=inf; for(j=1;j<=n;j++) if(lowcost[j]<min&&!visited[j]) { min=lowcost[j]; k=j; } if(min>max) max=min; visited[k]=true; for(j=1;j<=n;j++) if(lowcost[j]>villages[k][j]&&!visited[j]) { lowcost[j]=villages[k][j]; } } return max; //cout<<max<<endl; } void read_villages(int n){ for(int i = 1; i <= n; i++) for(int j = 1; j <= n; j++) { villages[i][j] = 0; scanf("%d",&villages[i][j]); } } void check_villages(int n){ for(int i = 1; i <= n; i++){ for(int j = 1; j <= n; j++) printf("%d ",villages[i][j]); // 应该是这个导致了 Output Limit Exceeded printf("\n"); } } int main(){ int T,n; #ifdef debug if(open_file() != 0) return -1; #endif scanf("%d",&T); while(T--){ scanf("%d",&n); read_villages(n); //check_villages(n); int ans=0; ans = prim(2,n); // 可以从任意一点开始 //ans = prim(n); printf("%d\n",ans); } return 0; }
