mirror of
https://gitcode.com/Zengtudor/alg2025.git
synced 2025-10-17 21:42:25 +00:00
feat: 添加P3956.cpp实现网格最短路径算法
实现一个基于优先队列的BFS算法,用于计算在特定规则下的网格最短路径。主要功能包括: - 处理网格中的颜色变化成本 - 支持魔法格子特殊处理 - 使用四维数组记录不同状态的最短路径
This commit is contained in:
parent
294780abb5
commit
66ae25f60c
148
src/10/1/P3956.cpp
Normal file
148
src/10/1/P3956.cpp
Normal file
@ -0,0 +1,148 @@
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <queue>
|
||||
#include <cstring>
|
||||
#include <climits>
|
||||
using namespace std;
|
||||
|
||||
const int MAXM = 105;
|
||||
const int INF = INT_MAX;
|
||||
|
||||
|
||||
const int dx[4] = {-1, 0, 1, 0};
|
||||
const int dy[4] = {0, 1, 0, -1};
|
||||
|
||||
struct Node {
|
||||
int x, y;
|
||||
int cost;
|
||||
int lastColor;
|
||||
bool used;
|
||||
|
||||
|
||||
bool operator>(const Node& other) const {
|
||||
return cost > other.cost;
|
||||
}
|
||||
};
|
||||
|
||||
int m, n;
|
||||
int grid[MAXM][MAXM];
|
||||
int dist[MAXM][MAXM][2][3];
|
||||
|
||||
void init() {
|
||||
|
||||
memset(grid, -1, sizeof(grid));
|
||||
|
||||
for (int i = 1; i <= m; i++) {
|
||||
for (int j = 1; j <= m; j++) {
|
||||
for (int k = 0; k < 2; k++) {
|
||||
for (int l = 0; l < 3; l++) {
|
||||
dist[i][j][k][l] = INF;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int bfs() {
|
||||
priority_queue<Node, vector<Node>, greater<Node>> pq;
|
||||
|
||||
|
||||
Node start;
|
||||
start.x = 1;
|
||||
start.y = 1;
|
||||
start.cost = 0;
|
||||
start.lastColor = grid[1][1];
|
||||
start.used = false;
|
||||
|
||||
dist[1][1][0][grid[1][1]] = 0;
|
||||
pq.push(start);
|
||||
|
||||
while (!pq.empty()) {
|
||||
Node current = pq.top();
|
||||
pq.pop();
|
||||
|
||||
int x = current.x;
|
||||
int y = current.y;
|
||||
int cost = current.cost;
|
||||
int lastColor = current.lastColor;
|
||||
bool used = current.used;
|
||||
|
||||
|
||||
if (cost > dist[x][y][used][lastColor]) {
|
||||
continue;
|
||||
}
|
||||
|
||||
|
||||
if (x == m && y == m) {
|
||||
return cost;
|
||||
}
|
||||
|
||||
|
||||
for (int i = 0; i < 4; i++) {
|
||||
int nx = x + dx[i];
|
||||
int ny = y + dy[i];
|
||||
|
||||
|
||||
if (nx < 1 || nx > m || ny < 1 || ny > m) {
|
||||
continue;
|
||||
}
|
||||
|
||||
int nxtColor = grid[nx][ny];
|
||||
if (nxtColor != -1) {
|
||||
int addCost = 0;
|
||||
if (lastColor != nxtColor) {
|
||||
addCost = 1;
|
||||
}
|
||||
int newCost = cost + addCost;
|
||||
int newLastColor = nxtColor;
|
||||
bool newused = false;
|
||||
if (newCost < dist[nx][ny][newused][newLastColor]) {
|
||||
dist[nx][ny][newused][newLastColor] = newCost;
|
||||
Node nxt;
|
||||
nxt.x = nx;
|
||||
nxt.y = ny;
|
||||
nxt.cost = newCost;
|
||||
nxt.lastColor = newLastColor;
|
||||
nxt.used = newused;
|
||||
pq.push(nxt);
|
||||
}
|
||||
}
|
||||
|
||||
else if (nxtColor == -1 && !used) {
|
||||
int magicCost = 2;
|
||||
for (int magicColor = 0; magicColor <= 1; magicColor++) {
|
||||
int addCost = 0;
|
||||
if (lastColor != magicColor) {
|
||||
addCost = 1;
|
||||
}
|
||||
int totalCost = cost + addCost + magicCost;
|
||||
int newLastColor = magicColor;
|
||||
bool newused = true;
|
||||
if (totalCost < dist[nx][ny][newused][newLastColor]) {
|
||||
dist[nx][ny][newused][newLastColor] = totalCost;
|
||||
Node nxt;
|
||||
nxt.x = nx;
|
||||
nxt.y = ny;
|
||||
nxt.cost = totalCost;
|
||||
nxt.lastColor = newLastColor;
|
||||
nxt.used = newused;
|
||||
pq.push(nxt);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
int main() {
|
||||
cin >> m >> n;
|
||||
init();
|
||||
for (int i = 0; i < n; i++) {
|
||||
int x, y, c;
|
||||
cin >> x >> y >> c;
|
||||
grid[x][y] = c;
|
||||
}
|
||||
int result = bfs();
|
||||
cout << result << endl;
|
||||
}
|
Loading…
Reference in New Issue
Block a user