HNUSTOJ-1674 水果消除(搜索或并查集)

1674: 水果消除

时间限制: 2 Sec 内存限制: 128 MB
提交: 335 解决: 164
[提交][状态][讨论版]

题目描述

“水果消除”是一款手机游戏，相信大家都玩过或玩过类似的游戏。

下面是“水果消除”游戏的一种初始状态。

消除的基本规则：如果有2个或2个以上的相同水果连在一起，则可以点选并消除。

请问在某一种状态下，有几种可以点选并消除的选择方案。

例如，对于上图所示的初始状态，将有6种点选并消除的选择方案。这6种方案依次如下图所示。

输入

先输入一个整数n，表示放水果的格子总数为n*n。n取3到1000之间的整数（含3和1000）。

然后依次输入n*n个表示水果的数据，不同的水果用不同的数字表示，同一种水果用相同的数字表示。

表示水果的数字编号从1开始，不超过100。

输出

在输入数据对应的初始状态下，有几种点选并消除的选择方案。

输出方案数。

样例输入

6
1 1 2 2 2 2
1 3 2 1 1 2
2 2 2 2 2 3
3 2 3 3 1 1
2 2 2 2 3 1
2 3 2 3 2 2

样例输出

提示

来源

Derson,2015年湖南科技大学校赛

这道题可以直接DFS求连通块，也可以标号以后并查集；求连通块的时候要注意包含的格子至少为两个计数才加一；
DFS求连通块：

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

const int N = 1000 + 50;
int mat[N][N];
bool visit[N][N];
int cur;
void DFS(int i,int j,int n,int color){
    if(i<0 || j<0 || i>=n || j>=n) return;
    if(visit[i][j]) return;
    if(mat[i][j]!=color) return;
    visit[i][j] = true;
    cur++;
    DFS(i+1,j,n,color);
    DFS(i-1,j,n,color);
    DFS(i,j+1,n,color);
    DFS(i,j-1,n,color);
}

int DFS_AL(int n){
    int Count=0;
    memset(visit ,0,sizeof(visit));
    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
    if(!visit[i][j]){
        cur=0;
        DFS(i,j,n,mat[i][j]);
        if(cur>1)
         Count++;
    }
    return Count;
}

void Input_data(int n){
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        scanf("%d",&mat[i][j]);
}

int main(){
    int n;
    scanf("%d",&n);
    Input_data(n);
    printf("%d
",DFS_AL(n));
}

并查集：

#include<cstdio>
#include<queue>

using namespace std;

const int MaxSize = 1000+5;
struct node1
{
    int p;
    int num;
}pre[MaxSize*MaxSize];

const int dir[4][2]= {{0,1},{0,-1},{1,0},{-1,0}};
short mat[MaxSize][MaxSize];
bool visit[MaxSize][MaxSize];
int Find(int x)
{
    return pre[x].p==x?x:(pre[x].p=Find(pre[x].p));
}

void Merge(int x,int y)
{
    x= Find(x),y=Find(y);
    if(x!=y)
    {
        pre[y].p=x;
        pre[x].num += pre[y].num;
    }
}
int main()
{
    int n,i,j,k=0,ans,h;
    scanf("%d",&n);
    for(i=0; i<n; i++)
        for(j=0; j<n; j++)
        {
            scanf("%hd",&mat[i][j]);
            pre[i*n+j].p=k++;
            pre[i*n+j].num=1;
            visit[i][j]=false;
        }
    for(i=0; i<n; i++)
        for(j=0; j<n; j++)
        {
            if(!visit[i][j])
            {
                visit[i][j] = true;
                for(h=0; h<4; h++)
                    if(i+dir[h][0]>=0 && i+dir[h][0]<n && j+dir[h][1]>=0 && j+dir[h][1]<n )
                        if(mat[i][j] == mat[i+dir[h][0]][j+dir[h][1]])
                            Merge((i*n+j),((i+dir[h][0])*n+j+dir[h][1]));
            }
        }
    for(ans=0,i=0; i<k; i++)
        if(pre[i].p==i && pre[i].num > 1 )
            ans++;
    printf("%d
",ans);
}


//如有错误，还请留言指出