Sicily 1889. Max’s game (优先队列dijkstra、哈希)

来源:互联网 发布:如何开通淘宝卖家 编辑:程序博客网 时间:2024/06/13 06:09


// 题意: 有n*m矩阵,从起点(sx,sy)出发,可以上下左右四个方向移动,// 若两个位置上是相同字符,则花费为0,否则为1,求到终点的最短距离// 用Dijkstra算法解决,但会 TLE  ,需要用 优先队列 优化时间#include <iostream>        // 邻接矩阵+优先队列实现Dijkstra算法#include <stdio.h>#include <queue>#include <cstring>using namespace std;                                const int INF=300000;const int MAXN=250000;int n,m,distD[MAXN],done[MAXN];    int sx,sy,tx,ty;    //起点(sx,sy)和终点(tx,ty)int dx[4]={0,0,1,-1},dy[4]={1,-1,0,0};char data[500][500];typedef pair<int,int> pii;void Dijkstra(int st)    //从源点st到其余各顶点的最短路径长度{    priority_queue<pii, vector<pii>, greater<pii> > q;    fill(distD,distD+n*m,INF);        //结点下标从0开始,共有 m * n 个结点    distD[st]=0;    fill(done,done+n*m,0);    q.push(make_pair(distD[st], st));    while(!q.empty())     {        pii u = q.top();        q.pop();        int p = u.second;        if(done[p])            continue;        done[p] = 1;        if( p == tx * m + ty )    // 如果探测到终点即可退出        {            printf("%d\n",distD[p]);                break;        }        for(int d=0;d<4;++d)        {            int newx=p/m+dx[d],newy=p%m+dy[d];            int newp=newx*m+newy;            if(newx>=0&&newx<n&&newy>=0&&newy<m&& !done[newp] )            {                int w = ( data[p/m][p%m] == data[newx][newy] ) ? 0 : 1 ;                if( distD[p] + w < distD[newp] )                {                    distD[newp]=distD[p]+w;                    q.push(make_pair(distD[newp], newp));                }            }        }    }}int main(){    while(scanf("%d%d",&n,&m)!=EOF)        // n 行 m 列    {        for(int i=0;i<n;++i)        {            scanf("%s",data[i]);        }        scanf("%d%d%d%d",&sx,&sy,&tx,&ty);        Dijkstra(sx*m+sy);    // 顶点坐标[x,y],则在队列对应的下标是 x * m + y    }    return 0;}