10855 - Rotated square

1. #include<iostream>
2. using namespace std;
3.  
4. char Big_array[1000][1000];
5. char small_array[4][1000][1000];
6.  
7. int checkin(int N, int n, int slice){
8.         int counter = 0;
9.         bool flag;
10.  
11.         for (int i = 0; i < N-n+1; i++){
12.                
13.                 for (int j = 0; j < N-n+1; j++){
14.                         flag = true;
15.                         for (int x = 0; x < n; x++){
16.                                 for (int y = 0; y < n; y++){
17.                                         if (Big_array[x + j][y + i] != small_array[slice][x][y]){
18.                                                 flag = false; break;
19.                                         }
20.                                 }
21.                                 if (!flag){
22.                                         break;
23.                                 }
24.                         }
25.                         if (flag){//the small square is found in the big one
26.                                 ++counter;
27.                         }
28.                 }
29.         }
30.         return counter;
31. }
32.  
33. int main()
34. {
35.         int N_big, n_small;
36.         for (;;){
37.                
38.                 cin >> N_big >> n_small;
39.                 if (N_big == 0 && n_small == 0) break;
40.  
41.                 int arr[4] = { 0 };
42.  
43.                 for (int i = 0; i < N_big; i++){
44.                         for (int j = 0; j < N_big; j++){
45.                                 cin >> Big_array[i][j];
46.                         }
47.                 }
48.                 for (int i = 0; i < n_small; i++){
49.                         for (int j = 0; j < n_small; j++){
50.                                 cin >> small_array[0][i][j];
51.                         }
52.                 }
53.  
54.                 for (int i = 0; i < n_small; i++){
55.                         for (int j = 0; j < n_small; j++){
56.                                 /****rotated 90 degrees
57.                                    (0,0) in slice#1 <-- (1,0) in slice#0
58.                                    (0,1) in slice#1 <-- (0,0) in slice#0
59.                                    (1,0) in slice#1 <-- (1,1) in slice#0
60.                                    (1,1) in slice#1 <-- (0,1) in slice#0
61.                                 */
62.                                 small_array[1][i][j] = small_array[0][n_small - 1 - j][i];
63.  
64.                                 /****rotated 180 degrees
65.                                    (0,0) in slice#2 <-- (1,1) in slice#0
66.                                    (0,1) in slice#2 <-- (1,0) in slice#0
67.                                    (1,0) in slice#2 <-- (0,1) in slice#0
68.                                    (1,1) in slice#2 <-- (0,0) in slice#0
69.                                 */
70.                                 small_array[2][i][j] = small_array[0][n_small - 1 - i][n_small - 1 - j];
71.  
72.  
73.                                 /****rotated 270 degrees
                                       (0,0) in slice#3 <-- (0,1) in slice#0
74.                                    (0,1) in slice#3 <-- (1,1) in slice#0
75.                                    (1,0) in slice#3 <-- (0,0) in slice#0
76.                                    (1,1) in slice#3 <-- (1,0) in slice#0
77.                                 */
78.                                 small_array[3][i][j] = small_array[0][j][n_small - 1 - i];
79.                         }
80.                 }
81.  
82.                 for (int i = 0; i < 4; i++){
83.                         arr[i] = checkin(N_big, n_small, i);
84.                 }
85.                 cout <<arr[0]<<" "<<arr[1]<<" "<<arr[2]<<" "<<arr[3]<< endl;
86.         }
87.  
88.         return 0;
89. }