灯泡游戏 （Kruskal)(并查集)

灯泡游戏

时间限制: 1 Sec  内存限制: 64 MB
提交: 9  解决: 4
[提交][状态][讨论版]

题目描述

有
一个n行m列的矩阵，左上角坐标是(0,0)，右下角坐标是(n-1,m-1)。每个格子有一个字符，
“0”至“9”表示数字0至9，“a”至“z”表示数字10至35，“A”至“Z”表示数字36至61。矩阵的每个格子都有一个灯泡，刚开始除了左上角的
灯泡是亮的，其他的灯泡都是灭的，刚开始你的得分是0。
游戏的过程是这样的：每次选一个灯泡是亮的格子X，同时选一个灯泡是灭的格子Y，而且要求
格子X和格子Y是相邻的格子。这个步骤会使你的得分增加，增加的值是：格子X与格子Y代表的数字的差的绝对值。当然，这个步骤会使你选中的灭的灯泡变亮。
重复上述操作，直到所有的灯泡都变成亮的。问你的最大得分是多少？

输入

第1行：两个整数n和m(1≤n，m≤50)。
　　接下来是n行m列的矩阵。
　　

输出

　　一个整数，表示最大得分。

样例输入

2  2
05
aB

样例输出

69

提示

第一次选择：格子(0，0)和格子(1，0)，得分是10-0=10；第二次选择：格子(1，0)和格子(1，1)，得分是37-10：27；第三次选择：格子(1，1)和格子(0，1)，得分是37-5=32，因此总得分是：10+27+32=69。

【分析】又学到了一种算法，求最小生成树的，当然这题是最大生成树，我用的是Kruskal算法，下面是AC代码。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <climits>
#include <cstring>
#include <string>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#define mod 1000000007
#define inf 0x3f3f3f3f
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int N=;
int n,m,k,cnt=-;
int d[][]= {,,,,,-,-,};
int parent[N+];
bool vis[N][N];
char mp[][];
struct man {
    int u,v,w;
} edg[N];//结构体存节点与节点之间的权值
bool cmp(man g,man h) {
    return g.w>h.w;
}//将权值即得分从大到小排序
int change(char ch) {
    if(ch>=''&&ch<='')return (ch-'');
    else if(ch>='a'&&ch<='z')return (ch-);
    else return (ch-);
}//字符转为数字
void init() {
    for(int i=; i<=; i++)parent[i]=i;
}//初始化
int Find(int x) {
    if(parent[x] != x) parent[x] = Find(parent[x]);
    return parent[x];
}//查找并返回节点x所属集合的根节点
void Union(int x,int y) {
    x = Find(x);
    y = Find(y);
    if(x == y) return;
    parent[y] = x;
}//将两个不同集合的元素进行合并
void Build()//将矩形图转化为树
{
    for(int i=; i<n; i++) {
        for(int j=; j<m; j++) {
            for(int k=; k<; k++) {
                int xx=i+d[k][],yy=j+d[k][];
                if(xx>=&&xx<n&&yy>=&&yy<m) {
                    int I=i*+j,J=xx*+yy;
                    if(!vis[I][J]) {
                        edg[++cnt].u=I;
                        edg[cnt].v=J;
                        edg[cnt].w=abs(change(mp[i][j])-change(mp[xx][yy]));
                        vis[I][J]=true;
                    }
                }
            }
        }
    }
}
void Kruskal() {
    int sumweight=;//生成树的总权值，即所得分数
    int num=;//已经选用边的数目
    int u,v;//顶点
    init();
    for(int i=; i<=cnt; i++) {
        u=edg[i].u;
        v=edg[i].v;
        if(Find(u)!=Find(v)) {
            sumweight+=edg[i].w;
            num++;
            Union(u,v);
        }
        if(num>=n*m-) break;//边已经全部建好
    }
    printf("%d\n",sumweight);
}
int main() {
    memset(vis,false,sizeof(vis));
    int u,v,w;
    cin>>n>>m;
    for(int i=; i<n; i++)cin>>mp[i];
    Build();
    sort(edg,edg+cnt+,cmp);
    Kruskal();
    return ;
}