其实这道题还是在考最短路,只是要在直线上方和下方各求一条最短路,我们可以执行两次求最短路操作。路径长度设为从s到该点路径上的像素和,路径更新可以仿照dijkstra,找到最近的点,最近点可以用优先队列(小根堆)求得,然后向其八个方向扩展,且斜线前行时要额外增加一个值(见题目要求)。
样例的解如下:
#include <iostream>
#include <cstring>
#include <queue>
#include <cmath>
#define N 110
using namespace std;
int n,m,mp[N][N],vis[N][N],f;
int dx[8]={0,0,1,-1,1,1,-1,-1},dy[8]={1,-1,0,0,1,-1,1,-1};
double dist[N][N],ans;
struct point{
int x,y;
double d;
bool operator < (const point &p) const{//优先队列
return d>p.d;
}
}s,e;
int judge(point p){
if(p.x<0||p.x>=n||p.y<0||p.y>=m||vis[p.x][p.y]) return 0;
if(p.x==e.x&&p.y==e.y) return 1;
double d=(e.y-s.y)*(p.x-s.x)-(p.y-s.y)*(e.x-s.x);//判断在se直线上方还是下方
if(f&&d>0) return 1;//直线上方
if(!f&&d<0) return 1;//直线下方
return 0;
}
void dijk(){
for(int i=0;i<n;i++)
for(int j=0;j<m;j++) dist[i][j]=1e9;
priority_queue<point> q;
dist[s.x][s.y]=0,vis[s.x][s.y]=1;
q.push(s);
point t,o;
while(!q.empty()){
t=q.top(); q.pop();
if(t.x==e.x&&t.y==e.y) break;
vis[t.x][t.y]=1;
for(int i=0;i<8;i++){//八个方向前进
o.x=t.x+dx[i],o.y=t.y+dy[i];
if(!judge(o)) continue;
double d=t.d+mp[o.x][o.y];//直线前进,距离+下一个点的分数
if(i>=4) d+=(mp[t.x][t.y]+mp[o.x][o.y])*(sqrt(2.0)-1.0);//斜线前进,距离再+斜线上两点分数和*(sqrt(2)-1)
if(d<dist[o.x][o.y]){//距离变小
o.d=dist[o.x][o.y]=d;
q.push(o);
}
}
}
}
int main(){
cin>>n>>m;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++) cin>>mp[i][j];
cin>>s.y>>s.x>>e.y>>e.x;
s.d=mp[s.x][s.y];
dijk();//直线下方的路线
ans+=dist[e.x][e.y];
f=1;
dijk();//直线上方的路线
ans+=dist[e.x][e.y];
printf("%.2lf\n",ans-mp[s.x][s.y]-mp[e.x][e.y]);//去掉重复计算点的分数
return 0;
}