牛客小白月赛75 D矩阵

 这题的边权有1,2所以不能用0,1bfs,虽然我也不是很会用

这题是可以说是个分层图

我们要利用小根堆进行排序,让边权小的排在前面,

实现小根堆有两种方式:

第一种是比较巧妙的,因为优先队列默认实现的是大根堆,所以我们可以把元素取反放进去,因为负数的绝对值越小越大,那么绝对值较小的元素就会被放在前面,我们在取出的时候再取个反,就瞒天过海地用大根堆实现了小根堆。

第二种:

小根堆有自己的声明方式,我们记住即可(我也说不明白道理):

priority_queue,greater >q;

accode:

#include 
using namespace std;
#define pi 3.1415926
#define X first
#define Y second
#define Ysanqian ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
#define endl "\n"
// #define int long long
#define pb push_back
typedef pair PII;
const int N = 1e6 + 100, M = 1010, inf = 0x3f3f3f3f, mod = 1e9 + 7;
int dx[4]={-1,1,0,0},dy[4]={0,0,-1,1};
bool st[M][M];
int dis[M][M];
char g[M][M];
int n,m;
void bfs()
{
  memset(st,0,sizeof st);
  memset(dis,inf,sizeof dis);
  priority_queue> q;
  dis[1][1]=0;
  q.push({0,{1,1}});
  while(q.size())
  {
	auto [x,y]=q.top().second;
	q.pop();
	if(st[x][y])continue;
    st[x][y]=1;
	for(int i=0;i<4;i++)
	{
		int a=x+dx[i],b=y+dy[i];
		if(a<1||a>n||b<1||b>m)continue;
		int w;
		if((a+b-1-1)%2)
		{
		if(g[a][b]==g[1][1])
		w=2;
		else w=1;
		}
		else {
		if(g[a][b]==g[1][1])
		w=1;
		else w=2;
		}
		int d=dis[x][y]+w;
       if(dis[a][b]>d)
	   {
         dis[a][b]=d;
		 q.push({-dis[a][b],{a,b}});
	   }

	}
  }
}
void solve()
{
	cin>>n>>m;
    for(int i=1;i<=n;i++)
    cin>>g[i]+1;
	bfs();
	cout<> T;
	while (T--)solve();
	return 0;
}

你可能感兴趣的:(矩阵,c++,算法,bfs)