【定期重构】2019牛客暑期多校训练营（第八场）- D - Distance

题目链接https://ac.nowcoder.com/acm/contest/888/D

---

题意

在一个三维空间(n,m,h)，做两种操作：

1. 添加一个标记点
2. 询问所有标记点距离(x,y,z)最小的曼哈顿距离

---

题解

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N=1e6+7;
const int M=1e5;
int n,m,h,Q;
int is(int x,int y,int z){
    return x*m*h+y*h+z;
}
int dis[N];
struct Node{
    int x,y,z;
};
int dir[6][3]{0,0,1,0,0,-1,0,1,0,0,-1,0,1,0,0,-1,0,0};
queue<Node>q;
vector<Node>add;
int main()
{
    scanf("%d%d%d%d",&n,&m,&h,&Q);
    for(int i=0;i<N;i++) dis[i]=N;
    int limit=sqrt(n*m*h);
    for(int i=1,o,x,y,z;i<=Q;i++){
        scanf("%d%d%d%d",&o,&x,&y,&z);
        if(o==1){
            add.push_back((Node){x,y,z});
            dis[is(x,y,z)]=0;
            if(add.size()>=limit){
                for(int i=0;i<limit;i++) q.push(add[i]);
                add.clear();
                while(!q.empty()){
                    Node t=q.front();q.pop();
                    int u=is(t.x,t.y,t.z);
                    for(int k=0;k<6;k++){
                        int dx=t.x+dir[k][0];
                        int dy=t.y+dir[k][1];
                        int dz=t.z+dir[k][2];
                        if(dx<1||dx>n||dy<1||dy>m||dz<1||dz>h) continue;
                        int v=is(dx,dy,dz);
                        if(dis[v]<=dis[u]+1) continue;
                        dis[v]=dis[u]+1;
                        q.push((Node){dx,dy,dz});
                    }
                }
            }
        }
        else{
            int u=is(x,y,z);
            int ans=dis[u];
            int sz=add.size();
            for(int i=0;i<sz;i++){
                Node t=add[i];
                ans=min(ans,abs(t.x-x)+abs(t.y-y)+abs(t.z-z));
            }
            printf("%d\n",ans);
        }
    }

}