NC15758 ——火车上的2连座

链接：https://ac.nowcoder.com/acm/problem/15758
来源：牛客网

题目描述
程序员A和B要去一起坐火车，这列火车上的每一排都有4个座位，而且这四个座位
被过道分成了两半。当A和B到了车上时，一些位子已经有人了。
A和B是好基友，于是他们想要找一对连在一起的座位。两个连在一起的座位是同一排但是不被过道隔开的两个座位。给定一列火车上的座位情况，请你寻找一下能否找到一对连座？

输入描述:
每组样例第一行为一个整数n，1<=n<=1000，表示火车一共有n排座位。
之后的n行每行为一个含有五个字符的字符串，第i个字符串的五个字符表示第i排座位的情况。每一个字符串的第三个字符都为字符‘|’，表示过道，其余每个字符表示一个座位的占用情况。字符’O’表示座位为空，字符’X’表示座位上已经有人，即被占用。

输出描述:
如果能够找到一组连座，则先输出一行字符串“YES”(不要输出引号)在第一行，在接下来的n
行输出车的座位情况，除了A和B的座位用字符’+'表示，其余输出格式与输入格式中车的座位情况一致。
如果不能够找到一组连座，仅输出一行“NO”（不要输出引号）即可。
有多组座位安排方式时，将A和B安排在前面的排，如果同一排还有两组可行解，选择将A和B排在左边。
例如一组车的情况为这样的时候，应该将其安排为箭头右边的情况。
OO|OO --> ++|OO
OO|OO --> OO|OO

示例1
输入

6
OO|OX
XO|XX
OX|OO
XX|OX
OO|OO
OO|XX
4
XO|OX
XO|XX
OX|OX
XX|OX
5
XX|XX
XX|XX
XO|OX
XO|OO
OX|XO

输出

YES
++|OX
XO|XX
OX|OO
XX|OX
OO|OO
OO|XX
NO
YES
XX|XX
XX|XX
XO|OX
XO|++
OX|XO

说明
注意
第一组样例中，下面的安排是不合法的。
O+|+X
XO|XX
OX|OO
XX|OX
OO|OO
OO|XX

ac代码：

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int n;
    while(cin>>n&&(n!=0))
    {
        char wei[n][6],x=-1,y=-1;
        for(int i=0; i<n; i++)
        {
            cin>>wei[i];
            if(x==-1&&wei[i][0]=='O'&&wei[i][1]=='O')
            {
                x=i;
                y=0;
            }
            if(x==-1&&wei[i][3]=='O'&&wei[i][4]=='O')
            {
                x=i;
                y=3;
            }
        }
        if(x==-1)
        {
            cout<<"NO"<<endl;
        }
        else
        {
            cout<<"YES"<<endl;
            for(int i=0; i<n; i++)
            {
                for(int j=0;j<5;j++)
                {
                    if(x==i&&(y==j||y+1==j))
                        cout<<"+";
                    else
                        cout<<wei[i][j];
                }
                cout<<endl;
            }
        }
    }
    return 0;
}