สมมติว่าเรามีเมทริกซ์แมท[][] มีตัวละคร Z และ P โดย Z คือซอมบี้ และ P เป็นพืช และตัวละครอื่น * คือดินแดนที่ว่างเปล่า ซอมบี้สามารถโจมตีต้นไม้ได้ เมื่อต้นไม้อยู่ติดกับซอมบี้ เราต้องหาต้นไม้จำนวนหนึ่งที่ปลอดภัยจากซอมบี้ สมมติว่าเมทริกซ์อยู่ด้านล่าง -
จึงมีพืชที่ปลอดภัยเพียงสองต้นเท่านั้น
เราจะสำรวจองค์ประกอบเมทริกซ์ทีละองค์ประกอบ จากนั้นเมื่อองค์ประกอบปัจจุบันเป็นต้นไม้ ให้ตรวจสอบว่าพืชนั้นล้อมรอบด้วยซอมบี้หรือไม่ ถ้าไม่ก็ให้เพิ่มจำนวน
ตัวอย่าง
#include<iostream> using namespace std; bool isZombie(int i, int j, int r, int c, string mat[]) { if (i < 0 || j < 0 || i >= r || j >= c || mat[i][j] != 'Z') return false; return true; } int countSafeCells(string matrix[], int row, int col) { int i, j, count = 0; for (i = 0; i < row; i++) { for (j = 0; j < col; j++) { if (matrix[i][j] == 'P') { if (!isZombie(i - 1, j - 1, row, col, matrix) && !isZombie(i - 1, j, row, col, matrix) && !isZombie(i - 1, j + 1, row, col, matrix) && !isZombie(i, j - 1, row, col, matrix) && !isZombie(i, j, row, col, matrix) && !isZombie(i, j + 1, row, col, matrix) && !isZombie(i + 1, j - 1, row, col, matrix) && !isZombie(i + 1, j, row, col, matrix) && !isZombie(i + 1, j + 1, row, col, matrix)) { count++; } } } } return count; } int main() { string mat[] = { "**P*", "Z***", "*P**", "***P" }; int row = sizeof(mat) / sizeof(mat[0]); int col = mat[0].length(); cout << "Number of safe cells: " << countSafeCells(mat, row, col); }
ผลลัพธ์
Number of safe cells: 2