P3956 [NOIP2017 普及组] 棋盘

P3956 [NOIP2017 普及组] 棋盘

有一个m×m的棋盘，棋盘上每一个格子可能是红色、黄色或没有任何颜色的。你现在要从棋盘的最左上角走到棋盘的最右下角。

任何一个时刻，你所站在的位置必须是有颜色的（不能是无色的）， 你只能向上、 下、左、 右四个方向前进。当你从一个格子走向另一个格子时，如果两个格子的颜色相同，那你不需要花费金币；如果不同，则你需要花费 1个金币。

另外， 你可以花费 2 个金币施展魔法让下一个无色格子暂时变为你指定的颜色。但这个魔法不能连续使用， 而且这个魔法的持续时间很短，也就是说，如果你使用了这个魔法，走到了这个暂时有颜色的格子上，你就不能继续使用魔法； 只有当你离开这个位置，走到一个本来就有颜色的格子上的时候，你才能继续使用这个魔法，而当你离开了这个位置（施展魔法使得变为有颜色的格子）时，这个格子恢复为无色。

现在你要从棋盘的最左上角，走到棋盘的最右下角，求花费的最少金币是多少？

解法优先队列维护的bfs找最短路径，权值相等

#include<iostream>
#include<string>
#include<queue>
#include<algorithm>
#include<utility>
#include<cstring>
using namespace std;
const int N = 1e3+11; 
int n,m;
int g[N][N];
int v[N][N];
int tt[N][N]; 
//int xx[8]={0,0,1,-1,1,1,-1,-1},
//	yy[8]={1,-1,0,0,-1,1,1,-1};
//int xx[12]={2,1,2,-1,-2,1,-2,-1,2,2,-2,-2},
//	yy[12]={1,2,-1,2,1,-2,-1,-2,-2,2,-2,2};
int xx[4]={0,0,1,-1},
	yy[4]={1,-1,0,0};
typedef pair<int,int> NO;
typedef pair<int,NO> E;
int bfs(int x,int y){
	if(g[x][y]==0) return -1;
	v[x][y]=1;
	priority_queue<E,vector<E>,greater<E> >que;
	que.push((E){0,(NO){x,y}});
	while(que.size()){
		E t = que.top();
		que.pop();int tx=t.second.first,ty=t.second.second;
		if(tx==n&&ty==n)return t.first;
		for(int i=0;i<4;i++){
			int dx=tx+xx[i],dy=ty+yy[i];
			if(dx<1||dy<1||dx>n||dy>n||v[dx][dy]) continue;
			if(g[tx][ty]==0&&g[dx][dy]==0) continue;
			if(g[tx][ty]==0){
				if(tt[tx][ty]==g[dx][dy]) que.push((E){t.first,(NO){dx,dy}});
				else que.push((E){t.first+1,(NO){dx,dy}});
				v[dx][dy]=1;
			}else{
				if(g[dx][dy]==0){
					tt[dx][dy]=g[tx][ty];
					que.push((E){t.first+2,(NO){dx,dy}});
					v[dx][dy]=1;
				}else{
					if(g[dx][dy]==g[tx][ty]) que.push((E){t.first,(NO){dx,dy}});
					else que.push((E){t.first+1,(NO){dx,dy}});
					v[dx][dy]=1;
				}
			}
		}
	}
	return -1;
}
int main(){
	scanf("%d%d",&n,&m);
	int x,y,z;
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&x,&y,&z);
		g[x][y]=z+1;
	}
//	for(int i=1;i<=n;i++,cout<<endl)for(int j=1;j<=n;j++) printf("%d ",g[i][j]);
	int ans = bfs(1,1);
	cout<<ans<<endl;
	return 0;
}