C++刷面试题（棋子翻转）【数据结构】【美团】【2020】

糖逗

题目描述：

1. 在4*4的棋盘上摆满了黑白棋子，黑白两色的位置和数目随机其中左上角坐标为(1,1),右下角坐标为(4,4),现在依次有一些翻转操作，要对一些给定支点坐标为中心的上下左右四个棋子的颜色进行翻转，请计算出翻转后的棋盘颜色。
   
2. 
   
3. 
   
4. 输入描述:
   
5. 给定两个数组，分两行
   
6. 
   
7. 第一行为分别为初始棋盘，为4*4矩阵，其中0表示白色棋子，1表示黑色棋子
   
8. 
   
9. 第二行为翻转位置，其中翻转位置共有3个
   
10. 
    
11. 
    
12. 
    
13. 输出描述:
    
14. 请返回翻转后的棋盘，为4*4矩阵
    
15. 示例1
    
16. 输入
    
17. [[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]]
    
18. [[2,2],[3,3],[4,4]]
    
19. 输出
    
20. [[0,1,1,1],[0,0,1,0],[0,1,1,0],[0,0,1,0]]

复制代码

题目来源：牛客网

1. #include <vector>
   
2. #include <iostream>
   
3. using namespace std;
   
4. 
   
5. 
   
6. vector<vector<int> > solution(vector<vector<int> > &matrix, vector<vector<int> > &change){
   
7.     int len = 4;
   
8.     for(int i = 0; i < change.size(); i++){
   
9.         int a = change[i][0]-1;
   
10.         int b = change[i][1]-1;
    
11.         if(a - 1 >= 0){
    
12.             if(matrix[a-1][b] == 0){
    
13.                 matrix[a-1][b] = 1;
    
14.             }else matrix[a-1][b] = 0;
    
15.         }
    
16.         if(a+1 < len){
    
17.             if(matrix[a+1][b] == 0){
    
18.                 matrix[a+1][b] = 1;
    
19.             }else matrix[a+1][b] = 0;
    
20.         }
    
21.         if(b - 1 >= 0){
    
22.             if(matrix[a][b-1] == 0){
    
23.                 matrix[a][b-1]  = 1;
    
24.             }else matrix[a][b-1]  = 0;
    
25.         }
    
26.         if(b+1 < len){
    
27.             if(matrix[a][b+1] == 0){
    
28.                 matrix[a][b+1] = 1;
    
29.             }else matrix[a][b+1] = 0;
    
30.         }
    
31.     }
    
32.     return matrix;
    
33. }
    
34. 
    
35. int main(void){
    
36.     string input, input2;
    
37.     cin >> input;
    
38.     cin.clear();
    
39.     cin >> input2;
    
40.     vector<vector<int> > matrix;
    
41.     vector<int> temp;
    
42.     for(int i = 0; i < input.size(); i++){
    
43.         if(input[i] == '[' || input[i] == ']' || input[i] == ',') continue;
    
44.         if(temp.size() == 4){
    
45.             matrix.push_back(temp);
    
46.             temp.clear();   
    
47.         }
    
48.         temp.push_back(input[i]- '0');
    
49.     }
    
50.     matrix.push_back(temp);
    
51. 
    
52. 
    
53.     vector<vector<int> > change;
    
54.     vector<int> temp1;
    
55.     for(int i = 0; i < input2.size(); i++){
    
56.         if(input2[i] == '[' || input2[i] == ']' || input2[i] == ',') continue;
    
57.         if(temp1.size() == 2){
    
58.             change.push_back(temp1);
    
59.             temp1.clear();   
    
60.         }
    
61.         temp1.push_back(input2[i]- '0');
    
62.     }
    
63.     change.push_back(temp1);
    
64. 
    
65.     vector<vector<int> > res = solution(matrix, change);
    
66.     string out = "[";
    
67.     for(int i = 0; i < res.size(); i++){
    
68.         string temp3 = "[";
    
69.         for(int j = 0; j < res[i].size(); j++){
    
70.             if(j != res[i].size()-1)temp3 += to_string(res[i][j]) + ",";
    
71.             else temp3 += to_string(res[i][j]) + "]";   
    
72.         }
    
73.         if(i != res.size()-1) out += temp3 + ",";
    
74.         else out += temp3 + "]";
    
75.     }
    
76.     cout << out << endl;
    
77. 
    
78. 
    
79.     return 0;
    
80. }

复制代码

糖逗

这C++读入要写到“吐血”

糖逗

这题感觉用python读入会更方便

糖逗