網格中的幻方在 C++ 中
假設我們有一個網格,我們必須找到該網格中魔法正方形子網格的數量。一個魔法正方形是一個 3 x 3 網格,裡面填有從 1 到 9 的不同數字,使得每一行、每一列和兩條對角線上的數字總和都相同。
因此,如果輸入如下:
| 4 | 3 | 8 | 4 |
| 9 | 5 | 1 | 9 |
| 2 | 7 | 6 | 2 |
則輸出為 1,因為魔法正方形是
| 4 | 3 | 8 |
| 9 | 5 | 1 |
| 2 | 7 | 6 |
為了解決這個問題,我們將遵循以下步驟:
- 用值定義一個集合:[816357492, 834159672, 618753294, 672159834, 492357816, 438951276, 294753618, 276951438]
- 定義大小為:9 x 2 的偏移陣列:{{ -2,-2},{-2,-1},{-2,0},{-1,-2},{-1,-1},{-1,0},{0,-2},{0,-1},{0,0}}
- ans := 0
- 對於初始化 i := 2,當 i < grid 行計數時,更新(將 i 增加 1),執行 -
- 對於初始化 j := 2,當 j < grid 行計數時,更新(將 j 增加 1),執行 -
- sum := 0
- 對於初始化 k := 0,當 k < 9 時,更新(將 k 增加 1),執行 -
- sum := sum * 10
- sum := sum + grid[i + offset[k, 0], j + offset[k, 1]]
- ans := ans + s 中 sum 的出現次數
- 對於初始化 j := 2,當 j < grid 行計數時,更新(將 j 增加 1),執行 -
- 返回 ans
讓我們看看以下實現以獲得更好的理解 -
示例
#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));
}投入
{{4,3,8,4},{9,5,1,9},{2,7,6,2}}輸出
1
廣告
資料結構
網路
RDBMS
作業系統
Java
iOS
HTML
CSS
安卓
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP