1. 程式人生 > >棋子翻轉---- 美團2016研發工程師線上程式設計題

棋子翻轉---- 美團2016研發工程師線上程式設計題

[程式設計題] 棋子翻轉

在4x4的棋盤上擺滿了黑白棋子,黑白兩色的位置和數目隨機其中左上角座標為(1,1),右下角座標為(4,4),現在依次有一些翻轉操作,要對一些給定支點座標為中心的上下左右四個棋子的顏色進行翻轉,請計算出翻轉後的棋盤顏色。

給定兩個陣列Af,分別為初始棋盤和翻轉位置。其中翻轉位置共有3個。請返回翻轉後的棋盤。

測試樣例:
[[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]],[[2,2],[3,3],[4,4]]
返回:[[0,1,1,1],[0,0,1,0],[0,1,1,0],[0,0,1,0]]

class Flip {
public:
    vector<vector<int> > flipChess(vector<vector<int> > vec, vector<vector<int> > f) {
        // write code here
        for ( int i = 0; i < f.size(); ++ i ) {
            int x = f[i][0] - 1 ;
            int y = f[i][1] - 1;
            
            if ( x - 1 >= 0 ) {
                int color = vec[x - 1][y] ;
                if ( color == 0 ) color = 1 ;
                else if ( color == 1 ) color = 0 ;
                    vec[x - 1][y] = color ;
            }
            if ( x + 1 <= 3 ) {
                int color = vec[x + 1][y] ;
                if ( color == 0 ) color = 1 ;
                else if ( color == 1 ) color = 0 ;
                    vec[x + 1][y] = color ;
            }
            
            if ( y - 1 >= 0 ) {
                int color = vec[x][y - 1] ;
                if ( color == 0 ) color = 1 ;
                else if ( color == 1 ) color = 0 ;
                    vec[x][y - 1] = color ;
            }
            if ( y + 1 <= 3 ) {
                int color = vec[x][y + 1] ;
                if ( color == 0 ) color = 1 ;
                else if ( color == 1 ) color = 0 ;
                    vec[x][y + 1] = color ;
            }
        }
        return vec ;
    }
 };