华为机试——CandyBrush游戏

CandyBrush游戏

描述: CandyBrush一款流行的消除游戏，在一个方阵中布满各种糖果，任意交换两个糖果，如果交换后出现横向或者竖向有连续三个相同的情况，则可以消除糖果并得分。

输入25个字符组成的字符串（编号为1到25），表示5×5的方阵，每个字符表示不同类型的糖果（区分大小写），判断是否有解，即交换某两个糖果后能够消除糖果。

举例，下面的方阵无解

T M T M O

X R U C q

A B C X R

U R M T O

T Q C R A

下面的方阵交换(0, 2)和(1, 2)两个糖果后，能消除M

T M T M O

X R M C q

A B C X R

U R M T O

T Q C R A

如果有解，则输出YES，同时输出所有交换方案中，被交换糖果最小编号（1到25），比如：上面的方阵中，被交换的糖果，编号最小的为糖果“T”，编号为3。

如果无解，则输出NO。

运行时间限制: 无限制

内存限制: 无限制

输入:

输入个字符组成的字符串，不含空格，表示5×5的方阵

输出:

如果有解，则输出YES，同时输出所有交换方案中，被交换糖果最小编号，以空格隔开。如果无解，则输出NO。

样例输入:

TMTMOXRMCqABCXRURMTOTQCRA

样例输出:

YES 3

答案提示:

输出最小糖果编号的意思，就是从左至右，从上到下遍历糖果，找到第一个跟其他交换后能消除的糖果即可。

#include<iostream>
#include<string>
using namespace std;

char a[5][5];
void swap(int i1,int j1,int i2,int j2)
{
char t=a[i1][j1];
a[i1][j1]=a[i2][j2];
a[i2][j2]=t;
}

int xiaochu()
{
for(int i=0;i<5;i++)
for(int j=0;j<3;j++)
{
if(a[i][j]==a[i][j+1] && a[i][j]==a[i][j+2])
return 1;
}

for(int j=0;j<5;j++)
for(int i=0;i<3;i++)
{
if(a[i][j]==a[i+1][j] && a[i][j]==a[i+2][j])
return 1;
}
return 0;
}
int main()
{
string str;
cin>>str;
int n=0,flag,min;
for(int i=0;i<5;i++)
for(int j=0;j<5;j++)
a[i][j]=str[n++];
for(int i=0;i<5;i++)
for(int j=0;j<5;j++)    //从左到右，从上到下，第一个满足的编号最小，知道陈寻终止
{
if(j+1<5)
{
swap(i,j,i,j+1);
flag=xiaochu();
if(flag==1)
{
min=5*i+j+1;
cout<<"Yes "<<min<<endl;
return 0;
}
swap(i,j,i,j+1);    //不符合时要交换回来，进行下次比较
}
if(i+1<5)
{
swap(i,j,i+1,j);
flag=xiaochu();
if(flag==1)
{
min=i*5+j+1;
cout<<"Yes "<<min<<endl;
return 0;
}
swap(i,j,i+1,j);
}
}
cout<<"NO"<<endl;
return 0;
}