zoukankan      html  css  js  c++  java
  • 刷题 | Lintcode 573. Build Post Office II 解题报告

    [Problem]

    Given a 2D grid, each cell is either a wall 2, an house 1 or empty 0 (the number zero, one, two), find a place to build a post office so that the sum of the distance from the post office to all the houses is smallest.

    Return the smallest sum of distance. Return -1 if it is not possible.

     Notice
    • You cannot pass through wall and house, but can pass through empty.
    • You only build post office on an empty.
    Example

    Given a grid:

    0 1 0 0 0
    1 0 0 2 1
    0 1 0 0 0
    

    return 8, You can build at (1,1). (Placing a post office at (1,1), the distance that post office to all the house sum is smallest.)

    [Idea]

    从房子的角度来看。也是用 BFS 来遍历所有的空地,不过这里就需要在每个空地中记录房子访问次数和房子距离之和了。记录访问次数是为了确保所有房子都能访问到该空地,否则就不是可行解。这种解法的优势是在于当房子数目远小于空地数目时比上面从空地出发进行 BFS 要更快一些。所以这也是一种权衡(trade-off)了,要视数据集的具体情况而定。

    [Code]

    class Node {
    public:
        
        Node(){}
        Node(int xx, int yy, int dist) {
            x = xx;
            y = yy;
            dis = dist;
        }
        int x, y, dis;
    };
    
    
    class Solution {
    private:
        int dx[4] = {1,0,-1,0};
        int dy[4] = {0,1,0,-1};
        
        bool valid(int nx, int ny, int n, int m, vector<vector<int>>& grid, vector<vector<bool>>& visited) {
            if(0 <= nx && nx < n && 0 <= ny && ny < m) {
                if(grid[nx][ny] == 0 && visited[nx][ny] == false) {
                    return  true;
                }
            }
            return false;
        }
        
        void bfs(Node cur, vector<vector<int>>& grid, vector<vector<int>>& dis, vector<vector<int>>& visit_num, int n, int m) {
            queue<Node> q;
            q.push(cur);
            vector<vector<bool>> visited(n, vector<bool>(m)) ;
    
            while(!q.empty()) {
                cur = q.front();
                q.pop();
                visit_num[cur.x][cur.y]++;
    
                for(int i = 0; i < 4; i++) {
                    int nx = cur.x + dx[i];
                    int ny = cur.y + dy[i];
                    if (valid(nx, ny, n, m, grid, visited)) {
                        dis[nx][ny] = dis[nx][ny] + cur.dis + 1;
                        visited[nx][ny] = true;
                        q.push(Node(nx, ny, cur.dis+1));
                    }
                }
            }
        }
        
    public:
        /*
         * @param grid: a 2D grid
         * @return: An integer
         */
        int shortestDistance(vector<vector<int>> &grid) {
            // write your code here
            if(grid.size() == 0)
                return 0;
            int n = grid.size();
            int m = grid[0].size();
            vector<vector<int>> dis(n, vector<int>(m, 0));
            vector<vector<int> > visit_num(n, vector<int>(m)) ;
            
            int house_num = 0;
            for (int i = 0; i < grid.size(); i++) {
                for (int j = 0; j < grid[i].size(); j++) {
                    if (grid[i][j] == 1) {
                        house_num++;
                        bfs(Node(i,j,0), grid, dis, visit_num, n, m);
                    }
                }
            }
            
            int ans = INT_MAX;
            for (int i = 0; i < grid.size(); i++) {
                for(int j = 0; j < grid[i].size(); j++) {
                    if (grid[i][j] == 0 && visit_num[i][j] == house_num){
                        ans = min(ans, dis[i][j]);
                    }
                }
            }
            
            return ans == INT_MAX ? -1 : ans;
        }
    };

    [Reference]

    1. https://www.jiuzhang.com/solution/build-post-office-ii
    2. http://blog.hyoung.me/cn/2017/02/build-post-office/
  • 相关阅读:
    进制
    流程控制
    运算符
    格式化输出
    数据结构-树的遍历
    A1004 Counting Leaves (30分)
    A1106 Lowest Price in Supply Chain (25分)
    A1094 The Largest Generation (25分)
    A1090 Highest Price in Supply Chain (25分)
    A1079 Total Sales of Supply Chain (25分)
  • 原文地址:https://www.cnblogs.com/casperwin/p/7518314.html
Copyright © 2011-2022 走看看