Computer >> Máy Tính >  >> Lập trình >> C ++

Magic Squares In Grid trong C ++

Giả sử chúng ta có một lưới, chúng ta phải tìm số ô vuông con ma thuật trong lưới đó. Hình vuông ma thuật là một lưới 3 x 3 chứa đầy các số phân biệt từ 1 đến 9 sao cho mỗi hàng, cột và cả hai đường chéo đều có tổng bằng nhau.

Vì vậy, nếu đầu vào giống như

4 3 8 4
9 5 1 9
2 7 6 2

thì đầu ra sẽ là 1, vì hình vuông ma thuật là

4 3 8
9 5 1
2 7 6

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • Xác định một tập hợp với các giá trị:[816357492, 834159672, 618753294, 672159834,492357816, 438951276, 294753618, 276951438]
  • Xác định độ lệch mảng có kích thước:9 x 2:={{-2, -2}, {- 2, -1}, {- 2,0}, {- 1, -2}, {- 1 , -1}, {- 1,0}, {0, -2}, {0, -1}, {0,0}}
  • ans:=0
  • để khởi tạo i:=2, khi i <đếm hàng lưới, cập nhật (tăng i lên 1), thực hiện -
    • để khởi tạo j:=2, khi j <đếm hàng lưới, cập nhật (tăng j lên 1), thực hiện -
      • sum:=0
      • để khởi tạo k:=0, khi k <9, cập nhật (tăng k lên 1), thực hiện -
        • sum:=sum * 10
        • sum:=sum + grid [i + offset [k, 0], j + offset [k, 1]]
      • ans:=ans + sự xuất hiện của tổng trong s
  • trả lại ans

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

Ví dụ

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   int numMagicSquaresInside(vector<vector<int>>& grid) {
      const unordered_set<int> s{816357492, 834159672, 618753294,
      672159834,492357816, 438951276, 294753618,276951438};
      const int offset[][2] = {{-2, -2}, {-2, -1}, {-2, 0},{-1, -2}, {-1, -1}, {-1, 0},
{ 0, -2}, { 0, -1}, { 0, 0}};
      int ans = 0;
      for(int i = 2; i< grid.size(); i++)
      {
         for(int j = 2; j<grid.size(); j++)
         {
            int sum = 0;
            for(int k = 0; k<9; k++)
            {
               sum *= 10;
               sum += grid[i + offset[k][0]][j+offset[k][1]];
            }
            ans += s.count(sum);
         }
      }
      return ans;
   }
};
main(){
   Solution ob;
   vector<vector<int>> v = {{4,3,8,4},{9,5,1,9},{2,7,6,2}};
   cout << (ob.numMagicSquaresInside(v));
}

Đầu vào

{{4,3,8,4},{9,5,1,9},{2,7,6,2}}

Đầu ra

1