[编程题]画家小Q 腾讯 C++

[编程题]画家小Q

时间限制：C/C++ 1秒，其他语言2秒空间限制：C/C++ 32M，其他语言64M
画家小Q又开始他的艺术创作。小Q拿出了一块有NxM像素格的画板, 画板初始状态是空白的,用’X’表示。
小Q有他独特的绘画技巧,每次小Q会选择一条斜线, 如果斜线的方向形如’/’,即斜率为1,小Q会选择这条斜线中的一段格子,都涂画为蓝色,用’B’表示;如果对角线的方向形如’’,即斜率为-1,小Q会选择这条斜线中的一段格子,都涂画为黄色,用’Y’表示。
如果一个格子既被蓝色涂画过又被黄色涂画过,那么这个格子就会变成绿色,用’G’表示。
小Q已经有想画出的作品的样子, 请你帮他计算一下他最少需要多少次操作完成这幅画。

输入描述:

每个输入包含一个测试用例。
每个测试用例的第一行包含两个正整数N和M(1 <= N, M <= 50), 表示画板的长宽。
接下来的N行包含N个长度为M的字符串, 其中包含字符’B’,‘Y’,‘G’,‘X’,分别表示蓝色,黄色,绿色,空白。整个表示小Q要完成的作品。

输出描述:

输出一个正整数, 表示小Q最少需要多少次操作完成绘画。

示例1
输入

4 4
YXXB
XYGX
XBYY
BXXY

输出

3

说明

XXXX
XXXX
XXXX
XXXX
->
YXXX
XYXX
XXYX
XXXY
->
YXXB
XYBX
XBYX
BXXY
->
YXXB
XYGX
XBYY
BXXY

思路

逐行扫描。由于上方一定是已经处理过了的，所以如果是B向左下画，如果是Y向右下画，如果是G就左右开弓。

我的解答

因此传参麻烦，干脆全都写一个函数里了，可读性不好。

#include <iostream>

using namespace std;

void printMatrix(char** mt,int m,int n) {
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            cout << mt[i][j] << "   ";
        }
        cout << endl;
    }
    cout << "-------------------------";
}

int main() {
    int n, m, cnt = 0;
    char b[50][50];
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> b[i][j];
        }
    }
    //printMatrix(b, 6, 5);
    int ii, jj;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (b[i][j] == 'X') continue;
            ii = i; jj = j; cnt++;
            if (b[i][j] == 'B') {
                while (ii < n-1 && jj > 0) {
                    if (b[ii + 1][jj - 1] == 'B') {
                        b[ii + 1][jj - 1] = 'X';
                    }
                    else if (b[ii + 1][jj - 1] == 'G') {
                        b[ii + 1][jj - 1] = 'Y';
                    }
                    else break;
                    ii++; jj--;
                }
            }
            if (b[i][j] == 'Y') {
                while (ii < n-1 && jj < m-1) {
                    if (b[ii + 1][jj + 1] == 'Y') {
                        b[ii + 1][jj + 1] = 'X';
                    }
                    else if (b[ii + 1][jj + 1] == 'G') {
                        b[ii + 1][jj + 1] = 'B';
                    }
                    else break;
                    ii++; jj++;
                }
            }
            if (b[i][j] == 'G') {
                cnt++;
                int iii = i; int jjj = j;
                while (ii < n-1 && jj > 0) {
                    if (b[ii + 1][jj - 1] == 'B') {
                        b[ii + 1][jj - 1] = 'X';
                    }
                    else if (b[ii + 1][jj - 1] == 'G') {
                        b[ii + 1][jj - 1] = 'Y';
                    }
                    else break;
                    ii++; jj--;
                }
                while (iii < n-1 && jjj < m-1) {
                    if (b[iii + 1][jjj + 1] == 'Y') {
                        b[iii + 1][jjj + 1] = 'X';
                    }
                    else if (b[iii + 1][jjj + 1] == 'G') {
                        b[iii + 1][jjj + 1] = 'B';
                    }
                    else break;
                    iii++; jjj++;
                }
            }
            //cout << endl;
            //printMatrix(b, 6, 5);
        }
    }
    cout << cnt;
    return 0;
}