uvaLive7303 Aquarium （kruskal）

题意：给R*C的房间，每个房间被左上-右下或右上-左下的墙分割为两个小房间，将分割移除有一定花费，问使所有小房间联通需要的最小花费

把每个房间分成左右（上下？）两个点，判一判，本来就联通的加零边，一个房间里的两个点间加花费的边，跑kruskal即可

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;
const int maxN=110;

typedef struct{
    int a,b,l;
}Edge;


Edge eg[maxN*maxN];
bool stage[maxN][maxN];  //true-> false->/
int bcj[maxN*maxN*2];  //(i-1)*C+j
int T,R,C;

bool cmp(Edge a,Edge b){
    return a.l<b.l;
}

int getf(int i){
    return bcj[i]==i?i:bcj[i]=getf(bcj[i]);
}

void add(int a,int b){  //a->b
    bcj[getf(a)]=getf(b);
}

int main(){
    int t,i,j,k,ind,a,b,num,ans;
    char cr;
    
    scanf("%d",&T);
    for(t=1;t<=T;t++){
        scanf("%d%d
",&R,&C);
        for(i=1;i<=R;i++){
            for(j=1;j<=C;j++){
                scanf("%c",&cr);
                stage[i][j]=(cr=='\');
            }
            if(i<R)  scanf("
");
        }
        ind=0;
        for(i=1;i<=R;i++){
            for(j=1;j<=C;j++){
                eg[ind].a=(i-1)*C*2+j*2-1;
                eg[ind].b=(i-1)*C*2+j*2;
                scanf("%d",&eg[ind++].l);
            }
        }
        for(i=1;i<=R;i++){
            ind=(i-1)*C*2;
            bcj[ind+1]=ind+1;
            bcj[ind+C*2]=ind+C*2;
            for(j=1;j<C;j++){
                bcj[ind+j*2]=ind+j*2;
                bcj[ind+j*2+1]=ind+j*2;
            }
        }
        for(i=1;i<R;i++){
            for(j=1;j<=C;j++){
                ind=(i-1)*C*2+j*2;
                a=stage[i][j]?ind-1:ind;
                b=stage[i+1][j]?ind+C*2:ind+C*2-1;
                add(a,b);
            }
        }
        sort(eg,eg+R*C,cmp);
        
        ans=0;
        for(i=0;i<R*C;i++){
            if(getf(eg[i].a)!=getf(eg[i].b)){
                add(eg[i].a,eg[i].b);
                ans+=eg[i].l;
            }
        }
        printf("Case %d: %d
",t,ans);
        
    }
}