数据结构与算法实验6——图论 7-8 森森美图

森森最近想让自己的朋友圈熠熠生辉，所以他决定自己写个美化照片的软件，并起名为森森美图。众所周知，在合照中美化自己的面部而不美化合照者的面部是让自己占据朋友圈高点的绝好方法，因此森森美图里当然得有这个功能。 这个功能的第一步是将自己的面部选中。森森首先计算出了一个图像中所有像素点与周围点的相似程度的分数，分数越低表示某个像素点越“像”一个轮廓边缘上的点。 森森认为，任意连续像素点的得分之和越低，表示它们组成的曲线和轮廓边缘的重合程度越高。为了选择出一个完整的面部，森森决定让用户选择面部上的两个像素点A和B，则连接这两个点的直线就将图像分为两部分，然后在这两部分中分别寻找一条从A到B且与轮廓重合程度最高的曲线，就可以拼出用户的面部了。 然而森森计算出来得分矩阵后，突然发现自己不知道怎么找到这两条曲线了，你能帮森森当上朋友圈的小王子吗？

为了解题方便，我们做出以下补充说明：

• 图像的左上角是坐标原点(0,0)，我们假设所有像素按矩阵格式排列，其坐标均为非负整数（即横轴向右为正，纵轴向下为正）。
• 忽略正好位于连接A和B的直线（注意不是线段）上的像素点，即不认为这部分像素点在任何一个划分部分上，因此曲线也不能经过这部分像素点。
• 曲线是八连通的（即任一像素点可与其周围的8个像素连通），但为了计算准确，某像素连接对角相邻的斜向像素时，得分额外增加两个像素分数和的2​倍减一。例如样例中，经过坐标为(3,1)和(4,2)的两个像素点的曲线，其得分应该是这两个像素点的分数和(2+2)，再加上额外的(2+2)乘以(−1)，即约为5.66。

输入格式：

输入在第一行给出两个正整数N和M（5≤N,M≤100），表示像素得分矩阵的行数和列数。

接下来N行，每行M个不大于1000的非负整数，即为像素点的分值。

最后一行给出用户选择的起始和结束像素点的坐标(Xstart​,Ystart​)和(Xend​,Yend​)。4个整数用空格分隔。

输出格式：

在一行中输出划分图片后找到的轮廓曲线的得分和，保留小数点后两位。注意起点和终点的得分不要重复计算。

输入样例：

6 6
9 0 1 9 9 9
9 9 1 2 2 9
9 9 2 0 2 9
9 9 1 1 2 9
9 9 3 3 1 1
9 9 9 9 9 9
2 1 5 4

输出样例：

27.04

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

#include<bits/stdc++.h>
using namespace std;
const int N = 1e2+5;
const int inf = 0x3f3f3f3f;
const double M = sqrt(2)-1;
int n, m, vis[N][N], sx, sy, ex, ey;
double s[N][N], dis[N][N], ans;
int ne[8][2] = {1,0,-1,0,0,1,0,-1,1,1,-1,-1,1,-1,-1,1};
struct node
{
    int x, y;
}a, b, p;
int cross(node a, node b, node p)
{
    return (b.x-a.x)*(p.y-a.y)-(b.y-a.y)*(p.x-a.x);
}
void bfs()
{
    memset(vis, 0, sizeof vis);
    //memset(dis, 0x5f, sizeof dis);
     for(int i=0; i<n; i++)
         for(int j=0; j<m; j++)
             dis[i][j]=inf;
    a={sx, sy}, b={ex, ey};
    dis[sx][sy]=s[sx][sy];
    queue<node>Q;
    Q.push(a);
    while(!Q.empty())
    {
        int x = Q.front().x, y = Q.front().y;
        Q.pop();
        vis[x][y]=0;
        for(int i=0; i<8; i++)
        {
            double w=0;
            p.x=x+ne[i][0], p.y=y+ne[i][1];
            if(p.x<0||p.x>=n||p.y<0||p.y>=m) continue;
            if(cross(a, b, p)>0||p.x==ex&&p.y==ey)
            {
                w=s[p.x][p.y];
                if(i>3) w+=(s[x][y]+s[p.x][p.y])*M;
                if(dis[p.x][p.y]>dis[x][y]+w)
                {
                    dis[p.x][p.y]=dis[x][y]+w;
                    if(!vis[p.x][p.y])
                    {
                        Q.push(p);
                        vis[p.x][p.y]=1;
                    }
                }
            }
        }
    }
    ans+=dis[ex][ey];
}
int main()
{
    cin >> n >> m;
    for(int i=0; i<n; i++)
        for(int j=0; j<m; j++)
            cin >> s[i][j];
    cin >> sy >> sx >> ey >> ex;
    ans = -s[sx][sy]-s[ex][ey];
    bfs();
    swap(ex, sx), swap(ey, sy);
    bfs();
    printf("%.2f",ans);
    return 0;
}