华为机试——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;
}