矩阵中任意多个点到某一点最短距离

输入

5 5 
0 0 0 0 0
8 8 8 0 0
0 0 8 0 0
8 8 8 8 0
0 0 0 0 0 

4 
1 1 
1 5 
5 1 
5 5 
#include <iostream>
#include <vector>
#include <queue>

using namespace std;

vector<int> direction{-1, 0, 1, 0, -1};

void dfs(queue<pair<int, int>>& points, vector<vector<int>> &grid, int m, int n
        , int i, int j) {
    if (i < 0 || j < 0 || i == m || j == n || grid[i][j] == 2 || grid[i][j] == 8) {
        return;
    }
    if (grid[i][j] == 0) {
        points.push({i, j});
        return;
    }
    grid[i][j] = 2;
    dfs(points, grid, m, n, i - 1, j);
    dfs(points, grid, m, n, i + 1, j);
    dfs(points, grid, m, n, i, j - 1);
    dfs(points, grid, m, n, i, j + 1);
}


int shortestDist(vector<vector<int>> grid,int x1,int y1,int x2,int y2) {

    grid[x2][y2] = 1;

    int m = grid.size(), n = grid[0].size();
    queue<pair<int, int>> points;
    points.push({x1, y1});
    grid[x1][y1] = 2;

// bfs寻找grid[x2][y2],并把过程中经过的0赋值为2
    int x, y;
    int level = 0;
    while (!points.empty()){
        ++level;
        int n_points = points.size();
        while (n_points--) {
            auto [r, c] = points.front();
            points.pop();
            for (int k = 0; k < 4; ++k) {
                x = r + direction[k], y = c + direction[k+1];
                if (x >= 0 && y >= 0 && x < m && y < n) {
                    if (grid[x][y] == 2) {
                        continue;
                    }
                    if (grid[x][y] == 8) {          //障碍
                        continue;
                    }
                    if (grid[x][y] == 1) {          //找到另一端
                        return level;
                    }
                    points.push({x, y});
                    grid[x][y] = 2;
                }
            }
        }
    }
    return -1;          //不通
}

int judge(vector<vector<int>> data,vector<vector<int>> &input)
{
    int min = INT_MAX;
    for(int i=0;i<data.size();i++)
    {
        for(int j=0;j<data[0].size();j++)
        {
            int max = 0;
            for(int l = 0;l<input.size();l++)
            {
                if(data[i][j] == 8)
                {
                    max = INT_MAX;
                    break;
                }
                else
                {
                    if(i == input[l][0] && j == input[l][1])
                        continue;
                    else{
                        int temp = shortestDist(data,input[l][0],input[l][1],i,j);
                        if(temp == -1)
                        {
                            max = INT_MAX;
                            break;
                        }
                        if(max < temp)
                            max = temp;
                    }
                }
            }
            cout<<"data["<<i<<"]["<<j<<"] temp is "<<max<<endl;
            if(i == input[0][0] && j == input[0][1] && max == -1)
            {
                return -1;
            }
            if(min > max )
                min = max;

        }
    }
    return min;
}


int main()
{

    freopen("D:\\desktop\\Leecode_dbg\\data.txt","r",stdin);
    int m,n;
    cin>>m>>n;
    vector<vector<int>> dist(m,vector<int>(n));
    for(int i=0;i<m;i++)
    {
        for(int j=0;j<n;j++)
        {
            cin>>dist[i][j];
        }
    }
    int k;
    cin>>k;
    vector<vector<int>> loc(k,vector<int>(2));

    for(int i=0;i<k;i++)
    {
        int temp;
        for(int j=0;j<2;j++)
        {
            cin>>temp;
            loc[i][j] = temp -1;
        }
    }

    cout<<judge(dist,loc)<<endl;
    return 0;
}




上一篇:452 用最少数量的箭引爆气球(求解区间相交的个数)


下一篇:3D点云语义分割篇——PointNet