Произвольные точки старта и финиша

kernel.cu

@@ -7,6 +7,11 @@
 #define BLOCK_SIZE 16
 #define MATRIX_SIZE 32
 #define OBSTACLE_PROB 10
+#define START_X 2
+#define START_Y 2
+#define FINISH_X (MATRIX_SIZE - 3)
+#define FINISH_Y (MATRIX_SIZE - 3)
+
 #define INF UINT_MAX // Используем беззнаковый максимум
 
 __global__ void wave_step(int* P, unsigned int* dist, int n, bool* changed) {
@@ -43,8 +48,8 @@ void generate_polygon(int* P, int n) {
     for (int i = 0; i < n*n; i++) {
         P[i] = (rand() % 100 < OBSTACLE_PROB) ? -1 : 0;
     }
-    P[0] = 0;       // Гарантируем, что старт свободен
-    P[n*n - 1] = 0; // Гарантируем, что финиш свободен
+    P[START_X + START_Y * MATRIX_SIZE] = 0;   // Гарантируем, что старт свободен
+    P[FINISH_X + FINISH_Y * MATRIX_SIZE] = 0; // Гарантируем, что финиш свободен
 }
 
 void print_distance_map(int* P, unsigned int* dist, int n) {
@@ -53,9 +58,9 @@ void print_distance_map(int* P, unsigned int* dist, int n) {
             int idx = i * n + j;
             if (P[idx] == -1) 
                 printf("████"); // Препятствие
-            else if (i == 0 && j == 0) 
+            else if (i == START_X && j == START_Y) 
                 printf("S   "); // Старт
-            else if (i == n-1 && j == n-1) 
+            else if (i == FINISH_X && j == FINISH_Y) 
                 printf("F   "); // Финиш
             else if (dist[idx] == INF) 
                 printf(".   "); // Недостижимая область
@@ -85,7 +90,7 @@ int main() {
     // Инициализация расстояний
     unsigned int* dist = (unsigned int*)malloc(n*n*sizeof(unsigned int));
     for (int i = 0; i < n*n; i++) dist[i] = INF;
-    dist[0] = 0; // Стартовая точка
+    dist[START_X + START_Y * MATRIX_SIZE] = 0; // Стартовая точка
 
     // Копирование данных на GPU
     cudaMemcpy(d_P, P, n*n*sizeof(int), cudaMemcpyHostToDevice);
@@ -121,10 +126,10 @@ int main() {
 
     // Проверка результата
     cudaMemcpy(dist, d_dist, n*n*sizeof(unsigned int), cudaMemcpyDeviceToHost);
-    if (dist[n*n - 1] == INF) {
+    if (dist[FINISH_X + FINISH_Y * MATRIX_SIZE] == INF) {
         printf("Path not found!\n");
     } else {
-        printf("Success! Path length: %u\n", dist[n*n - 1]);
+        printf("Success! Path length: %u\n", dist[FINISH_X + FINISH_Y * MATRIX_SIZE]);
     }
 
     printf("Time: %.2f ms\n", milliseconds);