Online JudgeProblem SetAuthorsOnline ContestsUser
Web Board
Home Page
F.A.Qs
Statistical Charts
Problems
Submit Problem
Online Status
Prob.ID:
Register
Update your info
Authors ranklist
Current Contest
Past Contests
Scheduled Contests
Award Contest
User ID:
Password:
  Register

kruskal过的

Posted by xym_sun at 2013-03-10 01:23:49 on Problem 3026
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;

#define MAXN 110
#define MAXM 10010

int r,c;
int pnx[MAXN],pny[MAXN],dis[55][55],vis[55][55],num;
int uu[MAXM],vv[MAXM],ww[MAXM],et;
char map[55][55];
int fa[MAXN],nr[MAXM];

const int dx[] = {-1,1,0,0};
const int dy[] = {0,0,-1,1};

void bfs(int k)
{
    memset(vis,0,sizeof(vis));
    queue<int> q;
    int u,v;
    int x,y,nx,ny;
    x = pnx[k];
    y = pny[k];
    u = x*c+y;
    dis[x][y] = 0;
    vis[x][y] = 1;
    q.push(u);
    while(!q.empty()){
        u = q.front();
        q.pop();
        x = u/c;
        y = u%c;
        for(int i=0;i<4;i++){
            nx = x+dx[i];
            ny = y+dy[i];
            if(nx>=0&&ny>=0&&nx<r&&ny<c&&!vis[nx][ny]&&map[nx][ny]!='#'){
                v = nx*c+ny;
                q.push(v);
                vis[nx][ny] = 1;
                dis[nx][ny] = dis[x][y]+1;
            }
        }
    }
}

int find(int x)
{
    return x==fa[x]? x:fa[x]=find(fa[x]);
}

int cmp(int a,int b)
{
    return ww[a]<ww[b];
}

int Kruskal()
{
    for(int i=0;i<num;i++) fa[i]=i;
    for(int i=0;i<et;i++) nr[i]=i;
    sort(nr,nr+et,cmp);
    int ans = 0;
    for(int i=0;i<et;i++){
        int e = nr[i];
        int x = find(uu[e]);
        int y = find(vv[e]);
        if(x!=y){
            ans += ww[e];
            fa[y] = x;
        }
    }
    return ans;
}

int main()
{
    //freopen("input.in","r",stdin);
    int cas;
    scanf("%d",&cas);
    while(cas--){
        scanf("%d%d",&c,&r);
        char tmp[115];gets(tmp);
        for(int i=0;i<r;i++){
            gets(map[i]);
        }
        num = 0;
        for(int i=0;i<r;i++){
            for(int j=0;j<c;j++){
                if(map[i][j]=='A'||map[i][j]=='S'){
                    pnx[num] = i;
                    pny[num++] = j;
                }
            }
        }
        et = 0;
        for(int i=0;i<num;i++){
            memset(dis,0,sizeof(dis));
            bfs(i);
            for(int j=i+1;j<num;j++){
                uu[et] = i;
                vv[et] = j;
                ww[et++] = dis[pnx[j]][pny[j]];
            }
        }
        int res = Kruskal();
        printf("%d\n",res);
    }
    return 0;
}

Followed by:

Post your reply here:
User ID:
Password:
Title:

Content:

Home Page   Go Back  To top


All Rights Reserved 2003-2013 Ying Fuchen,Xu Pengcheng,Xie Di
Any problem, Please Contact Administrator